diff --git a/configs/sqlite3.go b/configs/sqlite3.go index c9e2bc0..a8363f7 100644 --- a/configs/sqlite3.go +++ b/configs/sqlite3.go @@ -62,7 +62,7 @@ func init() { name TEXT, password TEXT, created_at TEXT, - updated_at TEXT, + updated_at TEXT ); CREATE TABLE IF NOT EXISTS tags( id INTEGER PRIMARY KEY AUTOINCREMENT, diff --git a/main.go b/main.go index d857917..a198a18 100644 --- a/main.go +++ b/main.go @@ -52,7 +52,6 @@ func models_get(w http.ResponseWriter, r *http.Request) { listview.Page = ParamInt(r.URL.Query().Get("page"), 1) listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10) listview.List = models.QueryModels(listview.Page, listview.PageSize) - fmt.Println(listview.List) listview.Total = models.CountModels() listview.Next = listview.Page*listview.PageSize < listview.Total w.Header().Set("Content-Type", "application/json; charset=utf-8") diff --git a/models/Model.go b/models/Model.go index 14cc933..854ad1e 100644 --- a/models/Model.go +++ b/models/Model.go @@ -27,13 +27,13 @@ func (model *Model) Create() error { return err } 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 { log.Println(err) return err } 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 { log.Println(err) return err @@ -111,7 +111,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) { return } 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 { log.Println(err) return @@ -119,7 +119,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) { defer rows.Close() for rows.Next() { 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 { log.Println(err) return