學(xué)習(xí)go也有一段時間了关顷，整理了一下學(xué)習(xí)go的心得體會膏蚓。結(jié)合一個web框架構(gòu)建來體會一下刚梭。demo地址
1 路由以及數(shù)據(jù)庫連接
func StartHttpServer() {
r := mux.NewRouter()
api := r.PathPrefix("/api").Subrouter()
go func() {
errs := http.ListenAndServe(":23456", r)
if errs != nil {
fmt.Println(errs)
}
}()
AddHandlerApi(api)
ch := make(chan os.Signal, 1)
signal.Notify(ch)
s := <-ch
fmt.Println("Got signal:", s)
}
func AddHandlerApi(r *mux.Router) {
// 利用工廠函數(shù)構(gòu)建sql
mysql, _ := db.NewMysqlService(db.IntConfig())
// 路由劃分 重點講解 CenterCtol函數(shù)實現(xiàn)
userApi, err := handler.NewUserHandler(mysql)
if err != nil {
fmt.Printf("userapi db connect is : %v", err)
} else {
r.HandleFunc("/login", util.CenterCtol(userApi.UserLogin, &dto.UserLoginReq{}))
r.HandleFunc("/register", util.CenterCtol(userApi.UserRegister, &dto.UserReigisterReq{}))
r.HandleFunc("/del_user", util.CenterCtol(userApi.DelUser, &dto.DelUserReq{})).Methods("POST")
r.HandleFunc("/user_list", util.CenterCtol(userApi.UserList, &dto.GetUserListReq{})).Methods("GET")
}
}
2 CenterCtol 對HTTP Res Req 封裝
type Response struct {
Code int `json:"code"` // status code
Msg string `json:"msg"`
Data interface{} `json:"data"`
NotEscapeHtml bool `json:"-"`
}
type DtoHandler func(w http.ResponseWriter, r *http.Request)
type HandlerFuncDto func(http.ResponseWriter, *http.Request, DtoAAA) *Response
type DtoAAA interface {
GetError() error
}
func CenterCtol(a HandlerFuncDto, b DtoAAA) http.HandlerFunc {
// 在返回函數(shù)中處理
return func(w http.ResponseWriter, r *http.Request) {
// 定義統(tǒng)一的返回數(shù)據(jù)
var res = &Response{}
// 返回處理函數(shù)
defer func() {
if res.Code == 0 {
res.Msg = "success"
}
// header處理
w.Header().Set("Access-Control-Allow-Origin", "*")
w.Header().Set("Content-Type", "application/json")
v, _ := json.Marshal(res)
w.Write(v)
}()
// body處理
if r != nil && r.Body != nil {
defer r.Body.Close()
}
if err := r.ParseForm(); err != nil {
}
// 定義統(tǒng)一的req數(shù)據(jù)接口
var requestDto DtoAAA = nil
// 接口數(shù)據(jù)處理
if b != nil {
// 反射
dtoType := reflect.TypeOf(b)
requestDto = reflect.New(dtoType.Elem()).Interface().(DtoAAA)
//json.Unmarshal(req, requestDto)
// 自定義解析函數(shù)(抄的)處理’application/x-www-form-urlencoded‘
Unmarshal(r.Form, requestDto)
err := requestDto.GetError()
if err != nil {
res.Code = 1
res.Msg = fmt.Sprintf("err:%v", err)
}
}
// 處理函數(shù)
res = a(w, r, requestDto)
}
}
3 對用戶的增刪改查
// 結(jié)構(gòu)體構(gòu)建 引入需要的結(jié)構(gòu)
type UserHandler struct {
ux *commonModel.UserSql
}
// 工廠函數(shù)創(chuàng)建實例
func NewUserHandler(base *sql.DB) (*UserHandler, error) {
db, err := commonModel.NewUserSql(base)
if err != nil {
return nil, err
}
return &UserHandler{
ux: db,
}, nil
}
// 以登錄來講
func (db *UserHandler) UserLogin(w http.ResponseWriter, r *http.Request, req util.DtoAAA) *util.Response {
res := &util.Response{}
args := req.(*dto.UserLoginReq)
user, err := db.ux.LoginSql(args.Username, args.Password)
if err != nil {
res.Code = 2
res.Msg = fmt.Sprintf("sql err:%v", err)
return res
}
res.Data = user
return res
}
4 sql操作
// 結(jié)構(gòu)體
type UserSql struct {
sql *sql.DB
}
// 工廠函數(shù)
func NewUserSql(config *sql.DB) (*UserSql, error) {
return &UserSql{
sql: config,
}, nil
}
//登錄方法
func (db *UserSql) LoginSql(username string, password string) (*UserTable, error) {
user := &UserTable{}
mysql := fmt.Sprintf("select id,username,password,sex,age from %v where username=? and password=?", user_table)
err := db.sql.QueryRow(mysql, username, password).Scan(&user.Id, &user.Username, &user.Password, &user.Sex, &user.Age)
if err != nil {
return nil, err
}
return user, nil
}
代碼里面還有很多的細節(jié)，可以去細看宁改，對于入門人員來說還是很值得一看的樊破。
換上自己的數(shù)據(jù)庫地址
func IntConfig() SqlConfig {
sql := SqlConfig{
Dsn: "自己的數(shù)據(jù)庫地址",
MaxIdle: 11,
MaxOpen: 11,
}
return sql
}
