wip go go agent

Author: gunlee01

.gitignore

@@ -13,3 +13,5 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+
+*.iml

go.mod

@@ -0,0 +1,10 @@
+module github.com/scouter-contrib/scouter-agent-golang
+
+go 1.15
+
+require (
+	github.com/emirpasic/gods v1.12.0
+	github.com/scouter-project/scouter-go-lib v0.0.0
+)
+
+replace github.com/scouter-project/scouter-go-lib => /Users/gunlee/Documents/workspace/gunlee01/scouter-go-lib

go.sum

@@ -0,0 +1,8 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/emirpasic/gods v1.12.0 h1:QAUIPSaCu4G+POclxeqb3F+WPpdKqFGlw36+yOzGlrg=
+github.com/emirpasic/gods v1.12.0/go.mod h1:YfzfFFoVP/catgzJb4IKIqXjX78Ha8FMSDh3ymbK86o=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

scouterx/agent.go

@@ -0,0 +1,19 @@
+package scouterx
+
+import (
+	"sync"
+)
+
+var agentOnce sync.Once
+
+func Init() {
+	agentOnce.Do(func() {
+		//confDir := "./nil"
+		//dir, err := os.Getwd()
+		//if err == nil {
+		//	confDir = dir
+		//}
+		//UdpSender = udpsender.GetInstance()
+	})
+}
+

scouterx/common/structure/cachemap/cachemap.go

@@ -0,0 +1,96 @@
+package cachemap
+
+import (
+	"github.com/emirpasic/gods/lists/singlylinkedlist"
+	"strconv"
+	"sync"
+)
+
+var lock sync.RWMutex
+
+type CacheMap struct {
+	table    map[interface{}]interface{}
+	ordering *singlylinkedlist.List
+	orderPos map[interface{}]int
+	maxSize  int
+}
+
+func New(maxSize int) *CacheMap {
+	m := &CacheMap{
+		table:    make(map[interface{}]interface{}),
+		ordering: singlylinkedlist.New(),
+		orderPos: make(map[interface{}]int),
+		maxSize:  maxSize,
+	}
+	return m
+}
+
+func (m *CacheMap) Add(key interface{}, item interface{}) {
+	lock.Lock()
+	defer lock.Unlock()
+	if _, contains := m.table[key]; !contains {
+		m.removeExceeded()
+		m.table[key] = item
+		m.ordering.Append(key)
+		m.orderPos[key] = m.ordering.Size() - 1
+	}
+}
+
+func (m *CacheMap) Remove(key interface{}) {
+	lock.Lock()
+	defer lock.Unlock()
+	delete(m.table, key)
+	index, contains := m.orderPos[key]
+	if contains {
+		m.ordering.Remove(index)
+		delete(m.orderPos, key)
+	}
+}
+
+func (m *CacheMap) removeExceeded() {
+	for m.ordering.Size() >= m.maxSize {
+		key, exist := m.ordering.Get(0)
+		if exist {
+			m.ordering.Remove(0)
+			delete(m.orderPos, key)
+			delete(m.table, key)
+		}
+	}
+}
+
+func (m *CacheMap) Contains(key interface{}) bool {
+	lock.RLock()
+	defer lock.RUnlock()
+	if _, contains := m.table[key]; !contains {
+		return false
+	}
+	return true
+}
+
+func (m *CacheMap) Get(key interface{}) interface{} {
+	lock.RLock()
+	defer lock.RUnlock()
+	return m.table[key]
+}
+
+func (m *CacheMap) Empty() bool {
+	lock.Lock()
+	defer lock.Unlock()
+	return m.Size() == 0
+}
+
+func (m *CacheMap) Size() int {
+	return m.ordering.Size()
+}
+
+func (m *CacheMap) Clear() {
+	lock.Lock()
+	defer lock.Unlock()
+	m.table = make(map[interface{}]interface{})
+	m.ordering.Clear()
+	m.orderPos = make(map[interface{}]int)
+}
+
+func (m *CacheMap) String() string {
+	return "CacheMap[" + strconv.Itoa(m.ordering.Size()) + "]"
+}

scouterx/common/structure/cachemap/cachemap_test.go

@@ -0,0 +1,63 @@
+package cachemap
+
+import (
+	"strconv"
+	"testing"
+)
+
+func TestCacheSet(t *testing.T) {
+	cacheMap := New(10)
+
+	//add "1" to "9"
+	for i := 0; i < 10; i++ {
+		cacheMap.Add(i, strconv.Itoa(i))
+	}
+
+	if cacheMap.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	//max size preserved
+	cacheMap.Add(3, "3")
+	if cacheMap.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	//0 exist
+	if (!cacheMap.Contains(0)) {
+		t.Error("contains error.")
+	}
+
+	//0 removed
+	cacheMap.Add(1000, "1000")
+	if cacheMap.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	if cacheMap.Contains(0) {
+		t.Error("contains error: 0 was removed.")
+	}
+
+	//1 exist
+	if !cacheMap.Contains(1) {
+		t.Error("contains error: 1 should be exist.")
+	}
+
+	//1 removed
+	cacheMap.Add(1001, "1001")
+	if cacheMap.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	if cacheMap.Contains(1) {
+		t.Error("contains error: 1 was removed.")
+	}
+
+	if !cacheMap.Contains(1001) {
+		t.Error("contains error: 1001 should be exist.")
+	}
+
+	if cacheMap.Get(1001) != "1001" {
+		t.Error("no matched value.")
+	}
+}

scouterx/common/structure/cacheset/cacheset.go

@@ -0,0 +1,78 @@
+package cacheset
+
+import (
+	"github.com/emirpasic/gods/lists/singlylinkedlist"
+	"strconv"
+	"sync"
+)
+
+var lock sync.RWMutex
+
+type CacheSet struct {
+	table    map[interface{}]struct{}
+	ordering *singlylinkedlist.List
+	maxSize  int
+}
+
+var itemExists = struct{}{}
+
+func New(maxSize int) *CacheSet {
+	set := &CacheSet{
+		table:    make(map[interface{}]struct{}),
+		ordering: singlylinkedlist.New(),
+		maxSize:  maxSize,
+	}
+	return set
+}
+
+func (set *CacheSet) Add(item interface{}) {
+	lock.Lock()
+	defer lock.Unlock()
+	var contains bool
+	if _, contains = set.table[item]; !contains {
+		set.removeExceeded()
+		set.table[item] = itemExists
+		set.ordering.Append(item)
+	}
+}
+
+func (set *CacheSet) removeExceeded() {
+	for set.ordering.Size() >= set.maxSize {
+		item, exist := set.ordering.Get(0)
+		if exist {
+			set.ordering.Remove(0)
+			delete(set.table, item)
+		}
+	}
+}
+
+func (set *CacheSet) Contains(item interface{}) bool {
+	lock.RLock()
+	defer lock.RUnlock()
+	if _, contains := set.table[item]; !contains {
+		return false
+	}
+	return true
+}
+
+func (set *CacheSet) Empty() bool {
+	lock.Lock()
+	defer lock.Unlock()
+	return set.Size() == 0
+}
+
+func (set *CacheSet) Size() int {
+	return set.ordering.Size()
+}
+
+func (set *CacheSet) Clear() {
+	lock.Lock()
+	defer lock.Unlock()
+	set.table = make(map[interface{}]struct{})
+	set.ordering.Clear()
+}
+
+func (set *CacheSet) String() string {
+	return "CacheSet[" + strconv.Itoa(set.ordering.Size()) + "]"
+}
+

scouterx/common/structure/cacheset/cacheset_test.go

@@ -0,0 +1,59 @@
+package cacheset
+
+import (
+	"strconv"
+	"testing"
+)
+
+func TestCacheSet(t *testing.T) {
+	cacheSet := New(10)
+
+	//add "1" to "9"
+	for i := 0; i < 10; i++ {
+		cacheSet.Add(strconv.Itoa(i))
+	}
+
+	if cacheSet.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	//max size preserved
+	cacheSet.Add("3")
+	if cacheSet.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	//0 exist
+	if (!cacheSet.Contains("0")) {
+		t.Error("contains error.")
+	}
+
+	//0 removed
+	cacheSet.Add("1000")
+	if cacheSet.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	if cacheSet.Contains("0") {
+		t.Error("contains error: 0 was removed.")
+	}
+
+	//1 exist
+	if !cacheSet.Contains("1") {
+		t.Error("contains error: 1 should be exist.")
+	}
+
+	//1 removed
+	cacheSet.Add("1001")
+	if cacheSet.Size() != 10 {
+		t.Error("size error.")
+	}
+
+	if cacheSet.Contains("1") {
+		t.Error("contains error: 1 was removed.")
+	}
+
+	if !cacheSet.Contains("1001") {
+		t.Error("contains error: 1001 should be exist.")
+	}
+}