fix bug

liuxiaolong

2019-07-05 9ce7c15c897e74c002fd33c07822a6b8a27aabe9

 main.go

@@ -1,31 +1,39 @@
package main

import (
   "basic.com/dbapi.git"
   "flag"
   "github.com/golang/glog"
   "strconv"
   "webserver/extend/config"
   "webserver/models"
   "webserver/router"
   "webserver/service"
)

var envirment = flag.String("e", "dev", "")
var dbIp = flag.String("dbIp","127.0.0.1","default dbIp=127.0.0.1")
var dbPort = flag.String("dbPort","8001","default dbPort=8001")
func main() {
   flag.Parse()
   service.InitService()

   service.TestPushImgMsg()
   p, err := strconv.Atoi(*dbPort)
   if err !=nil {
      p = 8001
   }
   dbapi.Init(*dbIp,p)

   envirment := flag.String("e", "dev", "")
   flag.Usage = func() {
      glog.Exit("flag parse usage !")
   }
   flag.Parse()

   flag.Lookup("alsologtostderr").Value.Set("true")
   flag.Lookup("log_dir").Value.Set("./log")
   config.Init(*envirment)
   models.Init()
   //models.Init()
   r := router.NewRouter()

   r.Run("0.0.0.0:8000")

   defer models.CloseDB()
   //defer models.CloseDB()
   //defer new(gorun.SimpleFaceDetect).FaceDetactClose()
}