下載圖像

routers/datasets.go

@@ -1,57 +1,31 @@
 package routers
 
 import (
-	"database/sql/driver"
 	"encoding/json"
 	"io/ioutil"
 	"main/configs"
 	"main/models"
 	"main/utils"
 	"net/http"
-	"time"
 
 	"github.com/gorilla/mux"
 )
 
-type ImageList []string
-
-func (list *ImageList) Scan(value interface{}) error {
-	return json.Unmarshal(value.([]byte), list)
-}
-func (list ImageList) Value() (driver.Value, error) {
-	return json.Marshal(list)
-}
-
-// 數據集模型
-type Dataset struct {
-	ID        int       `json:"id" gorm:"primary_key"`
-	Name      string    `json:"name"`
-	Info      string    `json:"info"`
-	Images    ImageList `json:"images"`
-	UserID    int       `json:"user_id"`
-	CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
-	UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
-}
-
-func init() {
-	configs.ORMDB().AutoMigrate(&Dataset{})
-}
-
 // 獲取數據集列表
 func DatasetsGet(w http.ResponseWriter, r *http.Request) {
 	var listview models.ListView
 	listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1)
 	listview.PageSize = utils.ParamInt(r.URL.Query().Get("pageSize"), 10)
-	var dataset_list []Dataset
+	var dataset_list []models.Dataset
 	db := configs.ORMDB()
 	db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&dataset_list)
 	for _, dataset := range dataset_list {
 		if dataset.Images == nil {
-			dataset.Images = ImageList{}
+			dataset.Images = models.ImageList{}
 		}
 		listview.List = append(listview.List, dataset)
 	}
-	db.Model(&Dataset{}).Count(&listview.Total)
+	db.Model(&models.Dataset{}).Count(&listview.Total)
 	listview.Next = listview.Page*listview.PageSize < int(listview.Total)
 	listview.WriteJSON(w)
 }
@@ -59,7 +33,7 @@ func DatasetsGet(w http.ResponseWriter, r *http.Request) {
 // 創建數據集
 func DatasetsPost(w http.ResponseWriter, r *http.Request) {
 	models.AccountRead(w, r, func(account *models.Account) {
-		var dataset Dataset
+		var dataset models.Dataset
 		body, err := ioutil.ReadAll(r.Body)
 		if err != nil {
 			w.WriteHeader(http.StatusInternalServerError)
@@ -73,7 +47,7 @@ func DatasetsPost(w http.ResponseWriter, r *http.Request) {
 			return
 		}
 		if dataset.Images == nil {
-			dataset.Images = ImageList{}
+			dataset.Images = models.ImageList{}
 		}
 		dataset.UserID = account.ID
 		if err := configs.ORMDB().Create(&dataset).Error; err != nil {
@@ -88,7 +62,7 @@ func DatasetsPost(w http.ResponseWriter, r *http.Request) {
 
 // 獲取數據集
 func DatasetsItemGet(w http.ResponseWriter, r *http.Request) {
-	dataset := Dataset{ID: utils.ParamInt(mux.Vars(r)["dataset_id"], 0)}
+	dataset := models.Dataset{ID: utils.ParamInt(mux.Vars(r)["dataset_id"], 0)}
 	if err := configs.ORMDB().Find(&dataset).Error; err != nil {
 		w.WriteHeader(http.StatusNotFound)
 		w.Write([]byte("404 - Not Found"))
@@ -101,7 +75,7 @@ func DatasetsItemGet(w http.ResponseWriter, r *http.Request) {
 // 修改數據集
 func DatasetsItemPatch(w http.ResponseWriter, r *http.Request) {
 	models.AccountRead(w, r, func(account *models.Account) {
-		var dataset Dataset = Dataset{ID: utils.ParamInt(mux.Vars(r)["dataset_id"], 0)}
+		var dataset models.Dataset = models.Dataset{ID: utils.ParamInt(mux.Vars(r)["dataset_id"], 0)}
 		if err := configs.ORMDB().Find(&dataset).Error; err != nil {
 			w.WriteHeader(http.StatusNotFound)
 			w.Write([]byte("404 - Not Found"))
@@ -120,7 +94,7 @@ func DatasetsItemPatch(w http.ResponseWriter, r *http.Request) {
 			dataset.Info = info
 		}
 		if images, ok := form["images"].([]interface{}); ok {
-			var image_list ImageList
+			var image_list models.ImageList
 			for _, image := range images {
 				if image, ok := image.(string); ok {
 					image_list = append(image_list, image)
@@ -142,7 +116,7 @@ func DatasetsItemPatch(w http.ResponseWriter, r *http.Request) {
 func DatasetsItemDelete(w http.ResponseWriter, r *http.Request) {
 	models.AccountRead(w, r, func(account *models.Account) {
 		// 獲取數據集
-		dataset := Dataset{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
+		dataset := models.Dataset{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
 		if err := configs.ORMDB().Find(&dataset).Error; err != nil {
 			w.WriteHeader(http.StatusNotFound)
 			w.Write([]byte("404 - Not Found"))

routers/models.go

@@ -92,8 +92,10 @@ func ModelsPost(w http.ResponseWriter, r *http.Request) {
 			log.Println(err)
 			return
 		}
+
 		// 直接提交訓練任務
 		go model.Train()
+
 		// 返回創建的模型
 		w.Header().Set("Content-Type", "application/json; charset=utf-8")
 		w.Write(utils.ToJSON(model))