websocket

models/Model.go

@@ -104,6 +104,21 @@ func (model *Model) Get() error {
 	return nil
 }
 
+func QueryModel(id int) (model Model) {
+	db, err := configs.GetDB()
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer db.Close()
+	err = db.QueryRow("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id FROM models WHERE id = ?", id).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
+	}
+	return
+}
+
 func QueryModels(page int, pagesize int) (models []interface{}) {
 	db, err := configs.GetDB()
 	if err != nil {

models/Task.go

@@ -99,6 +99,21 @@ func (task *Task) Get() error {
 	return nil
 }
 
+func QueryTask(id int) (task Task) {
+	db, err := configs.GetDB()
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer db.Close()
+	err = db.QueryRow("SELECT id, name, type, created_at, updated_at FROM tasks WHERE id = ?", id).Scan(&task.ID, &task.Name, &task.Type, &task.CreatedAt, &task.UpdatedAt)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	return
+}
+
 func QueryTasks(page int, pagesize int) (tasks []interface{}) {
 	db, err := configs.GetDB()
 	if err != nil {