Refactor everything

repo/appRepo.go

@@ -0,0 +1,42 @@
+package repo
+
+import (
+	"encoding/json"
+
+	"bloat/kv"
+	"bloat/model"
+)
+
+type appRepo struct {
+	db *kv.Database
+}
+
+func NewAppRepo(db *kv.Database) *appRepo {
+	return &appRepo{
+		db: db,
+	}
+}
+
+func (repo *appRepo) Add(a model.App) (err error) {
+	data, err := json.Marshal(a)
+	if err != nil {
+		return
+	}
+	err = repo.db.Set(a.InstanceDomain, data)
+	return
+}
+
+func (repo *appRepo) Get(instanceDomain string) (a model.App, err error) {
+	data, err := repo.db.Get(instanceDomain)
+	if err != nil {
+		err = model.ErrAppNotFound
+		return
+	}
+
+	err = json.Unmarshal(data, &a)
+	if err != nil {
+		return
+	}
+
+	return
+}

repo/sessionRepo.go

@@ -0,0 +1,42 @@
+package repo
+
+import (
+	"encoding/json"
+
+	"bloat/kv"
+	"bloat/model"
+)
+
+type sessionRepo struct {
+	db *kv.Database
+}
+
+func NewSessionRepo(db *kv.Database) *sessionRepo {
+	return &sessionRepo{
+		db: db,
+	}
+}
+
+func (repo *sessionRepo) Add(s model.Session) (err error) {
+	data, err := json.Marshal(s)
+	if err != nil {
+		return
+	}
+	err = repo.db.Set(s.ID, data)
+	return
+}
+
+func (repo *sessionRepo) Get(id string) (s model.Session, err error) {
+	data, err := repo.db.Get(id)
+	if err != nil {
+		err = model.ErrSessionNotFound
+		return
+	}
+
+	err = json.Unmarshal(data, &s)
+	if err != nil {
+		return
+	}
+
+	return
+}