Create & Init Project...

app/service/ep/saga-agent/conf/BUILD

@@ -0,0 +1,44 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["conf_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = ["//vendor/github.com/smartystreets/goconvey/convey:go_default_library"],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["conf.go"],
+    importpath = "go-common/app/service/ep/saga-agent/conf",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//library/conf:go_default_library",
+        "//library/log:go_default_library",
+        "//vendor/github.com/BurntSushi/toml:go_default_library",
+        "//vendor/github.com/pkg/errors:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/service/ep/saga-agent/conf/conf.go

@@ -0,0 +1,117 @@
+package conf
+
+import (
+	"flag"
+	"os"
+
+	"go-common/library/conf"
+	"go-common/library/log"
+
+	"github.com/BurntSushi/toml"
+	"github.com/pkg/errors"
+)
+
+// Runner ...
+type Runner struct {
+	URL   string
+	Token string
+	Name  string
+}
+
+type config struct {
+	Offline bool
+	Runner  []Runner
+}
+
+// Conf ...
+var (
+	confPath  string
+	client    *conf.Client
+	filename  string
+	reload    chan bool
+	Conf      config
+	RunnerMap map[string]Runner
+	HostName  string
+)
+
+func load() (err error) {
+	var (
+		s  string
+		ok bool
+	)
+	if s, ok = client.Value(filename); !ok {
+		err = errors.Errorf("load config center error [%s]", filename)
+		return
+	}
+
+	log.Info("config data: %s", s)
+	Conf = config{}
+	if _, err = toml.Decode(s, &Conf); err != nil {
+		err = errors.Wrapf(err, "could not decode config err(%+v)", err)
+		return
+	}
+	return
+}
+
+func configCenter() (err error) {
+	if filename == "" {
+		return errors.New("filename is null,please set the environment(RUNNER_REPONAME)")
+	}
+	if client, err = conf.New(); err != nil {
+		panic(err)
+	}
+	if err = load(); err != nil {
+		return
+	}
+	client.Watch(filename)
+	go func() {
+		for range client.Event() {
+			log.Info("config reload")
+
+			if load() != nil {
+				log.Error("config reload error (%v)", err)
+			} else {
+				reload <- true
+			}
+		}
+	}()
+	return
+}
+
+func osHostName() string {
+	var (
+		host string
+		err  error
+	)
+	if host, err = os.Hostname(); err != nil {
+		host = "unKown-Hostname"
+		log.Error("%+v", errors.WithStack(err))
+	}
+	return host
+}
+
+func init() {
+	flag.StringVar(&confPath, "conf", "", "config path")
+}
+
+// Init ...
+func Init() (err error) {
+	filename = os.Getenv("CONF_FILENAME")
+	RunnerMap = make(map[string]Runner)
+	HostName = osHostName()
+	reload = make(chan bool, 1)
+	if confPath == "" {
+		err = configCenter()
+		return
+	}
+	if _, err = toml.DecodeFile(confPath, &Conf); err != nil {
+		log.Error("toml.DecodeFile(%s) err(%+v)", confPath, err)
+		return
+	}
+	return
+}
+
+// ReloadEvent ...
+func ReloadEvent() <-chan bool {
+	return reload
+}

app/service/ep/saga-agent/conf/conf_test.go

@@ -0,0 +1,30 @@
+package conf
+
+import (
+	"flag"
+	"fmt"
+	"testing"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func showConf() {
+	for _, v := range Conf.Runner {
+		fmt.Println(v.URL, v.Token)
+	}
+}
+
+func init() {
+	var err error
+	flag.Set("conf", "../service/agent/runners.toml")
+	if err = Init(); err != nil {
+		panic(err)
+	}
+	showConf()
+}
+
+func TestConf(t *testing.T) {
+	Convey("test conf", t, func() {
+		So(Conf.Runner, ShouldNotBeEmpty)
+	})
+}