package main import ( "encoding/gob" "errors" "flag" "log" "os" "path" "runtime" "strconv" "strings" "time" "git.trj.tw/golang/mtfosbot/module/apis/twitch" "git.trj.tw/golang/mtfosbot/module/cmd" "git.trj.tw/golang/mtfosbot/module/options" "git.trj.tw/golang/mtfosbot/module/utils" "git.trj.tw/golang/mtfosbot/model" "git.trj.tw/golang/mtfosbot/module/background" "git.trj.tw/golang/mtfosbot/module/config" twitchirc "git.trj.tw/golang/mtfosbot/module/twitch-irc" "git.trj.tw/golang/mtfosbot/router/routes" "github.com/gin-gonic/gin" ) var server *gin.Engine func init() { options.RegFlag() flag.Parse() } func main() { runOptions := options.GetFlag() if runOptions.Help { flag.Usage() return } err := config.LoadConfig(runOptions.Config) if err != nil { log.Fatal(err) } // connect to database db, err := model.NewDB() if err != nil { log.Fatal(err) } defer db.Close() if runOptions.DBTool { cmd.DBTool() } registerTypes() background.SetBackground() // create http server server = routes.NewServ() routes.SetRoutes(server) go twitchirc.InitIRC() // create thumbnail directory conf := config.GetConf() if !utils.CheckExists(conf.ImageRoot, true) { log.Fatal(errors.New("image root not exists")) } if !utils.CheckExists(path.Join(conf.ImageRoot, "thumbnail"), true) { err = os.MkdirAll(path.Join(conf.ImageRoot, "thumbnail"), 0775) if err != nil { log.Fatal(err) } } if !utils.CheckExists(conf.LogImageRoot, true) { log.Fatal(errors.New("log image root not exists")) } go func() { for { PrintMemUsage() time.Sleep(time.Second * 20) } }() server.Run(strings.Join([]string{":", strconv.Itoa(config.GetConf().Port)}, "")) } // PrintMemUsage - func PrintMemUsage() { var m runtime.MemStats runtime.ReadMemStats(&m) // For info on each, see: https://golang.org/pkg/runtime/#MemStats // fmt.Printf("Alloc = %v MiB", bToMb(m.Alloc)) // fmt.Printf("\tTotalAlloc = %v MiB", bToMb(m.TotalAlloc)) // fmt.Printf("\tSys = %v MiB", bToMb(m.Sys)) // fmt.Printf("\tNumGC = %v\n", m.NumGC) // fmt.Printf("HeapAlloc = %v MiB", bToMb(m.HeapAlloc)) // fmt.Printf("\t HeapSys = %v MiB", bToMb(m.HeapSys)) // fmt.Printf("\t NextGC = %v MiB\n", bToMb(m.NextGC)) } func bToMb(b uint64) uint64 { return b / 1024 / 1024 } func registerTypes() { gob.Register(model.Account{}) gob.Register(model.Commands{}) gob.Register(model.DonateSetting{}) gob.Register(model.FacebookPage{}) gob.Register(model.KeyCommands{}) gob.Register(model.LineGroup{}) gob.Register(model.LineMessageLog{}) gob.Register(model.LineUser{}) gob.Register(model.OpayDonateList{}) gob.Register(model.TwitchChannel{}) gob.Register(model.YoutubeChannel{}) gob.Register(twitch.TwitchTokenData{}) gob.Register(twitch.UserInfo{}) gob.Register(map[string]interface{}{}) }