gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

sql

Browse Source
This commit is contained in:
桜華
2023-04-28 07:28:40 +08:00
parent 8b14c7b7ec
commit f264d6921d
3 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
configs/sqlite3.go
View File

@@ -62,7 +62,7 @@ func init() {
name TEXT, name TEXT,
password TEXT, password TEXT,
created_at TEXT, created_at TEXT,
updated_at TEXT, updated_at TEXT
); );
CREATE TABLE IF NOT EXISTS tags( CREATE TABLE IF NOT EXISTS tags(
id INTEGER PRIMARY KEY AUTOINCREMENT, id INTEGER PRIMARY KEY AUTOINCREMENT,

1
main.go
View File

@@ -52,7 +52,6 @@ func models_get(w http.ResponseWriter, r *http.Request) {
listview.Page = ParamInt(r.URL.Query().Get("page"), 1) listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10) listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
listview.List = models.QueryModels(listview.Page, listview.PageSize) listview.List = models.QueryModels(listview.Page, listview.PageSize)
fmt.Println(listview.List)
listview.Total = models.CountModels() listview.Total = models.CountModels()
listview.Next = listview.Page*listview.PageSize < listview.Total listview.Next = listview.Page*listview.PageSize < listview.Total
w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Header().Set("Content-Type", "application/json; charset=utf-8")

8
models/Model.go
View File

@@ -27,13 +27,13 @@ func (model *Model) Create() error {
return err return err
} }
defer db.Close() defer db.Close()
stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?)") stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return err return err
} }
defer stmt.Close() defer stmt.Close()
result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Tags, model.CreatedAt, model.UpdatedAt) result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Tags, model.CreatedAt, model.UpdatedAt, model.UserID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return err return err
@@ -111,7 +111,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) {
return return
} }
defer db.Close() defer db.Close()
rows, err := db.Query("SELECT * FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize) rows, err := db.Query("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return return
@@ -119,7 +119,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) {
defer rows.Close() defer rows.Close()
for rows.Next() { for rows.Next() {
var model Model var model Model
err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.CreatedAt, &model.UpdatedAt) err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return return