aboutsummaryrefslogtreecommitdiffstats
path: root/src/dma/cmd/threshold/main.go
blob: b98328d183a6c2c17e76a1caa21c84730e530abb (plain)
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
/*
 * 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")
}