change database scheema

2019-07-02 09:02:31 +00:00
parent dd09bfda6e
commit 462804ed07
5 changed files with 150 additions and 78 deletions
@@ -0,0 +1,128 @@
+package public
+
+import (
+	"dorisbot/models"
+	"errors"
+	"time"
+)
+
+// CommandsModel -
+type CommandModel struct {
+	Key           string    `db:"key"`
+	Platform      string    `db:"platform"`
+	Binding       string    `db:"binding"`
+	RequireManage bool      `db:"require_manage"`
+	RequireInit   bool      `db:"require_init"`
+	Value         string    `db:"value"`
+	Ctime         time.Time `db:"ctime"`
+	Mtime         time.Time `db:"mtime"`
+}
+
+// NewCommand -
+func NewCommand(key, value, platform, binding string, requireManage, requireInit bool) (c *CommandModel, err error) {
+	if len(key) == 0 || len(value) == 0 {
+		return nil, errors.New("cmd, server or value is empty")
+	}
+
+	c = &CommandModel{
+		Key:           key,
+		Value:         value,
+		Platform:      platform,
+		Binding:       binding,
+		RequireInit:   requireInit,
+		RequireManage: requireManage,
+	}
+
+	return
+}
+
+// GetCommandListByPlatformBinding -
+func GetCommandListByPlatformBinding(binding, platform string) (list []*CommandModel, err error) {
+	if len(binding) == 0 || len(platform) == 0 {
+		return nil, errors.New("server id or platform is empty")
+	}
+	db := models.GetConn()
+	query := `select "key", "platform", "binding", "value", "require_manage", "require_init", "ctime", "mtime"
+			from "command"
+			where 
+				"binding" = $1
+				and "platform" = $2`
+	err = db.Select(&list, query, binding, platform)
+	if err != nil {
+		return nil, err
+	}
+	return
+}
+
+// GetCommandByPlatformBinding -
+func GetCommandByPlatformBinding(key, binding, platform string) (list []*CommandModel, err error) {
+	if len(key) == 0 || len(binding) == 0 || len(platform) == 0 {
+		return nil, errors.New("key, server id or platform is empty")
+	}
+	db := models.GetConn()
+	query := `select "key", "platform", "binding", "value", "require_manage", "require_init", "ctime", "mtime"
+			from "command"
+			where 
+				"key" = $1
+				and (
+				("binding" = $2 and "platform" = $3)
+				or ("binding" = '' and "platform" = '')
+				)
+				order by "binding" desc`
+	err = db.Select(&list, query, key, binding, platform)
+	if err != nil {
+		return nil, err
+	}
+	return
+}
+
+// GetCommandByKey -
+func GetCommandByKey(key string) (list []*CommandModel, err error) {
+	if len(key) == 0 {
+		return nil, errors.New("key is empty")
+	}
+
+	db := models.GetConn()
+	query := `select "key", "platform", "binding", "value", "require_manage", "require_init", "ctime", "mtime"
+			from "command"
+			where 
+				"key" = $1`
+	err = db.Select(&list, query, key)
+	if err != nil {
+		return nil, err
+	}
+	return
+}
+
+// GetGlobalCommand -
+func GetGlobalCommand(key string) (list []*CommandModel, err error) {
+	if len(key) == 0 {
+		return nil, errors.New("key is empty")
+	}
+	db := models.GetConn()
+	query := `select "cmd", "server", "value", "require_manage", "require_init", "ctime", "mtime"
+			from "command"
+			where 
+				"cmd" = $1
+				and	"platform" = ''
+				and "binding" = ''`
+	err = db.Select(&list, query, key)
+	if err != nil {
+		return nil, err
+	}
+	return
+}
+
+// Write -
+func (p *CommandModel) Write() (err error) {
+	db := models.GetConn()
+	query := `insert into "command" ("key", "paltform", "binding", "value", "require_init", "require_manage") values 
+(:key, :platform, :binding, :value, :require_init, :require_manage) returning *`
+	row, err := db.NamedQuery(query, p)
+	if err != nil {
+		return err
+	}
+	err = row.StructScan(p)
+
+	return
+}