gcache/main.go at main · mamil/gcache · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main

/*
$ curl "http://localhost:9999/api?key=Tom"
630
$ curl "http://localhost:9999/api?key=kkk"
kkk not exist
*/

import (
	"flag"
	"fmt"
	"gcache/cache"
	"strconv"
	"strings"

	log "github.com/sirupsen/logrus"
	"github.com/spf13/viper"
)

const (
	HttpPre       = "http://"
	DefaultNodeId = 0
)

var db = map[string]string{
	"Tom":  "630",
	"Jack": "589",
	"Sam":  "567",
}

func initConfig() error {
	viper.SetConfigFile("./config.ini")
	if err := viper.ReadInConfig(); err != nil {
		log.Fatalf("read config fail, err:%v", err)
		return err
	}
	return nil
}

func InitLogger() {
	level := GetLogLevel(viper.GetString("LOG_LEVEL"))
	log.SetLevel(level)
	formatter := &log.TextFormatter{
		ForceColors:   true,
		FullTimestamp: true,
		// DisableQuote:    true,
		TimestampFormat: "2006-01-02 15:04:05.999999999",
	}
	log.SetFormatter(formatter)
	log.Debug("debug log level")
	log.Info("start")
}

func GetLogLevel(logLevelConfig string) log.Level {
	level := log.InfoLevel

	logLevelConfig = strings.ToUpper(logLevelConfig)

	switch logLevelConfig {
	case "DEBUG":
		level = log.DebugLevel
	case "INFO":
		level = log.InfoLevel
	case "ERROR":
		level = log.ErrorLevel
	case "FATAL":
		level = log.FatalLevel
	case "TRACE":
		level = log.TraceLevel
	case "WARN":
		level = log.WarnLevel
	}
	return level
}

func NodePrepare(nodeId int) (startAddr string, addrs []string, g *cache.Group) {
	// 获取节点地址
	countStr := viper.GetString("CACHENODE.Count")
	log.Infof("CACHENODE.Count:%s", countStr)
	countInt, err := strconv.Atoi(countStr)
	if err != nil {
		log.Fatalf("convert countStr fail err:%v", err)
	}
	// var addrs []string
	for i := 0; i < countInt; i++ {
		addrName := fmt.Sprintf("CACHENODE.Addr%d", i)
		addr := HttpPre + viper.GetString(addrName)
		addrs = append(addrs, addr)
	}

	g = createGroup()
	startAddr = HttpPre + viper.GetString(fmt.Sprintf("CACHENODE.Addr%d", nodeId))
	return
}

func StartNode(startAddr string, addrs []string, g *cache.Group) {
	startCacheServer(startAddr, addrs, g)
}

func main() {
	if err := initConfig(); err != nil {
		return
	}

	InitLogger()

	var node int
	var api int
	flag.IntVar(&node, "node", -1, "gcache server node")
	flag.IntVar(&api, "api", -1, "gcache api node")
	flag.Parse()

	// 此节点是api还是node
	if node >= 0 {
		startAddr, addrs, g := NodePrepare(node)
		StartNode(startAddr, addrs, g)
	}
	if api >= 0 {
		apiAddr := HttpPre + viper.GetString("APINODE.InterfaceAddr")
		log.Infof("apiAddr:%s", apiAddr)
		startApiNode(api, apiAddr, DefaultNodeId)
	}
}