package main

import (
	"fmt"
	"github.com/sirupsen/logrus"
	"gitlab.wodcloud.com/apaas/apaas-meshproxy/src/router"
	"net/http"
	"os"
	"time"

	"gitlab.wodcloud.com/apaas/apaas-meshproxy/src/config"

	"github.com/gin-gonic/contrib/ginrus"
	"github.com/spf13/pflag"
)

var (
	argPort   = pflag.Int("port", 8088, "")
	argPrefix = pflag.String("prefix", "/test", "")
	redisUrl  = pflag.String("redisUrl", "redis://k8s.wodcloud.com:16379", "")
	redisTag  = pflag.String("redisTag", "apaas-mesh-proxy", "")
)

func main() {
	pflag.Parse()
	initEnv()
	initConfig()
	server()
}

func server() error {
	config.Prefix = *argPrefix
	handler := router.Load(ginrus.Ginrus(logrus.StandardLogger(), time.RFC3339, true))
	fmt.Println(fmt.Sprintf(":%d", *argPort))
	return http.ListenAndServe(
		fmt.Sprintf(":%d", *argPort),
		handler,
	)
}

//注册环境变量
func initEnv() {
	if len(os.Getenv("AWE_REDIS_CONNECTION")) > 0 {
		*redisUrl = os.Getenv("AWE_REDIS_CONNECTION")
	}
}

//注册公共config参数信息
func initConfig() {
	config.RedisTag = *redisTag
	config.RedisURL = *redisUrl
	config.Prefix = *argPrefix
}