v2.2.0

Author: liupan1890

source-6panserver/src/data/DBHelper.go

@@ -0,0 +1,71 @@
+package data
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+
+	"../utils"
+	buntdb "github.com/tidwall/buntdb"
+)
+
+var WWWDB *buntdb.DB = nil
+
+var APPDB *buntdb.DB = nil
+
+func Open() bool {
+	exepath := utils.ExePath()
+	dbpath := filepath.Join(exepath, "AppData")
+	log.Println("Start Path", dbpath)
+	errmk := os.MkdirAll(dbpath, 0777)
+	if errmk != nil {
+		log.Println(errmk)
+		return false
+	}
+	if WWWDB != nil {
+		return true
+	}
+	var err error
+	WWWDB, err = buntdb.Open(filepath.Join(dbpath, "www.db"))
+	if WWWDB == nil || err != nil {
+		return false
+	}
+
+	if APPDB != nil {
+		return true
+	}
+	APPDB, err = buntdb.Open(filepath.Join(dbpath, "user.db"))
+	if APPDB != nil && err == nil {
+		initData()
+	}
+	return APPDB != nil && WWWDB != nil
+}
+
+func Close() {
+	if APPDB != nil {
+		APPDB.Close()
+		APPDB = nil
+	}
+	if WWWDB != nil {
+		WWWDB.Close()
+		WWWDB = nil
+	}
+}
+
+func initData() {
+
+	var config buntdb.Config
+	config.SyncPolicy = buntdb.Always
+	APPDB.SetConfig(config)
+
+	APPDB.Update(func(tx *buntdb.Tx) error {
+		tx.CreateIndex("SixUser", "User:SixUser:*", buntdb.IndexString)
+		tx.CreateIndex("Downing", "Down:Downing:*", buntdb.IndexJSON("DownTime"))
+		tx.CreateIndex("Downed", "Down:Downed:*", buntdb.IndexJSON("DownTime"))
+		tx.CreateIndex("Uploading", "Down:Uploading:*", buntdb.IndexJSON("DownTime"))
+		tx.CreateIndex("Upload", "Down:Upload:*", buntdb.IndexJSON("DownTime"))
+		return nil
+	})
+
+	LoadSetting()
+}

source-6panserver/src/data/Down.go

@@ -0,0 +1,74 @@
+package data
+
+import (
+	buntdb "github.com/tidwall/buntdb"
+)
+
+func _ValToStr6(val string) (bool, string) {
+	if val == "" {
+		return false, ""
+	}
+	v := []byte(val)
+	l := len(v)
+	for i := 0; i < l; i++ {
+		v[i] ^= 6
+	}
+	return true, string(v)
+}
+
+func GetDownedAll() (bool, map[string]string) {
+
+	isget := false
+	list := make(map[string]string)
+	str := ""
+	APPDB.View(func(tx *buntdb.Tx) error {
+		tx.Ascend("Downed", func(key, val string) bool {
+
+			isget, str = _ValToStr6(val)
+			if isget {
+				list[key] = str
+			}
+			return true
+		})
+		return nil
+	})
+	return isget, list
+}
+
+func GetDowningAll() (bool, map[string]string) {
+
+	isget := false
+	list := make(map[string]string)
+	str := ""
+	APPDB.View(func(tx *buntdb.Tx) error {
+		tx.Ascend("Downing", func(key, val string) bool {
+			isget, str = _ValToStr6(val)
+			if isget {
+				list[key] = str
+			}
+			return true
+		})
+		return nil
+	})
+	return isget, list
+}
+
+func SetDown(downid, jsonval string) {
+	APPDB.Update(func(tx *buntdb.Tx) error {
+		v := []byte(jsonval)
+		l := len(v)
+		for i := 0; i < l; i++ {
+			v[i] ^= 6
+		}
+
+		tx.Set(downid, string(v), nil)
+		return nil
+	})
+}
+
+func DeleteDown(downid string) {
+	APPDB.Update(func(tx *buntdb.Tx) error {
+		tx.Delete(downid)
+		return nil
+	})
+}

source-6panserver/src/data/Range.go

@@ -0,0 +1,42 @@
+package data
+
+import (
+	"strings"
+
+	buntdb "github.com/tidwall/buntdb"
+)
+
+func GetRange(key string) (isget bool, jsonstr string) {
+	key = strings.Replace(key, "Down:", "", -1)
+	key = strings.Replace(key, "Downing:", "", -1)
+	isget = false
+	jsonstr = ""
+
+	APPDB.View(func(tx *buntdb.Tx) error {
+		jsonstr, _ = tx.Get("Range:" + key)
+		if jsonstr != "" {
+			isget = true
+		}
+		return nil
+	})
+	return isget, jsonstr
+}
+
+func SetRange(key string, jsonval string) {
+	key = strings.Replace(key, "Down:", "", -1)
+	key = strings.Replace(key, "Downing:", "", -1)
+
+	APPDB.Update(func(tx *buntdb.Tx) error {
+		tx.Set("Range:"+key, jsonval, nil)
+		return nil
+	})
+}
+
+func DelRange(key string) {
+	key = strings.Replace(key, "Down:", "", -1)
+	key = strings.Replace(key, "Downing:", "", -1)
+	APPDB.Update(func(tx *buntdb.Tx) error {
+		tx.Delete("Range:" + key)
+		return nil
+	})
+}