服務器狀態

models/Model.go

@@ -21,6 +21,29 @@ type Model struct {
 	UserID       int    `json:"user_id"`
 }
 
+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) Create() error {
 	db, err := configs.GetDB()
 	if err != nil {