TikBase/main.go at main · T4t4KAU/TikBase · 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
package main

import (
	"github.com/T4t4KAU/TikBase/pkg/config"
	"github.com/T4t4KAU/TikBase/proxy"
)

var logo = " _____ _ _    ____                 \n|_   _(_) | _| __ )  __ _ ___  ___ \n  | | | | |/ /  _ \\ / _` / __|/ _ \\\n  | | | |   <| |_) | (_| \\__ \\  __/\n  |_| |_|_|\\_\\____/ \\__,_|___/\\___|\n"

var (
	ServerConfFile    = "./conf/server-config.yaml"
	BaseConfigFile    = "./conf/base-config.yaml"
	CacheConfigFile   = "./conf/cache-config.yaml"
	ReplicaConfigFile = "./conf/replica-config-1.yaml"
	SliceConfigFile   = "./conf/slice-config.yaml"
)

func start() {
	// 读取服务器配置文件
	serverConf, err := config.ReadServerConfigFile(ServerConfFile)
	if err != nil {
		panic(err)
	}

	var storeConf config.StoreConfig

	// 读取存储配置文件
	switch serverConf.EngineName {
	case "base":
		storeConf, err = config.ReadBaseConfigFile(BaseConfigFile)
	case "cache":
		storeConf, err = config.ReadCacheConfigFile(CacheConfigFile)
	default:
		panic("unknown engine name")
	}
	if err != nil {
		panic(err)
	}

	// 读取副本配置文件
	replicaConf, err := config.ReadReplicaConfigFile(ReplicaConfigFile)
	if err != nil {
		panic(err)
	}

	print("listening at port:", serverConf.Port)
	println("  using engine:", serverConf.EngineName)

	// 启动代理
	err = proxy.Start(serverConf, storeConf, replicaConf)
	if err != nil {
		panic(err)
	}
}

func main() {
	println(logo)
	start()
}