且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

go语言

更新时间:2022-09-27 17:55:40

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
package controllers
 
import (
   "github.com/revel/revel"
   "fmt"
   "time"
   "strconv"
   "strings"
    "database/sql"
   "text/template"
    _"github.com/go-sql-driver/mysql"
)
 
type Spark struct {
   *revel.Controller
}
 
var (
   starlist = map[string]string{
      "xuezhiqian":"薛之谦",
      "wanghan":"汪涵",
      "yangdi":"杨迪",
      "guoxuefu":"郭雪芙",
      "zhangyu":"张宇",
      "liuwei":"刘维",
      "tianyuan":"田源",
      "qianfeng":"钱枫",
      "shenmengchen":"沈梦辰",
      "zhaosilu":"赵思璐",
   }
  db *sql.DB
 
)
type dbObj struct{
   db *sql.DB
}
 
func (c Spark) getcase() *dbObj{
   if db != nil{
      return &dbObj{db:db}
   }else{
      var conf map[string]string
      conf = c.getconf()
      fmt.Println(conf)
      server := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s",conf["user"],conf["pwd"],conf["host"],conf["port"],conf["db"],conf["charset"])
      db, _ = sql.Open("mysql", server) //第一个参数为驱动名
      db.SetMaxOpenConns( 100 )
      db.SetMaxIdleConns( 10 )
      return &dbObj{db}
   }
}
 
func (c Spark) getdb() {
   if db == nil {
      var conf map[string]string
      conf = c.getconf()
      fmt.Println(conf)
      server := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=%s",conf["user"],conf["pwd"],conf["host"],conf["port"],conf["db"],conf["charset"])
      db, _ = sql.Open("mysql", server) //第一个参数为驱动名
      db.SetMaxOpenConns( 100 )
      db.SetMaxIdleConns( 10 )
   }
}
 
func (c Spark) getconf() map[string]string{
   var conf map[string]string
   conf = make(map[string]string,10)
   conf["host"],_ = revel.Config.String("host")
   conf["db"],_ = revel.Config.String("db")
   conf["pwd"],_ = revel.Config.String("pwd")
   conf["user"],_ = revel.Config.String("user")
   conf["port"],_ = revel.Config.String("port")
   conf["charset"],_ = revel.Config.String("charset")
   return conf
}
 
func (c Spark) Add (starname string,url string , username string,cb string) revel.Result {
    starname = template.HTMLEscapeString(strings.TrimSpace(starname))
    username = template.HTMLEscapeString(strings.TrimSpace(username))
    url = template.HTMLEscapeString(strings.TrimSpace(url))
   if starname == "" {
      return c.display(1,"请填写好你支持的明星,再提交!","",cb)
   }
   if url == "" {
      return c.display(2,"请填写好你剪辑好的视频,再提交!","",cb)
   }
 
   newlist := make(map[string]string,len(starlist))
   for key,name := range starlist {
      newlist[name] = key
   }
   _, ok := newlist[starname]
   if(!ok){
      return c.display(3,"填写的明星不存在","",cb)
   }
 
   t := time.Now().Unix()
   tt := strconv.FormatInt(t,10)
   //c.getdb()
   c.getcase()
   //stmt,err := db.Prepare("insert into star (starname,username,videourl,addtime,status) values (?,?,?,?,?)")
   //checkErr(err)
   //res,err := stmt.Exec(starname,username,url,tt,0)
   //checkErr(err)
   //id,err := res.LastInsertId()
   //fmt.Println(id)
   fmt.Println(db)
   _,err := db.Exec("insert into star (starname,username,videourl,addtime,status) values (?,?,?,?,?)",starname,username,url,tt,0)
   _,err = db.Exec("update info set num=num+1 where starname=?",starname)
   //checkErr(err)
   //id,err = res.LastInsertId()
   if err == nil {
      return c.display(0,"ok","提交成功,请耐心等待工作人员审核",cb)
   else {
      return c.display(4,"插入数据库失败","",cb)
   }
 
}
 
func (c Spark) Getstarlist(cb string) revel.Result{
 
   c.getcase()
   rows,err := db.Query("select id,starname, num ,imgurl,videourl ,headimg from info order by num desc")
   checkErr(err)
   defer rows.Close()
   list := make( []map[string]interface{},0,1 )
   for rows.Next() {
      row := make(map[string]interface{})
      var id int
      var starname string
      var num int
      var imgurl string
      var videourl string
      var headimg string
      err = rows.Scan(&id,&starname,&num,&imgurl,&videourl,&headimg)
      row["id"] = id
      row["starname"] = starname
      row["num"] = num
      row["imgurl"] = imgurl
      row["videourl"] = videourl
      row["headimg"] = headimg
      checkErr(err)
      list = append(list,row)
   }
 
   return c.display(0,"ok",list,cb)
}
 
func (c Spark) Getone(name string,cb string) revel.Result{
   //c.getdb()
   c.getcase()
   stmt,err := db.Prepare("select id,starname, num ,imgurl,videourl ,headimg from info where starname=?")
   checkErr(err)
   var id int
   var starname string
   var num int
   var imgurl string
   var videourl string
   var headimg string
   error := stmt.QueryRow(name).Scan(&id,&starname,&num,&imgurl,&videourl,&headimg)
   checkErr(error)
   row := make(map[string]interface{})
   row["id"] = id
   row["starname"] = starname
   row["num"] = num
   row["imgurl"] = imgurl
   row["videourl"] = videourl
   row["headimg"] = headimg
   return c.display(0,"ok",row,cb)
}
 
func checkErr(err error) {
   if err != nil {
      panic(err)
   }
}
 
func (c Spark) display (errno int,msg string,data interface{},cb string)revel.Result{
   var result = make(map[string]interface{},2)
   result["errno"] = errno
   result["msg"] = msg
   result["data"] = data
   if cb != "" {
      return c.RenderJSONP( cb,result )
   else {
      return c.RenderJSON( result )
   }
 
}
本文转自  陈小龙哈   51CTO博客,原文链接:http://blog.51cto.com/chenxiaolong/1944665