New standard GO project layout

[ric-plt/appmgr.git] / cmd / appmgr / config.go

diff --git a/cmd/appmgr/config.go b/cmd/appmgr/config.go
new file mode 100755 (executable)
index 0000000..5e1975e
--- /dev/null
+++ b/cmd/appmgr/config.go
@@ -0,0 +1,56 @@
+/*
+==================================================================================
+  Copyright (c) 2019 AT&T Intellectual Property.
+  Copyright (c) 2019 Nokia
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================================
+*/
+
+package main
+
+import (
+       "flag"
+       "github.com/fsnotify/fsnotify"
+       "github.com/spf13/viper"
+       "log"
+)
+
+const DEFAULT_CONFIG_FILE = "config/appmgr.yaml"
+
+func parseCmd() string {
+       var fileName *string
+       fileName = flag.String("f", DEFAULT_CONFIG_FILE, "Specify the configuration file.")
+       flag.Parse()
+
+       return *fileName
+}
+
+func loadConfig() {
+       viper.SetConfigFile(parseCmd())
+
+       if err := viper.ReadInConfig(); err != nil {
+               log.Fatalf("Error reading config file, %s", err)
+       }
+       log.Printf("Using config file: %s\n", viper.ConfigFileUsed())
+
+       // Watch for config file changes and re-read data ...
+       watch()
+}
+
+func watch() {
+       viper.WatchConfig()
+       viper.OnConfigChange(func(e fsnotify.Event) {
+               log.Println("config file changed ", e.Name)
+       })
+}