启动RESTful API应用
在启动http server前，还需要完成以下四步：
- 初始化配置值
- 载入授权用的键值
- 创建一个mongodb的session对象
- 给mongodb添加索引
初始化配置值
src/taskmanager2/common2/utils.go
package common2
import (
"encoding/json"
"os"
"log"
"gopkg.in/gin-gonic/gin.v1"
"net/http"
)
type (
configuration struct {
Server, MongoDBHost, DBUser, DBPwd, Database string
LogLevel int
}
)
var AppConfig configuration
func initConfig() {
loadAppConfig()
}
func loadAppConfig() {
file, err := os.Open("src/taskmanager2/common2/cfg.json")
defer file.Close()
if err != nil {
log.Fatalf("[loadConfig]: %s\n", err)
}
decoder := json.NewDecoder(file)
AppConfig = configuration{}
err = decoder.Decode(&AppConfig)
if err != nil {
log.Fatalf("[loadAppConfig]: %s\n", err)
}
}
loadAppConfig()中载入cfg.json文件以获取应用参数。
cfg.json（src/taskmanager2/common2/cfg.json）的内容如下（其中的参数都可以自定义）：
{
"Server" : "0.0.0.0:8890",
"MongoDBHost" : "192.168.56.99",
"MongoDBUser" : "",
"MongoDBPwd" : "",
"Database" : "taskdb",
"LogLevel" : 4
}
载入授权用的键值
这里没有使用书中的RSA方式生成公钥私钥，因为其与jwt-go的代码有冲突。可能在新版的jwt-go中有改善。此处的app.k文件中，仅是一个随机字符串而已。
src/taskmanager2/common2/auth.go
const (
keyPath = "src/taskmanager2/key/app.k"
)
var (
verifyKey, signKey []byte
)
func initKeys() {
var err error
signKey, err = ioutil.ReadFile(keyPath)
if err != nil {
log.Fatalf("[initKeys]: %s\n", err)
}
verifyKey, err = ioutil.ReadFile(keyPath)
if err != nil {
log.Fatalf("[initKeys]: %s\n", err)
}
}
创建一个mongodb的session对象
src/taskmanager2/common2/mgoUtils.go
package common2
import (
"log"
"time"
"gopkg.in/mgo.v2"
)
var session *mgo.Session
func GetSession() *mgo.Session {
if session == nil {
var err error
session, err = mgo.DialWithInfo(&mgo.DialInfo{
Addrs: []string{AppConfig.MongoDBHost},
Username:AppConfig.DBUser,
Password:AppConfig.DBUser,
Timeout:60 * time.Second,
})
if err != nil {
log.Fatalf("[GetSession]: %s\n", err)
}
}
return session
}
func createDbSession() {
var err error
session, err = mgo.DialWithInfo(&mgo.DialInfo{
Addrs: []string{AppConfig.MongoDBHost},
Username: AppConfig.DBUser,
Password: AppConfig.DBPwd,
Timeout: 60 * time.Second,
})
if err != nil {
log.Fatalf("[createDbSession]: %s\n", err)
}
}
给mongodb添加索引
src/taskmanager2/common2/mgoUtils.go
func addIndexes() {
var err error
userIndex := mgo.Index{
Key: []string{"email"},
Unique: true,
Background: true,
Sparse: true,
}
taskIndex := mgo.Index{
Key: []string{"createdby"},
Unique: false,
Background: true,
Sparse: true,
}
noteIndex := mgo.Index{
Key: []string{"taskid"},
Unique: false,
Background: true,
Sparse: true,
}
session := GetSession().Copy()
defer session.Close()
userCol := session.DB(AppConfig.Database).C("users")
taskCol := session.DB(AppConfig.Database).C("tasks")
noteCol := session.DB(AppConfig.Database).C("notes")
err = userCol.EnsureIndex(userIndex)
if err != nil {
log.Fatalf("[addIndexes]: %s\n", err)
}
err = taskCol.EnsureIndex(taskIndex)
if err != nil {
log.Fatalf("[addIndexes]: %s\n", err)
}
err = noteCol.EnsureIndex(noteIndex)
if err != nil {
log.Fatalf("[addIndexes]: %s\n", err)
}
}
common包中的初始化逻辑
src/taskmanager2/common2/bootstrapper.go
package common2
// init bootstrapps the application
func init() {
// Initialize AppConfig variable
initConfig()
// Initialize private/public keys for JWT authentication
initKeys()
// Initialize Logger objects with Log Level
setLogLevel(Level(AppConfig.LogLevel))
// Start a MongoDB session
createDbSession()
// Add indexes into MongoDB
addIndexes()
}
启动http server
src/taskmanager2/main.go
package main
import (
"log"
"net/http"
"time"
"taskmanager2/routers2"
)
func main() {
router := routers2.InitRoutes()
// set server
server := &http.Server{
Addr: ":8890",
Handler: router,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
MaxHeaderBytes: 1 << 20,
}
log.Println("Listening...")
server.ListenAndServe()
}
在gogland中启动：
