admin
a6de9ebd12
git-subtree-dir: server git-subtree-mainline:c4e617ada7git-subtree-split:e64421295f
37 lines
1009 B
Go
37 lines
1009 B
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"slogan-agent/styleagent/consts"
|
|
"slogan-agent/styleagent/model/entity"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
var ScoringRule = &scoringRuleDao{}
|
|
|
|
type scoringRuleDao struct{}
|
|
|
|
func init() {
|
|
ctx := context.Background()
|
|
_, err := g.DB().Exec(ctx, `CREATE TABLE IF NOT EXISTS `+consts.TableNameScoringRule+` (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
dimension TEXT NOT NULL DEFAULT '',
|
|
rule_type TEXT NOT NULL DEFAULT '',
|
|
rules_json TEXT NOT NULL DEFAULT '',
|
|
enabled INTEGER NOT NULL DEFAULT 1,
|
|
version INTEGER NOT NULL DEFAULT 1,
|
|
created_at DATETIME DEFAULT (datetime('now','localtime'))
|
|
)`)
|
|
if err != nil {
|
|
g.Log().Warningf(ctx, "create scoring_rule table failed: %v", err)
|
|
}
|
|
}
|
|
|
|
func (d *scoringRuleDao) ListEnabled(ctx context.Context) ([]*entity.ScoringRule, error) {
|
|
var list []*entity.ScoringRule
|
|
err := g.DB().Model(consts.TableNameScoringRule).Ctx(ctx).
|
|
Where("enabled", 1).OrderAsc("id").Scan(&list)
|
|
return list, err
|
|
}
|