rename connectors

connector/mock.go

@@ -0,0 +1,126 @@
+// +build !release
+
+package connector
+
+import (
+	"math/rand"
+	"sort"
+	"strings"
+	"time"
+
+	"github.com/bcicen/ctop/container"
+	"github.com/bcicen/ctop/metrics"
+	"github.com/jgautheron/codename-generator"
+	"github.com/nu7hatch/gouuid"
+)
+
+type Mock struct {
+	containers container.Containers
+}
+
+func NewMock() *Mock {
+	cs := &Mock{}
+	go cs.Init()
+	go cs.Loop()
+	return cs
+}
+
+// Create Mock containers
+func (cs *Mock) Init() {
+	rand.Seed(int64(time.Now().Nanosecond()))
+
+	for i := 0; i < 4; i++ {
+		cs.makeContainer(3)
+	}
+
+	for i := 0; i < 16; i++ {
+		cs.makeContainer(1)
+	}
+
+}
+
+func (cs *Mock) makeContainer(aggression int64) {
+	collector := metrics.NewMock(aggression)
+	c := container.New(makeID(), collector)
+	c.SetMeta("name", makeName())
+	c.SetState(makeState())
+	cs.containers = append(cs.containers, c)
+}
+
+func (cs *Mock) Loop() {
+	iter := 0
+	for {
+		// Change state for random container
+		if iter%5 == 0 && len(cs.containers) > 0 {
+			randC := cs.containers[rand.Intn(len(cs.containers))]
+			randC.SetState(makeState())
+		}
+		iter++
+		time.Sleep(3 * time.Second)
+	}
+}
+
+// Get a single container, by ID
+func (cs *Mock) Get(id string) (*container.Container, bool) {
+	for _, c := range cs.containers {
+		if c.Id == id {
+			return c, true
+		}
+	}
+	return nil, false
+}
+
+// Return array of all containers, sorted by field
+func (cs *Mock) All() container.Containers {
+	sort.Sort(cs.containers)
+	cs.containers.Filter()
+	return cs.containers
+}
+
+// Remove containers by ID
+func (cs *Mock) delByID(id string) {
+	for n, c := range cs.containers {
+		if c.Id == id {
+			cs.del(n)
+			return
+		}
+	}
+}
+
+// Remove one or more containers by index
+func (cs *Mock) del(idx ...int) {
+	for _, i := range idx {
+		cs.containers = append(cs.containers[:i], cs.containers[i+1:]...)
+	}
+	log.Infof("removed %d dead containers", len(idx))
+}
+
+func makeID() string {
+	u, err := uuid.NewV4()
+	if err != nil {
+		panic(err)
+	}
+	return strings.Replace(u.String(), "-", "", -1)[:12]
+}
+
+func makeName() string {
+	n, err := codename.Get(codename.Sanitized)
+	nsp := strings.Split(n, "-")
+	if len(nsp) > 2 {
+		n = strings.Join(nsp[:2], "-")
+	}
+	if err != nil {
+		panic(err)
+	}
+	return strings.Replace(n, "-", "_", -1)
+}
+
+func makeState() string {
+	switch rand.Intn(10) {
+	case 0, 1, 2:
+		return "exited"
+	case 3:
+		return "paused"
+	}
+	return "running"
+}