src: Add DMA localagent

[barometer.git] / src / dma / cmd / threshold / main.go

diff --git a/src/dma/cmd/threshold/main.go b/src/dma/cmd/threshold/main.go
new file mode 100644 (file)
index 0000000..b98328d
--- /dev/null
+++ b/src/dma/cmd/threshold/main.go
@@ -0,0 +1,107 @@
+/*
+ * Copyright 2018 NEC Corporation
+ *
+ *   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 (
+       "context"
+       "fmt"
+       "github.com/BurntSushi/toml"
+       "log"
+       "sync"
+       "time"
+)
+
+// Config is ...
+type Config struct {
+       Common    CommonConfig
+       Threshold ThresholdConfig
+}
+
+// CommonConfig is ...
+type CommonConfig struct {
+       RedisHost     string `toml:"redis_host"`
+       RedisPort     string `toml:"redis_port"`
+       RedisPassword string `toml:"redis_password"`
+       RedisDB       int    `toml:"redis_db"`
+}
+
+// ThresholdConfig is ...
+type ThresholdConfig struct {
+       RedisHost     string `toml:"redis_host"`
+       RedisPort     string `toml:"redis_port"`
+       RedisPassword string `toml:"redis_password"`
+       RedisDB       int    `toml:"redis_db"`
+
+       Interval int `toml:"interval"`
+       Min      int `toml:"min"`
+
+       CollectdPlugin string `toml:"collectd_plugin"`
+       CollectdType   string `toml:"collectd_type"`
+}
+
+type rawData struct {
+       key      string
+       datalist []int
+}
+
+type evalData struct {
+       key   string
+       label int
+}
+
+func main() {
+       var config Config
+       _, err := toml.DecodeFile("/etc/barometer-dma/config.toml", &config)
+       if err != nil {
+               log.Fatalf("Read error of config: %s", err)
+       }
+
+       thresConfig := config.Threshold
+       log.Printf("Raw data redis config Addr:%s:%s DB:%d", thresConfig.RedisHost, thresConfig.RedisPort, thresConfig.RedisDB)
+       if thresConfig.RedisPassword == "" {
+               log.Printf("Raw data redis password is not set")
+       }
+       annoConfig := config.Common
+       log.Printf("Annotate redis config Addr:%s:%s DB:%d", annoConfig.RedisHost, annoConfig.RedisPort, annoConfig.RedisDB)
+       if annoConfig.RedisPassword == "" {
+               log.Printf("Annotate redis password is not set")
+       }
+
+       var waitgroup sync.WaitGroup
+       ctx := context.Background()
+
+       waitgroup.Add(1)
+       go func() {
+               defer waitgroup.Done()
+               ticker := time.NewTicker(time.Duration(config.Threshold.Interval) * time.Second)
+               for {
+                       select {
+                       case <-ctx.Done():
+                               return
+                       case <-ticker.C:
+                               result1 := read(&config)
+                               // analysis()
+                               result2 := evaluate(&config, result1)
+                               transmit(&config, result2)
+                       }
+               }
+       }()
+
+       waitgroup.Wait()
+
+       fmt.Println("End")
+}