main.go

package main

import (
	"RESTfulGo/config"
	"RESTfulGo/model"
	"RESTfulGo/router"
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/lexkong/log"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
	"net/http"
	_ "net/http/pprof"
	"runtime"
	"strconv"
	"time"
)

var (
	cfg     = pflag.StringP("config", "c", "", "apiserver config file path.")
	version = pflag.BoolP("version", "v", false, "show version info")
)

func main() {
	numbers := runtime.NumCPU()
	fmt.Println("nums cpu : ", numbers)
	//pflag.Parse()
	//if *version {
	//	v := v.Get()
	//	marshalled, err := json.MarshalIndent(&v, "", "  ")
	//	if err != nil {
	//		fmt.Printf("%v\n", err)
	//		os.Exit(1)
	//	}
	//
	//	fmt.Println(string(marshalled))
	//	return
	//}
	// init config
	if err := config.Init(*cfg); err != nil {
		panic(err)
	}

	// 配置文件设置gin运行模式
	gin.SetMode(viper.GetString("runmode"))

	// 测试日志打印转存效果
	//testLog()

	g := gin.New()

	middlewares := []gin.HandlerFunc{}

	// routers
	router.Load(g, middlewares...)

	// init db
	go model.DB.Init()
	//defer model.DB.Close()

	//cfg := tars.GetServerConfig()
	//profMux := &tars.TarsHttpMux{}
	//profMux.HandleFunc("/debug/pprof/", pprof.Index)
	//profMux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
	//profMux.HandleFunc("/debug/pprof/profile", pprof.Profile)
	//profMux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
	//profMux.HandleFunc("/debug/pprof/trace", pprof.Trace)
	//tars.AddHttpServant(profMux, cfg.App+"."+cfg.Server+".ProfObj")

	log.Infof("Start to requests on http address: %s", viper.GetString("addr"))
	log.Info(http.ListenAndServe(viper.GetString("addr"), g).Error())

}

func testLog(s chan<- string, index, value int) {
	fmt.Println("testLog:", index, value)
	s <- "testLog" + strconv.Itoa(value)
}

func testLogInfo() {
	for {
		// 延迟1秒
		log.Info("2333333333333333333333333333233333333333333333333333333323333333333333333333333333332333333333333333333333333333")
		time.Sleep(time.Second * 2)
	}
}

// API 服务器健康状态自检
func pingServer() error {
	for i := 9; i < 10; i++ {
		// ping the server By Get request to "health"
		res, err := http.Get(viper.GetString("url") + "/sd/health")
		if err == nil && res.StatusCode == http.StatusOK {
			return nil
		}
		log.Info("time sleep 1 ")
		//time.Sleep(time.Second)
	}
	return errors.New("服务检查错误")
}