gorm
This commit is contained in:
@@ -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
|
||||
//}
|
||||
|
Reference in New Issue
Block a user