initial commit

Author: 김경남

.gitignore

@@ -0,0 +1,4 @@
+debug
+.swp
+test/*
+**/*.scouter

README.md

@@ -0,0 +1,3 @@
+# argos
+
+arogs is mariadb agent for scouter.

_argos.home/.DS_Store

@@ -0,0 +1,23 @@
+{ 
+    "configurations": {
+        "collector.ip": "127.0.0.1",
+        "collector.udp.port": "6100",
+        "collector.tcp.port": "6100",
+        "db.instances": [
+            {
+                "db.ip": "db addr",
+                "db.port": "3306",
+                "db.user": "user_name",
+                "db.password": "user_password",
+                "db.slowquery": "/path/to/slow_query"
+            },
+            {
+                "db.server": "db addr",
+                "db.port": "3308",
+                "db.user": "user_name",
+                "db.password": "user_password",
+                "db.slowquery": "/path/to/slow_query"
+            }
+        ]
+    }
+}

_argos.home/conf/scouter.json

@@ -0,0 +1,102 @@
+package config
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"os"
+	"sync"
+	"time"
+)
+
+type Configuration struct {
+	stopRunning chan bool
+}
+
+var confFilePath string
+var confFile *os.File
+var running = make(chan bool)
+var confFileModdTime time.Time
+var confFileSize int64
+
+type ConfObject struct {
+	Configurations Collector
+}
+
+type Collector struct {
+	IP        string       `json:"collector.ip"`
+	Udpport   string       `json:"collector.udp.port"`
+	Tcpport   string       `json:"collector.tcp.port"`
+	Instances []DBInstance `json:"db.instances"`
+}
+
+type DBInstance struct {
+	IP        string `json:"db.ip"`
+	Port      string `json:"db.port"`
+	User      string `json:"db.user"`
+	Password  string `json:"db.password"`
+	Slowquery string `json:"db.slowquery"`
+}
+
+func (conf *Configuration) load() {
+	var confobj ConfObject
+	fileInfo, e := confFile.Stat()
+	if e != nil {
+		//todo: error handling
+	}
+
+	if confFileModdTime != fileInfo.ModTime() || confFileSize != fileInfo.Size() {
+		file, err := ioutil.ReadFile(confFilePath)
+		if err != nil {
+			//todo :
+		}
+		json.Unmarshal(file, &confobj)
+		confFileSize = fileInfo.Size()
+		confFileModdTime = fileInfo.ModTime()
+	}
+
+}
+
+func (conf *Configuration) init() bool {
+	f, err := os.Open(confFilePath)
+	if err != nil {
+		return false
+	} else {
+		confFile = f
+	}
+	return true
+}
+
+func (conf *Configuration) Start() {
+	go conf.run()
+}
+
+func (conf *Configuration) Stop() {
+	conf.stopRunning <- true
+}
+
+func (conf *Configuration) run() {
+	for {
+		conf.load()
+		time.Sleep(1 * time.Second)
+		select {
+		case <-conf.stopRunning:
+			break
+		default:
+			continue
+		}
+	}
+}
+
+var configure *Configuration
+var once sync.Once
+
+// GetInstance returns configuraton singleton instance
+func GetInstance() *Configuration {
+	once.Do(func() {
+		configure = &Configuration{}
+		if configure.init() {
+			configure.Start()
+		}
+	})
+	return configure
+}

agent/config/configuration.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/scouter-argos/agent/config"
+	"github.com/scouter-argos/agent/manager"
+)
+
+var pidfile string
+var f *os.File
+
+func main() {
+	if len(os.Args) > 1 {
+		readArgs(os.Args)
+	}
+	start()
+
+}
+
+func start() {
+	displayLogo()
+	//workGroup.Add(1)
+	//var conf *config.Configuration
+	config.GetInstance()
+	manager.ServiceStart()
+	writePid()
+
+	for fileExist(pidfile) {
+		time.Sleep(1 * time.Second)
+	}
+
+	return
+}
+
+func stop() {
+	if fileExist(pidfile) {
+		os.Remove(pidfile)
+	} else {
+		fmt.Println("Agent is not running.")
+	}
+}
+
+func displayLogo() {
+	fmt.Println("MySQL Agent.")
+}
+
+func fileExist(filename string) bool {
+	if _, err := os.Stat(filename); err != nil {
+		if os.IsNotExist(err) {
+			return false
+		}
+	}
+	return true
+}
+
+func writePid() error {
+	path, err := os.Getwd()
+	if err == nil {
+		pidfile = path + "/" + strconv.Itoa(os.Getpid()) + ".scouter"
+		f, err = os.Create(pidfile)
+		if err != nil {
+			return err
+		}
+		defer f.Close()
+	} else {
+		return nil
+	}
+	return nil
+}
+
+func usage() {
+	fmt.Printf("need arguements")
+}
+
+func readArgs(args []string) {
+	for _, value := range args {
+		argsItem := strings.Split(value, "=")
+		if argsItem[0] == "scouter.config" {
+			config.ConfFilePath = argsItem[1]
+		}
+	}
+}

agent/main/app.go

@@ -0,0 +1,5 @@
+package manager
+
+func ServiceStart() {
+    
+}

agent/manager/service_manager.go

@@ -0,0 +1 @@
+package dataio

common/dataio/datainputx.go

@@ -0,0 +1,66 @@
+package dataio
+
+import (
+	"bytes"
+	"encoding/binary"
+	"fmt"
+)
+
+/*var INT3_MIN_VALUE int = 0xff800000
+var INT3_MAX_VALUE int = 0x007fffff
+var LONG5_MIN_VALUE int64 = 0xffffff8000000000
+var	LONG5_MAX_VALUE int64 = 0x0000007fffffffff */
+
+// A DataOutputX is a output stream which used write various kinds of data.
+type DataOutputX struct {
+	written int // the wrtten bytes.
+	buffer  *bytes.Buffer
+}
+
+// NewDataOutputX returns DataOutputX object
+func NewDataOutputX() *DataOutputX {
+	out := new(DataOutputX)
+	out.written = 0
+	out.buffer = new(bytes.Buffer)
+	return out
+}
+
+// WriteInt32 write int32 number to buffer.
+func (out *DataOutputX) WriteInt32(value int32) *DataOutputX {
+	out.written += 4
+	err := binary.Write(out.buffer, binary.BigEndian, value)
+	if err != nil {
+		fmt.Println("Failed to binary write : ", err)
+	}
+	return out
+}
+
+// WriteInt16 write int16 number to buffer.
+func (out *DataOutputX) WriteInt16(value int16) *DataOutputX {
+	out.written += 2
+	err := binary.Write(out.buffer, binary.BigEndian, value)
+	if err != nil {
+		fmt.Println("Failed to binary write : ", err)
+	}
+	return out
+}
+
+// WriteInt64 write int64 number to buffer.
+func (out *DataOutputX) WriteInt64(value int64) *DataOutputX {
+	out.written += 8
+	err := binary.Write(out.buffer, binary.BigEndian, value)
+	if err != nil {
+		fmt.Println("Failed to binary write : ", err)
+	}
+	return out
+}
+
+//ReadInt32 reads int32 number from buffer and assign to value.
+func (out *DataOutputX) ReadInt32(value *int32) {
+	err := binary.Read(out.buffer, binary.BigEndian, value)
+	if err != nil {
+		fmt.Println("Faileed to binary read :", err)
+		value = nil
+	}
+
+}

common/dataio/dataoutputx.go

@@ -0,0 +1,14 @@
+package dataio
+
+import "testing"
+
+func TestWriteInt(t *testing.T) {
+	out := NewDataOutputX()
+	out.WriteInt32(100)
+	out.WriteInt32(200)
+	var a int32
+	out.ReadInt32(&a)
+	t.Log("the value of a : %d", a)
+	out.ReadInt32(&a)
+	t.Log("the value of a : %d", a)
+}

common/dataio/dataoutputx_test.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/iwindfree/argos/common/dataio"
+)
+
+func main() {
+	out := dataio.NewDataOutputX()
+	out.WriteInt32(10)
+	out.WriteInt32(100)
+	out.WriteInt32(200)
+	var value int32
+	out.ReadInt32(&value)
+	fmt.Println("the value of  a : %d", value)
+	out.ReadInt32(&value)
+	fmt.Println("the value of  a : %d", value)
+	out.ReadInt32(&value)
+	fmt.Println("the value of  a : %d", value)
+
+}