src: Add DMA localagent

Change-Id: Ibcee814fbc9a904448eeb368a1a26bbb69cf54aa
Signed-off-by: Toshiaki Takahashi <takahashi.tsc@ncos.nec.co.jp>

1 files changed, 97 insertions, 0 deletions

diff --git a/src/dma/cmd/threshold/transmit.go b/src/dma/cmd/threshold/transmit.go
new file mode 100644
index 00000000..8cac2a88
--- /dev/null
+++ b/src/dma/cmd/threshold/transmit.go
@@ -0,0 +1,97 @@
+/*
+ * 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 (
+	"bytes"
+	"fmt"
+	"github.com/distributed-monitoring/agent/pkg/common"
+	"github.com/go-redis/redis"
+	"strconv"
+	"strings"
+	"time"
+)
+
+type collectdNotifier struct {
+	pluginName string
+	typeName   string
+}
+
+func send(cn collectdNotifier, message string, severity string, metaData [][2]string) error {
+	unixNow := float64(time.Now().UnixNano()) / 1000000000
+
+	var metaDataStr bytes.Buffer
+	for _, data := range metaData {
+		metaDataStr.WriteString(" s:")
+		metaDataStr.WriteString(data[0])
+		metaDataStr.WriteString("=\"")
+		metaDataStr.WriteString(strings.Replace(data[1], "\"", "\\\"", -1))
+		metaDataStr.WriteString("\"")
+	}
+
+	fmt.Printf("PUTNOTIF message=\"%s\" severity=%s time=%f "+
+		"host=localhost plugin=%s type=%s %s\n",
+		message, severity, unixNow, cn.pluginName, cn.typeName, metaDataStr.String())
+
+	return nil
+}
+
+func transmit(config *Config, edlist []evalData) {
+	annoConfig := config.Common
+
+	client := redis.NewClient(&redis.Options{
+		Addr:     annoConfig.RedisHost + ":" + annoConfig.RedisPort,
+		Password: annoConfig.RedisPassword,
+		DB:       annoConfig.RedisDB,
+	})
+	pool := common.RedisPool{Client: client}
+
+	notifier := collectdNotifier{
+		pluginName: config.Threshold.CollectdPlugin,
+		typeName:   config.Threshold.CollectdType}
+
+	for _, ed := range edlist {
+		if ed.label == 1 {
+
+			fmt.Println("kick action")
+
+			item := strings.Split(ed.key, "/")
+			ifItem := strings.SplitN(item[3], "-", 2)
+			virtName := item[1]
+			virtIF := ifItem[1]
+
+			var message bytes.Buffer
+			message.WriteString("Value exceeded threshold ")
+			message.WriteString(strconv.Itoa(config.Threshold.Min))
+			message.WriteString(".")
+
+			nameVal, _ := pool.Get(fmt.Sprintf("%s/%s/vminfo", "vm", virtName))
+			ifVal, _ := pool.Get(fmt.Sprintf("%s/%s/neutron_network", "if", virtIF))
+
+			nameInfo := fmt.Sprintf("{\"%s\": %s}", virtName, nameVal)
+			ifInfo := fmt.Sprintf("{\"%s\": %s}", virtIF, ifVal)
+
+			fmt.Println(nameInfo)
+			fmt.Println(ifInfo)
+
+			send(notifier, message.String(),
+				"warning",
+				[][2]string{{"vminfo", nameInfo}, {"neutron_network", ifInfo}})
+
+		}
+	}
+}