This commit is contained in:
2023-05-14 07:00:24 +08:00
parent 2423213e9a
commit ee3b60eccc
18 changed files with 273 additions and 1221 deletions

View File

@@ -1,7 +1,6 @@
package models
import (
"log"
"main/configs"
)
@@ -25,25 +24,25 @@ func init() {
configs.ORMDB().AutoMigrate(&Model{})
}
func (model *Model) SendToTrain() error {
db, err := configs.GetDB()
if err != nil {
log.Println(err)
return err
}
defer db.Close()
stmt, err := db.Prepare("UPDATE models SET status = ?, progress = ?, updated_at = ? WHERE id = ?")
if err != nil {
log.Println(err)
return err
}
defer stmt.Close()
_, err = stmt.Exec(model.Status, model.Progress, model.UpdatedAt, model.ID)
if err != nil {
log.Println(err)
return err
}
// TODO: 創建一個新線程管理訓練任務
// 將任務放入隊列中, 將自動回調更新任務狀態
return nil
}
//func (model *Model) SendToTrain() error {
// db, err := configs.GetDB()
// if err != nil {
// log.Println(err)
// return err
// }
// defer db.Close()
// stmt, err := db.Prepare("UPDATE models SET status = ?, progress = ?, updated_at = ? WHERE id = ?")
// if err != nil {
// log.Println(err)
// return err
// }
// defer stmt.Close()
// _, err = stmt.Exec(model.Status, model.Progress, model.UpdatedAt, model.ID)
// if err != nil {
// log.Println(err)
// return err
// }
// // TODO: 創建一個新線程管理訓練任務
// // 將任務放入隊列中, 將自動回調更新任務狀態
// return nil
//}