refactor: databse functions and data structure

TODO.md

@@ -1,7 +1,5 @@
 ### Todo
 - [ ] Get Sets Name
-- [ ] Drop Set
-- [ ] Clean Set
 - [ ] Get Elements of a set
 - [ ] Count elements of a set
 - [ ] Remove Element from a set
@@ -24,3 +22,5 @@
 
 
 ### ✓ Done
+- [X] Clean Set
+- [X] Drop Set

cmd/main.go

@@ -7,6 +7,6 @@ import (
 )
 
 func main() {
-	database := database.CreateDataBase(os.Args[0])
-	database.InitSocket()
+	database := database.Init(os.Args[0])
+	database.AddSet("test")
 }

core/database/database.go

@@ -1,29 +1,81 @@
 package database
 
 import (
-	"net"
-	"os"
-
 	"github.com/zurvan-lab/TimeTraceDB/config"
 )
 
 type Database struct {
 	Sets   Sets
-	Config config.Config
-	socket net.Listener
+	Config *config.Config
 }
 
-func CreateDataBase(path string) *Database {
+func Init(path string) *Database {
 	return &Database{
-		Sets:   *NewSets(),
-		Config: *config.LoadFromFile(path),
+		Sets:   make(Sets),
+		Config: config.LoadFromFile(path),
+	}
+}
+
+func (db *Database) AddSet(name string) string {
+	db.Sets[name] = make(Set)
+	return "DONE"
+}
+
+func (db *Database) AddSubSet(set, name string) string {
+	s, ok := db.Sets[set]
+	if !ok {
+		return "SETNF"
+	}
+	s[name] = make(SubSet, 0)
+	return "DONE"
+}
+
+func (db *Database) PushElement(set, subset string, e Element) string {
+	_, ok := db.Sets[set][subset]
+	if !ok {
+		return "SUBSETNF"
+	}
+	db.Sets[set][subset] = append(db.Sets[set][subset], e)
+	return "DONE"
+}
+
+func (db *Database) DropSet(name string) string {
+	_, ok := db.Sets[name]
+	if !ok {
+		return "SETNF"
+	}
+	delete(db.Sets, name)
+	return "DONE"
+}
+
+func (db *Database) DropSubSet(set, subset string) string {
+	_, ok := db.Sets[set][subset]
+	if !ok {
+		return "SUBETNF"
+	}
+	delete(db.Sets[set], subset)
+	return "DONE"
+}
+
+func (db *Database) CleanSets() string {
+	db.Sets = make(Sets)
+	return "DONE"
+}
+
+func (db *Database) CleanSet(name string) string {
+	_, ok := db.Sets[name]
+	if !ok {
+		return "SETNF"
 	}
+	db.Sets[name] = make(Set)
+	return "DONE"
 }
 
-func (db *Database) InitSocket() {
-	var err error
-	db.socket, err = net.Listen("tcp", db.Config.Listen.IP+":"+db.Config.Listen.Port)
-	if err != nil {
-		os.Exit(1)
+func (db *Database) CleanSubSet(set, subset string) string {
+	_, ok := db.Sets[set][subset]
+	if !ok {
+		return "SUBETNF"
 	}
+	db.Sets[set][subset] = make(SubSet, 0)
+	return "DONE"
 }

core/database/element.go

@@ -2,11 +2,6 @@ package database
 
 import "time"
 
-type Element struct {
-	value string
-	time  time.Time
-}
-
 func NewElement(v string) Element {
 	return Element{value: v, time: time.Now()}
 }

core/database/types.go

@@ -0,0 +1,14 @@
+package database
+
+import "time"
+
+type Element struct {
+	value string
+	time  time.Time
+}
+
+type (
+	Sets   map[string]Set
+	Set    map[string]SubSet
+	SubSet []Element
+)