新建config目錄，初始化并監聽文件：
package config
import (
"fmt"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
)
type Config struct {
Name string
}
// 初始化配置
func Init(cfg string) error {
c := Config{
Name: cfg,
}
if err := c.initConfig(); err != nil {
return err
}
c.watchConfig()
return nil
}
func (c *Config) initConfig() error {
if c.Name != "" {
viper.SetConfigFile(c.Name)
} else {
// 默認配置文件是conf/config.yaml
viper.AddConfigPath("conf")
viper.SetConfigName("config")
}
viper.SetConfigType("yaml")
// viper解析配置文件
err := viper.ReadInConfig()
if err != nil {
panic(fmt.Errorf("Fatal error config file: %s \n", err))
}
fmt.Println(viper.GetString("name"))
return nil
}
func (c *Config) watchConfig() {
viper.WatchConfig()
viper.OnConfigChange(func(e fsnotify.Event) {
fmt.Println("Config file changed:", e.Name)
})
}conf/config.yaml，語法可自行研究下，比較簡單。
YAML入門
code
name: gin-frame
db:
name: blog
host: 127.0.0.1:3306
username: root
password: 111111
charset: utf8mb4連接數據庫，構建連接池：
go
package model
import (
"fmt"
"sync"
"errors"
orm "github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
"github.com/spf13/viper"
)
type MySqlPool struct {}
var instance *MySqlPool
var once sync.Once
var db *orm.DB
var err error
// 單例模式
func GetInstance() *MySqlPool {
once.Do(func() {
instance = &MySqlPool{}
})
return instance
}
func (pool *MySqlPool) InitPool() (isSuc bool) {
dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=%s", viper.GetString("db.username"), viper.GetString("db.password"), viper.GetString("db.host"), viper.GetString("db.name"), viper.GetString("db.charset"))
db, err = orm.Open("mysql", dsn)
if err != nil {
panic(errors.New("mysql連接失敗"))
return false
}
// 連接數配置也可以寫入配置，在此讀取
db.DB().SetMaxIdleConns(50)
db.DB().SetMaxOpenConns(50)
// db.LogMode(true)
return true
}我們完善一下main.go，初始化配置，并構建連接池：
go
package main
// import 這里我習慣把官方庫，開源庫，本地module依次列出
import (
"log"
"os"
"errors"
"github.com/spf13/pflag"
"local.com/sai0556/gin-frame/config"
"local.com/sai0556/gin-frame/model"
)
var (
conf = pflag.StringP("config", "c", "", "config filepath")
)
func main() {
pflag.Parse()
// 初始化配置
if err := config.Init(*conf); err != nil {
panic(err)
}
// 連接mysql數據庫
isSuc := model.GetInstance().InitPool()
if !isSuc {
log.Println("init database pool failure...")
panic(errors.New("init database pool failure"))
}
}
寫完不妨運行一下，看看效果吧！
code
go run main.go -c=./conf/config.yaml
在上篇里，我介紹了讀取配置，并嘗試連接了數據庫，那么這一篇呢，我們主要利用gin框架來寫寫簡單的接口。
為了便于管理，還是將路由文件單獨出來，新建routes：
go
package router
import (
"net/http"
"github.com/gin-gonic/gin"
"local.com/sai0556/gin-frame/controller"
)
func Load(g *gin.Engine) *gin.Engine {
g.Use(gin.Recovery())
// 404
g.NoRoute(func (c *gin.Context) {
c.String(http.StatusNotFound, "404 not found");
})
g.GET("/", controller.Index)
return g
}上面的代碼中我們看到了controller，我們建一個目錄controller：
先建base.go文件，用于寫一些基礎的方法，如SendResponse返回json。
go
package controller
import (
"net/http"
"github.com/gin-gonic/gin"
)
type Response struct {
Code int json:"code"
Message string json:"message"
Data interface{} json:"data"
}
func SendResponse(c *gin.Context, code int, message string, data interface{}) {
c.JSON(http.StatusOK, Response{
Code: code,
Message: message,
Data: data,
})
}再來寫個index.go，處理邏輯。
go
package controller
import (
"github.com/gin-gonic/gin"
)
func Index(c *gin.Context) {
SendResponse(c, 0, "success", nil)
}go
// main.go
// 在連接數據庫后加入以下代碼
gin.SetMode("debug")
g := gin.New()
g = router.Load(g)
g.Run(":8080")不妨啟動看看效果。
go run main.go -c=./conf/config.yaml
當然，這里的服務啟動、停止可以寫得再優雅一些。
前兩篇我們已經完成了gin+gorm部分，今天我們來補充go-Redis，并進行測試。
我們把Redis相關也放在model下面，使用的是常見的go-redis：
go
// redis.go
package model
import (
"fmt"
"github.com/spf13/viper"
"github.com/go-redis/redis"
)
var RedisClient *redis.Client
func RedisInit() {
RedisClient = redis.NewClient(&redis.Options{
Addr: fmt.Sprintf("%s:%s", viper.GetString("redis.host"), viper.GetString("redis.port")),
Password: viper.GetString("redis.auth"),
DB: 0,
})
_, err := RedisClient.Ping().Result()
if err != nil {
panic("redis ping error")
}
}
然后在連接Mysql的前面加入初始化redis連接的操作即可。
go
// redis 初始化
model.RedisInit()你可以做一些簡單操作，或者在redis.go做一些常用方法的封裝，比較簡單，就不贅述了，更多go-redis操作可見：
新建測試目錄test，建立三個文件：
// index.go
package test
import (
"net/http"
"io/ioutil"
)
func Sum(a int, b int) int {
return a+b
}
func HttpIndex() []byte {
resp, err := http.Get("http://127.0.0.1:8080/")
if err != nil && resp.StatusCode != 200 {
panic(err)
}
//關閉連接
defer resp.Body.Close()
//讀取報文中所有內容
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
panic(err)
}
//輸出內容
return body// index_test.go
package test
import (
"testing"
"encoding/json"
"local.com/sai0556/gin-frame/controller"
)
func TestSum(t *testing.T) {
ret := Sum(2, 7)
if ret != 9 {
t.Error("Expected 9 ~wow~")
}
}
func TestHttpIndex(t *testing.T) {
data := HttpIndex()
target := controller.Response{}
// json轉換
if err := json.Unmarshal(data, &target); err != nil {
t.Error(target)
}
ret := controller.Response{0, "success", nil}
if target != ret {
t.Error("json error")
}
}// index_bench_test.go
package test
import (
"testing"
)
func BenchmarkSum(b *testing.B) {
for i := 0; i < b.N; i++ {
Sum(2, 7)
}
}
func BenchmarkHttpIndex(b *testing.B) {
for i := 0; i < b.N; i++ {
HttpIndex()
}
}私以為go的測試相比其他語言還是比較簡潔的，這里需要注意幾點：
運行起來，看一下：
對圖中做一些說明：
code
// -run="none"不執行基礎單元測試，bench指定基準測試方法
go test -v -run="none" -bench="Bench*"
// 最后一個BenchmarkHttpIndex-4后面測試結果表示
一共執行了11010次，每次執行耗時107392ns（～0.107ms）test標準庫
文章很基礎，主要是介紹了結合了gin+gorm+go-redis，并寫了簡單的測試，是相對基礎的文章，但足以應付一些api接口了。希望對你有幫助，有問題可留言或私信。
點擊查看項目DEMO
文章轉自微信公眾號@SaiWeng