like list

models/account.go

@@ -1,19 +1,38 @@
 package models
 
 import (
+	"log"
 	"main/configs"
 	"net/http"
 	"time"
 )
 
 type Account struct {
-	ID        int       `json:"id"`
+	ID        int         `json:"id"`
-	Name      string    `json:"name"`
+	Name      string      `json:"name"`
-	Email     string    `json:"email"`
+	Email     string      `json:"email"`
-	Admin     bool      `json:"admin"`
+	Admin     bool        `json:"admin"`
-	SessionID string    `json:"session_id"`
+	SessionID string      `json:"session_id"`
-	CreatedAt time.Time `json:"created_at"`
+	CreatedAt time.Time   `json:"created_at"`
-	UpdatedAt time.Time `json:"updated_at"`
+	UpdatedAt time.Time   `json:"updated_at"`
+	LikeList  interface{} `json:"likes" gorm:"-"`
+}
+
+func (account *Account) ReadLikeList() {
+	is, err := LikeImage.GetA(string(rune(account.ID)))
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	ms, err := LikeModel.GetA(string(rune(account.ID)))
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	account.LikeList = map[string]interface{}{
+		"images": is,
+		"models": ms,
+	}
 }
 
 func AccountRead(w http.ResponseWriter, r *http.Request, cb func(account *Account)) {

routers/account.go

@@ -9,6 +9,7 @@ import (
 // 獲取當前賬戶信息(重寫, 爲輸出增加sid字段)
 func AccountGet(w http.ResponseWriter, r *http.Request) {
 	models.AccountRead(w, r, func(account *models.Account) {
+		account.ReadLikeList()
 		w.Header().Set("Content-Type", "application/json; charset=utf-8")
 		w.Write(utils.ToJSON(account))
 	})