diff --git a/README.md b/README.md
index 6c5fa77..b9fa4aa 100644
--- a/README.md
+++ b/README.md
@@ -44,6 +44,21 @@ WS | /api/{name}/{item_id} | Websocket 連接對象
對象模型:
+```go
+type Model struct {
+ ID int `json:"id"`
+ Name string `json:"name"`
+ Type string `json:"type"` // (lora|ckp|hyper|ti)
+ TriggerWords string `json:"trigger_words"` // 觸發詞
+ BaseModel string `json:"base_model"` // (SD1.5|SD2)
+ ModelPath string `json:"model_path"` // 模型路徑
+ Status string `json:"status"` // (waiting|running|success|error)
+ Progress int `json:"progress"` // (0-100)
+ Tags []string `json:"tags"`
+ CreatedAt string `json:"created_at"`
+ UpdatedAt string `json:"updated_at"`
+}
+```
```javascript
//// TODO
////训练专属模型模块
@@ -94,17 +109,6 @@ WS | /api/{name}/{item_id} | Websocket 連接對象
}
}
-// Model 模型對象
-{
- id: 'xxxxx', // 模型ID
- name: 'xxx', // 模型名稱
- type: '', // 模型類型(RoLa|SD2|SD1.5)
- status: 'xxx', // 模型狀態(waiting|running|success|error)
- progress: 100, // 訓練進度(0~100)
- createdAt: '', // 創建時間
- updatedAt: '', // 更新時間
-}
-
// Task 任務對象
{
id: 'xxxxx', // 任務ID
diff --git a/configs/sqlite3.go b/configs/sqlite3.go
index 8d7eccf..a3eee7d 100644
--- a/configs/sqlite3.go
+++ b/configs/sqlite3.go
@@ -36,6 +36,9 @@ func init() {
CREATE TABLE IF NOT EXISTS models(
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT,
+ type TEXT,
+ status TEXT,
+ progress INTEGER,
created_at TEXT,
updated_at TEXT
);
diff --git a/main.go b/main.go
index 7b3c0c5..d857917 100644
--- a/main.go
+++ b/main.go
@@ -8,6 +8,7 @@ import (
"net/http"
"runtime"
"strconv"
+ "text/template"
"time"
"main/models"
@@ -15,130 +16,244 @@ import (
"github.com/gorilla/mux"
)
-func CreateObject(w http.ResponseWriter, r *http.Request) {
- defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
- // 從body中獲取收到的JSON數據
- body, err := ioutil.ReadAll(r.Body)
- if err != nil {
- log.Println(err)
- return
- }
- defer r.Body.Close()
-
- // 將JSON數據轉換成目標對象
- object := models.GetObject(mux.Vars(r)["name"], 0)
- if err = json.Unmarshal(body, object); err != nil {
- log.Println(err)
- return
- }
-
- // 將對象存入數據庫
- err = object.Create()
- if err != nil {
- log.Println(err)
- return
- }
-
- // 返回結果
- w.Header().Set("Content-Type", "application/json; charset=utf-8")
- w.Write(object.ToJSON())
-}
-
-func GetObjetc(w http.ResponseWriter, r *http.Request) {
- defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
- // 獲取對象
- object := models.GetObject(mux.Vars(r)["name"], ParamInt(mux.Vars(r)["id"], 0))
- err := object.Get()
- if err != nil {
- log.Println(err)
- return
- }
-
- // 返回結果
- w.Header().Set("Content-Type", "application/json; charset=utf-8")
- w.Write(object.ToJSON())
-}
-
-func UpdateObject(w http.ResponseWriter, r *http.Request) {
- defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
- // 從body中獲取收到的JSON數據
- body, err := ioutil.ReadAll(r.Body)
- if err != nil {
- log.Println(err)
- return
- }
- defer r.Body.Close()
-
- // 將JSON數據轉換成目標對象
- object := models.GetObject(mux.Vars(r)["name"], ParamInt(mux.Vars(r)["id"], 0))
- if err = json.Unmarshal(body, object); err != nil {
- log.Println(err)
- return
- }
-
- // 更新對象
- err = object.Update()
- if err != nil {
- log.Println(err)
- return
- }
-
- // 返回結果
- w.Header().Set("Content-Type", "application/json; charset=utf-8")
- w.Write(object.ToJSON())
-}
-
-func DeleteObject(w http.ResponseWriter, r *http.Request) {
- defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
- // 刪除對象
- object := models.GetObject(mux.Vars(r)["name"], ParamInt(mux.Vars(r)["id"], 0))
- err := object.Delete()
- if err != nil {
- log.Println(err)
- return
- }
-
- // 返回結果
- w.Header().Set("Content-Type", "application/json; charset=utf-8")
- w.Write(object.ToJSON())
-}
-
-func GetObjects(w http.ResponseWriter, r *http.Request) {
- defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
- var listview models.ListView
-
- // 獲取查詢參數: page, pageSize
- name := mux.Vars(r)["name"]
- page := ParamInt(r.URL.Query().Get("page"), 1)
- pagesize := ParamInt(r.URL.Query().Get("pageSize"), 10)
-
- // 獲取對象列表
- listview.Read(name, page, pagesize)
-
- // 返回結果
- w.Header().Set("Content-Type", "application/json; charset=utf-8")
- w.Write(listview.ToJSON())
-}
-
func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
r := mux.NewRouter()
- r.HandleFunc("/api/{name}", GetObjects).Methods("GET")
- r.HandleFunc("/api/{name}", CreateObject).Methods("POST")
- r.HandleFunc("/api/{name}/{id}", GetObjetc).Methods("GET")
- r.HandleFunc("/api/{name}/{id}", UpdateObject).Methods("PATCH")
- r.HandleFunc("/api/{name}/{id}", DeleteObject).Methods("DELETE")
+ r.Use(middleware)
+ r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+ t, _ := template.ParseFiles("templates/index.html")
+ t.Execute(w, nil)
+ })
+ r.HandleFunc("/api/models", models_get).Methods("GET")
+ r.HandleFunc("/api/models", models_post).Methods("POST")
+ r.HandleFunc("/api/models/{id}", models_item_get).Methods("GET")
+ r.HandleFunc("/api/models/{id}", models_item_patch).Methods("PATCH")
+ r.HandleFunc("/api/models/{id}", models_item_delete).Methods("DELETE")
+
+ r.HandleFunc("/api/images", images_get).Methods("GET")
+ r.HandleFunc("/api/images", images_post).Methods("POST")
+ r.HandleFunc("/api/images/{id}", images_item_get).Methods("GET")
+ r.HandleFunc("/api/images/{id}", images_item_patch).Methods("PATCH")
+ r.HandleFunc("/api/images/{id}", images_item_delete).Methods("DELETE")
+
+ r.HandleFunc("/api/tasks", tasks_get).Methods("GET")
+ r.HandleFunc("/api/tasks", tasks_post).Methods("POST")
+ r.HandleFunc("/api/tasks/{id}", tasks_item_get).Methods("GET")
+ r.HandleFunc("/api/tasks/{id}", tasks_item_patch).Methods("PATCH")
+ r.HandleFunc("/api/tasks/{id}", tasks_item_delete).Methods("DELETE")
log.Println("Web Server is running on http://localhost:8080")
http.ListenAndServe(":8080", r)
}
+func models_get(w http.ResponseWriter, r *http.Request) {
+ var listview models.ListView
+ 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")
+ w.Write(listview.ToJSON())
+}
+
+func models_post(w http.ResponseWriter, r *http.Request) {
+ var model models.Model
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ if err = json.Unmarshal(body, &model); err != nil {
+ log.Println(err)
+ return
+ }
+ model.Create()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(model))
+}
+
+func models_item_get(w http.ResponseWriter, r *http.Request) {
+ model := models.Model{ID: ParamInt(mux.Vars(r)["id"], 0)}
+ model.Get()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(model))
+}
+
+func models_item_patch(w http.ResponseWriter, r *http.Request) {
+ model := models.Model{}
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ if err = json.Unmarshal(body, &model); err != nil {
+ log.Println(err)
+ return
+ }
+ model.ID = ParamInt(mux.Vars(r)["id"], 0)
+ model.Update()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(model))
+}
+
+func models_item_delete(w http.ResponseWriter, r *http.Request) {
+ model := models.Model{ID: ParamInt(mux.Vars(r)["id"], 0)}
+ model.Delete()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(model))
+}
+
+func images_get(w http.ResponseWriter, r *http.Request) {
+ var listview models.ListView
+ listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+ listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+ listview.List = models.QueryImages(listview.Page, listview.PageSize)
+ listview.Total = models.CountImages()
+ listview.Next = listview.Page*listview.PageSize < listview.Total
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(listview))
+}
+
+func images_post(w http.ResponseWriter, r *http.Request) {
+ var image models.Image
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ if err = json.Unmarshal(body, &image); err != nil {
+ log.Println(err)
+ return
+ }
+ image.Create()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(image))
+}
+
+func images_item_get(w http.ResponseWriter, r *http.Request) {
+ image := models.Image{ID: ParamInt(mux.Vars(r)["id"], 0)}
+ image.Get()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(image))
+}
+
+func images_item_patch(w http.ResponseWriter, r *http.Request) {
+ image := models.Image{}
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ if err = json.Unmarshal(body, &image); err != nil {
+ log.Println(err)
+ return
+ }
+ image.ID = ParamInt(mux.Vars(r)["id"], 0)
+ image.Update()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(image))
+}
+
+func images_item_delete(w http.ResponseWriter, r *http.Request) {
+ image := models.Image{ID: ParamInt(mux.Vars(r)["id"], 0)}
+ image.Delete()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(image))
+}
+
+func tasks_get(w http.ResponseWriter, r *http.Request) {
+ var listview models.ListView
+ listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+ listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+ listview.List = models.QueryTasks(listview.Page, listview.PageSize)
+ listview.Total = models.CountTasks()
+ listview.Next = listview.Page*listview.PageSize < listview.Total
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(listview))
+}
+
+func tasks_post(w http.ResponseWriter, r *http.Request) {
+ var task models.Task
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ if err = json.Unmarshal(body, &task); err != nil {
+ log.Println(err)
+ return
+ }
+ task.Create()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(task))
+}
+
+func tasks_item_get(w http.ResponseWriter, r *http.Request) {
+ task := models.Task{ID: ParamInt(mux.Vars(r)["id"], 0)}
+ task.Get()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(task))
+}
+
+func tasks_item_patch(w http.ResponseWriter, r *http.Request) {
+ task := models.Task{}
+ body, err := ioutil.ReadAll(r.Body)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer r.Body.Close()
+ if err = json.Unmarshal(body, &task); err != nil {
+ log.Println(err)
+ return
+ }
+ task.ID = ParamInt(mux.Vars(r)["id"], 0)
+ task.Update()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(task))
+}
+
+func tasks_item_delete(w http.ResponseWriter, r *http.Request) {
+ task := models.Task{ID: ParamInt(mux.Vars(r)["id"], 0)}
+ task.Delete()
+ w.Header().Set("Content-Type", "application/json; charset=utf-8")
+ w.Write(ToJSON(task))
+}
+
+// 中間件, 通用
+func middleware(next http.Handler) http.Handler {
+ return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+ defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
+ // 處理跨域請求
+ w.Header().Set("Access-Control-Allow-Origin", "*")
+ w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization, X-Requested-With")
+ w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, PATCH, DELETE, OPTIONS")
+ // 處理OPTIONS請求
+ if r.Method == "OPTIONS" {
+ w.WriteHeader(http.StatusOK)
+ return
+ }
+ next.ServeHTTP(w, r)
+ })
+}
+
+func ToJSON(object interface{}) []byte {
+ json, err := json.MarshalIndent(object, "", " ")
+ if err != nil {
+ log.Println(err)
+ return []byte{}
+ }
+ return json
+}
+
func LogComponent(startTime int64, r *http.Request) {
ms := (time.Now().UnixNano() - startTime) / 1000000
color := "\033[1;32m%d\033[0m"
@@ -162,11 +277,11 @@ func LogComponent(startTime int64, r *http.Request) {
}
// 獲取查詢參數(int 類型)
-func ParamInt(value string, defaultValue int64) int64 {
+func ParamInt(value string, defaultValue int) int {
if value == "" {
return defaultValue
}
- result, err := strconv.ParseInt(value, 10, 64)
+ result, err := strconv.Atoi(value)
if err != nil {
return defaultValue
}
diff --git a/models/Image.go b/models/Image.go
new file mode 100644
index 0000000..4dec198
--- /dev/null
+++ b/models/Image.go
@@ -0,0 +1,136 @@
+package models
+
+import (
+ "log"
+ "main/configs"
+)
+
+type Image struct {
+ ID int `json:"id"`
+ Name string `json:"name"`
+ CreatedAt string `json:"created_at"`
+ UpdatedAt string `json:"updated_at"`
+}
+
+func (image *Image) Create() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("INSERT INTO images(name, created_at, updated_at) values(?, ?, ?)")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ result, err := stmt.Exec(image.Name, image.CreatedAt, image.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ id, err := result.LastInsertId()
+ if err != nil {
+ return err
+ }
+ image.ID = int(id)
+ return nil
+}
+
+func (image *Image) Delete() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("DELETE FROM images WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(image.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (image *Image) Update() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("UPDATE images SET name = ?, updated_at = ? WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(image.Name, image.UpdatedAt, image.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (image *Image) Get() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT id, name, created_at, updated_at FROM images WHERE id = ?", image.ID).Scan(&image.ID, &image.Name, &image.CreatedAt, &image.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func QueryImages(page int, pagesize int) (images []interface{}) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer db.Close()
+ rows, err := db.Query("SELECT id, name, created_at, updated_at FROM images LIMIT ?, ?", (page-1)*pagesize, pagesize)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer rows.Close()
+ for rows.Next() {
+ image := Image{}
+ err := rows.Scan(&image.ID, &image.Name, &image.CreatedAt, &image.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ images = append(images, image)
+ }
+ return
+}
+
+func CountImages() (count int) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT COUNT(*) FROM images").Scan(&count)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ return
+}
diff --git a/models/ListView.go b/models/ListView.go
index 5b2d446..072d794 100644
--- a/models/ListView.go
+++ b/models/ListView.go
@@ -3,14 +3,13 @@ package models
import (
"encoding/json"
"log"
- "main/configs"
)
type ListView struct {
- Code int64 `json:"code"`
- Page int64 `json:"page"`
- PageSize int64 `json:"page_size"`
- Total int64 `json:"total"`
+ Code int `json:"code"`
+ Page int `json:"page"`
+ PageSize int `json:"page_size"`
+ Total int `json:"total"`
Next bool `json:"next"`
List []interface{} `json:"list"`
}
@@ -32,292 +31,72 @@ func (listview *ListView) ToJSON() []byte {
return b
}
-// 查詢數據(分頁, 每頁數量, 排序, 條件, 關聯, 獲取總數)
-func (listview *ListView) Read(name string, page int64, page_size int64) error {
-
- // 獲取數據庫連接
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return err
- }
-
- // 獲取總數
- var total int64
- err = db.QueryRow("SELECT COUNT(*) FROM " + name).Scan(&total)
- if err != nil {
- log.Println(err)
- return err
- }
-
- // 獲取數據
- rows, err := db.Query("SELECT * FROM "+name+" LIMIT ?, ?", (page-1)*page_size, page_size)
- if err != nil {
- log.Println(err)
- return err
- }
- defer rows.Close()
-
- // 將數據轉換成對象
- switch name {
- case "images":
- for rows.Next() {
- var image Image
- err = rows.Scan(&image.ID, &image.Name, &image.CreatedAt, &image.UpdatedAt)
- if err != nil {
- log.Println(err)
- return err
- }
- listview.List = append(listview.List, image)
- }
- case "models":
- for rows.Next() {
- var model Model
- err = rows.Scan(&model.ID, &model.Name, &model.CreatedAt, &model.UpdatedAt)
- if err != nil {
- log.Println(err)
- return err
- }
- listview.List = append(listview.List, model)
- }
- default:
- for rows.Next() {
- var object Object
- err = rows.Scan(&object.ID, &object.Name, &object.CreatedAt, &object.UpdatedAt)
- if err != nil {
- log.Println(err)
- return err
- }
- listview.List = append(listview.List, object)
- }
- }
-
- // 設置返回值
- listview.Code = 0
- listview.Page = page
- listview.PageSize = page_size
- listview.Total = total
- listview.Next = total > page*page_size
-
- return nil
-}
-
-type Object struct {
- ID int64 `json:"id"`
- Name string `json:"name"`
- CreatedAt string `json:"created_at"`
- UpdatedAt string `json:"updated_at"`
- tablename string
-}
-
-type Image struct {
- Object
-}
-
-type Model struct {
- Object
- Type string `json:"type"`
-}
-
-type Tag struct {
- Object
-}
-
-type User struct {
- Object
-}
-
-type Task struct {
- Object
-}
-
-type ObjectInterface interface {
- Create() error
- Delete() error
- Update() error
- Get() error
- GetAll() ([]Object, error)
- GetPage(page int64, page_size int64) (ListView, error)
- ToJSON() []byte
-}
-
-// 按照名稱返回響應的對象類型
-func GetObject(name string, id int64) ObjectInterface {
- switch name {
- case "images":
- return &Image{
- Object{tablename: name, ID: id},
- }
- default:
- return &Object{tablename: name, ID: id}
- }
-}
-
-// 創建對象
-func (d *Object) Create() error {
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return err
- }
- stmt, err := db.Prepare("INSERT INTO " + d.tablename + "(name, created_at, updated_at) values(?, ?, ?)")
- if err != nil {
- log.Println(err)
- return err
- }
- defer stmt.Close()
- result, err := stmt.Exec(d.Name, d.CreatedAt, d.UpdatedAt)
- if err != nil {
- log.Println(err)
- return err
- }
- id, err := result.LastInsertId()
- if err != nil {
- return err
- }
- d.ID = id
- return nil
-}
-
-// 刪除對象
-func (d *Object) Delete() error {
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return err
- }
- stmt, err := db.Prepare("DELETE FROM " + d.tablename + " WHERE id = ?")
- if err != nil {
- log.Println(err)
- return err
- }
- defer stmt.Close()
- _, err = stmt.Exec(d.ID)
- if err != nil {
- log.Println(err)
- return err
- }
- return nil
-}
-
-// 更新對象
-func (d *Object) Update() error {
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return err
- }
- stmt, err := db.Prepare("UPDATE " + d.tablename + " SET name = ?, updated_at = ? WHERE id = ?")
- if err != nil {
- log.Println(err)
- return err
- }
- defer stmt.Close()
- _, err = stmt.Exec(d.Name, d.UpdatedAt, d.ID)
- if err != nil {
- log.Println(err)
- return err
- }
- return nil
-}
-
-// 獲取對象
-func (d *Object) Get() error {
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return err
- }
- stmt, err := db.Prepare("SELECT * FROM " + d.tablename + " WHERE id = ?")
- if err != nil {
- log.Println(err)
- return err
- }
- defer stmt.Close()
- row := stmt.QueryRow(d.ID)
- err = row.Scan(&d.ID, &d.Name, &d.CreatedAt, &d.UpdatedAt)
- if err != nil {
- log.Println(err)
- return err
- }
- return nil
-}
-
-// 獲取所有對象
-func (d *Object) GetAll() ([]Object, error) {
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return nil, err
- }
- stmt, err := db.Prepare("SELECT * FROM " + d.tablename)
- if err != nil {
- log.Println(err)
- return nil, err
- }
- defer stmt.Close()
- rows, err := stmt.Query()
- if err != nil {
- log.Println(err)
- return nil, err
- }
- defer rows.Close()
- var list []Object
- for rows.Next() {
- var item Object
- err = rows.Scan(&item.ID, &item.Name, &item.CreatedAt, &item.UpdatedAt)
- if err != nil {
- log.Println(err)
- return nil, err
- }
- list = append(list, item)
- }
- return list, nil
-}
-
-// 獲取分頁對象
-func (d *Object) GetPage(page int64, page_size int64) (ListView, error) {
- db, err := configs.GetDB()
- if err != nil {
- log.Println(err)
- return ListView{}, err
- }
- stmt, err := db.Prepare("SELECT * FROM " + d.tablename + " LIMIT ?, ?")
- if err != nil {
- log.Println(err)
- return ListView{}, err
- }
- defer stmt.Close()
- rows, err := stmt.Query((page-1)*page_size, page_size)
- if err != nil {
- log.Println(err)
- return ListView{}, err
- }
- defer rows.Close()
- var list []interface{}
- for rows.Next() {
- var item Object
- err = rows.Scan(&item.ID, &item.Name, &item.CreatedAt, &item.UpdatedAt)
- if err != nil {
- log.Println(err)
- return ListView{}, err
- }
- list = append(list, item)
- }
- var listview ListView
- listview.Code = 0
- listview.Page = page
- listview.PageSize = page_size
- listview.List = list
- return listview, nil
-}
-
-// 轉換為格式化的JSON
-func (d *Object) ToJSON() []byte {
- b, err := json.MarshalIndent(d, "", " ")
- if err != nil {
- log.Println(err)
- return nil
- }
- return b
-}
+//// 查詢數據(分頁, 每頁數量, 排序, 條件, 關聯, 獲取總數)
+//func (listview *ListView) Read(name string, page int64, page_size int64) error {
+//
+// // 獲取數據庫連接
+// db, err := configs.GetDB()
+// if err != nil {
+// log.Println(err)
+// return err
+// }
+//
+// // 獲取總數
+// var total int64
+// err = db.QueryRow("SELECT COUNT(*) FROM " + name).Scan(&total)
+// if err != nil {
+// log.Println(err)
+// return err
+// }
+//
+// // 獲取數據
+// rows, err := db.Query("SELECT * FROM "+name+" LIMIT ?, ?", (page-1)*page_size, page_size)
+// if err != nil {
+// log.Println(err)
+// return err
+// }
+// defer rows.Close()
+//
+// // 將數據轉換成對象
+// switch name {
+// case "images":
+// for rows.Next() {
+// var image Image
+// err = rows.Scan(&image.ID, &image.Name, &image.CreatedAt, &image.UpdatedAt)
+// if err != nil {
+// log.Println(err)
+// return err
+// }
+// listview.List = append(listview.List, image)
+// }
+// case "models":
+// for rows.Next() {
+// var model Model
+// err = rows.Scan(&model.ID, &model.Name, &model.CreatedAt, &model.UpdatedAt)
+// if err != nil {
+// log.Println(err)
+// return err
+// }
+// listview.List = append(listview.List, model)
+// }
+// default:
+// for rows.Next() {
+// var object Object
+// err = rows.Scan(&object.ID, &object.Name, &object.CreatedAt, &object.UpdatedAt)
+// if err != nil {
+// log.Println(err)
+// return err
+// }
+// listview.List = append(listview.List, object)
+// }
+// }
+//
+// // 設置返回值
+// listview.Code = 0
+// listview.Page = page
+// listview.PageSize = page_size
+// listview.Total = total
+// listview.Next = total > page*page_size
+//
+// return nil
+//}
diff --git a/models/Model.go b/models/Model.go
new file mode 100644
index 0000000..338ff79
--- /dev/null
+++ b/models/Model.go
@@ -0,0 +1,144 @@
+package models
+
+import (
+ "log"
+ "main/configs"
+)
+
+type Model struct {
+ ID int `json:"id"`
+ Name string `json:"name"`
+ Type string `json:"type"` // (lora|ckp|hyper|ti)
+ TriggerWords string `json:"trigger_words"` // 觸發詞
+ BaseModel string `json:"base_model"` // (SD1.5|SD2)
+ ModelPath string `json:"model_path"` // 模型路徑
+ Status string `json:"status"` // (waiting|running|success|error)
+ Progress int `json:"progress"` // (0-100)
+ Tags []string `json:"tags"`
+ CreatedAt string `json:"created_at"`
+ UpdatedAt string `json:"updated_at"`
+}
+
+func (model *Model) Create() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("INSERT INTO models(name, type, created_at, updated_at) values(?, ?, ?, ?)")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ result, err := stmt.Exec(model.Name, model.Type, model.CreatedAt, model.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ id, err := result.LastInsertId()
+ if err != nil {
+ return err
+ }
+ model.ID = int(id)
+ return nil
+}
+
+func (model *Model) Delete() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("DELETE FROM models WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(model.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (model *Model) Update() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("UPDATE models SET name = ?, type = ?, updated_at = ? WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(model.Name, model.Type, model.UpdatedAt, model.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (model *Model) Get() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT * FROM models WHERE id = ?", model.ID).Scan(&model.ID, &model.Name, &model.Type, &model.CreatedAt, &model.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+
+ return nil
+}
+
+func QueryModels(page int, pagesize int) (models []interface{}) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer db.Close()
+ rows, err := db.Query("SELECT * FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer rows.Close()
+ for rows.Next() {
+ var model Model
+ err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.CreatedAt, &model.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ models = append(models, model)
+ }
+ return
+}
+
+func CountModels() (count int) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT COUNT(*) FROM models").Scan(&count)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ return
+}
diff --git a/models/Tag.go b/models/Tag.go
new file mode 100644
index 0000000..f26dab7
--- /dev/null
+++ b/models/Tag.go
@@ -0,0 +1,141 @@
+package models
+
+import (
+ "log"
+ "main/configs"
+)
+
+type Tag struct {
+ ID int `json:"id"`
+ Name string `json:"name"`
+ CreatedAt string `json:"created_at"`
+ UpdatedAt string `json:"updated_at"`
+}
+
+func (tag *Tag) Create() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("INSERT INTO tags(name, created_at, updated_at) values(?, ?, ?)")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ result, err := stmt.Exec(tag.Name, tag.CreatedAt, tag.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ id, err := result.LastInsertId()
+ if err != nil {
+ return err
+ }
+ tag.ID = int(id)
+ return nil
+}
+
+func (tag *Tag) Delete() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("DELETE FROM tags WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(tag.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (tag *Tag) Update() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("UPDATE tags SET name = ?, updated_at = ? WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(tag.Name, tag.UpdatedAt, tag.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func GetTags() ([]Tag, error) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ defer db.Close()
+ rows, err := db.Query("SELECT * FROM tags")
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ defer rows.Close()
+ var tags []Tag
+ for rows.Next() {
+ var tag Tag
+ err := rows.Scan(&tag.ID, &tag.Name, &tag.CreatedAt, &tag.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ tags = append(tags, tag)
+ }
+ return tags, nil
+}
+
+func GetTag(id int) (*Tag, error) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ defer db.Close()
+ row := db.QueryRow("SELECT * FROM tags WHERE id = ?", id)
+ var tag Tag
+ err = row.Scan(&tag.ID, &tag.Name, &tag.CreatedAt, &tag.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ return &tag, nil
+}
+
+func GetTagByName(name string) (*Tag, error) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ defer db.Close()
+ row := db.QueryRow("SELECT * FROM tags WHERE name = ?", name)
+ var tag Tag
+ err = row.Scan(&tag.ID, &tag.Name, &tag.CreatedAt, &tag.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ return &tag, nil
+}
diff --git a/models/Task.go b/models/Task.go
new file mode 100644
index 0000000..01c6d0d
--- /dev/null
+++ b/models/Task.go
@@ -0,0 +1,137 @@
+package models
+
+import (
+ "log"
+ "main/configs"
+)
+
+type Task struct {
+ ID int `json:"id"`
+ Name string `json:"name"`
+ Type string `json:"type"`
+ CreatedAt string `json:"created_at"`
+ UpdatedAt string `json:"updated_at"`
+}
+
+func (task *Task) Create() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("INSERT INTO tasks(name, type, created_at, updated_at) values(?, ?, ?, ?)")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ result, err := stmt.Exec(task.Name, task.Type, task.CreatedAt, task.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ id, err := result.LastInsertId()
+ if err != nil {
+ return err
+ }
+ task.ID = int(id)
+ return nil
+}
+
+func (task *Task) Delete() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("DELETE FROM tasks WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(task.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (task *Task) Update() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("UPDATE tasks SET name = ?, type = ?, updated_at = ? WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(task.Name, task.Type, task.UpdatedAt, task.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (task *Task) Get() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT name, type, created_at, updated_at FROM tasks WHERE id = ?", task.ID).Scan(&task.Name, &task.Type, &task.CreatedAt, &task.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func QueryTasks(page int, pagesize int) (tasks []interface{}) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer db.Close()
+ rows, err := db.Query("SELECT id, name, type, created_at, updated_at FROM tasks LIMIT ?, ?", (page-1)*pagesize, pagesize)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer rows.Close()
+ for rows.Next() {
+ task := Task{}
+ err := rows.Scan(&task.ID, &task.Name, &task.Type, &task.CreatedAt, &task.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ tasks = append(tasks, task)
+ }
+ return
+}
+
+func CountTasks() (count int) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT COUNT(*) FROM tasks").Scan(&count)
+ if err != nil {
+ log.Println(err)
+ return
+ }
+ return
+}
diff --git a/models/User.go b/models/User.go
new file mode 100644
index 0000000..9030a25
--- /dev/null
+++ b/models/User.go
@@ -0,0 +1,138 @@
+package models
+
+import (
+ "log"
+ "main/configs"
+)
+
+type User struct {
+ ID int `json:"id"`
+ Name string `json:"name"`
+ Email string `json:"email"`
+ CreatedAt string `json:"created_at"`
+ UpdatedAt string `json:"updated_at"`
+}
+
+func (user *User) Create() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("INSERT INTO users(name, email, created_at, updated_at) values(?, ?, ?, ?)")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ result, err := stmt.Exec(user.Name, user.Email, user.CreatedAt, user.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ id, err := result.LastInsertId()
+ if err != nil {
+ return err
+ }
+ user.ID = int(id)
+ return nil
+}
+
+func (user *User) Delete() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("DELETE FROM users WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(user.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (user *User) Update() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ stmt, err := db.Prepare("UPDATE users SET name = ?, email = ?, updated_at = ? WHERE id = ?")
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer stmt.Close()
+ _, err = stmt.Exec(user.Name, user.Email, user.UpdatedAt, user.ID)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (user *User) Get() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT name, email, created_at, updated_at FROM users WHERE id = ?", user.ID).Scan(&user.Name, &user.Email, &user.CreatedAt, &user.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
+
+func (user *User) GetAll() ([]User, error) {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ defer db.Close()
+ rows, err := db.Query("SELECT id, name, email, created_at, updated_at FROM users")
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ defer rows.Close()
+ var users []User
+ for rows.Next() {
+ var user User
+ err := rows.Scan(&user.ID, &user.Name, &user.Email, &user.CreatedAt, &user.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return nil, err
+ }
+ users = append(users, user)
+ }
+ return users, nil
+}
+
+func (user *User) GetByEmail() error {
+ db, err := configs.GetDB()
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ defer db.Close()
+ err = db.QueryRow("SELECT id, name, email, created_at, updated_at FROM users WHERE email = ?", user.Email).Scan(&user.ID, &user.Name, &user.Email, &user.CreatedAt, &user.UpdatedAt)
+ if err != nil {
+ log.Println(err)
+ return err
+ }
+ return nil
+}
diff --git a/templates/index.html b/templates/index.html
new file mode 100644
index 0000000..69e55d8
--- /dev/null
+++ b/templates/index.html
@@ -0,0 +1,26 @@
+
+
+
+
+ Home
+
+
+ Home
+ Hi, {{ .Name }}!
+
+
+
+
+