fix

models/chain.go

@@ -18,6 +18,11 @@ type Chain struct {
 	BrowerUrl  string  `json:"brower_url"`  //浏览器链接
 }
 
+type UserChain struct {
+	Id       int    `json:"id"`
+	Platform string `json:"platform"` //平行链名称
+}
+
 func (c Chain) TableName() string {
 	return setting.DatabaseSetting.Name_Sources + ".wallet_chain"
 }
@@ -71,6 +76,17 @@ func GetChains(pageNum, pageSize int, maps interface{}) ([]*Chain, error) {
 	return chains, nil
 }
 
+func GetUserChains(maps interface{}) ([]*UserChain, error) {
+	var chains []*UserChain
+
+	err := db.Select("id, platform").Table("wallet_chain").Where(maps).Scan(&chains).Error
+	if err != nil && err != gorm.ErrRecordNotFound {
+		return nil, err
+	}
+
+	return chains, nil
+}
+
 /**
  * 通过id获取指定链信息
  * @param id

routers/api/v1/chain.go

@@ -99,6 +99,42 @@ func GetChains(c *gin.Context) {
 	handler.SendResponse(c, nil, data)
 }
 
+func GetUserChains(c *gin.Context) {
+	valid := validation.Validation{}
+
+	platform := ""
+	if arg := c.Query("platform"); arg != "" {
+		platform = c.Query("platform")
+	}
+
+	if valid.HasErrors() {
+		handler.SendResponse(c, errno.ErrValidation, nil)
+		return
+	}
+
+	chainService := chain_service.Chain{
+		Platform: platform,
+	}
+
+	total, err := chainService.Count()
+	if err != nil {
+		handler.SendResponse(c, errno.ErrCountChain, nil)
+		return
+	}
+
+	chains, err := chainService.GetUserChain()
+	if err != nil {
+		handler.SendResponse(c, errno.InternalServerError, nil)
+		return
+	}
+
+	data := make(map[string]interface{})
+	data["items"] = chains
+	data["total"] = total
+
+	handler.SendResponse(c, nil, data)
+}
+
 func AddChain(c *gin.Context) {
 	token := c.Request.Header.Get("Token")
 	authService := auth_service.Auth{Token: token}

routers/router.go

@@ -53,6 +53,7 @@ func InitRouter() *gin.Engine {
 		api.GET("/chain", v1.GetChain)
 		api.PUT("/chain", v1.EditChain)
 		api.DELETE("/chain", v1.DeleteChain)
+		api.GET("/user-chains", v1.GetUserChains)
 
 		api.GET("/recommend-categories", v1.GetWalletRecommendCategories)
 		api.POST("/recommend-category", v1.AddWalletRecommendCategory)

service/chain_service/chain.go

@@ -36,6 +36,17 @@ func (c *Chain) GetAll() ([]*models.Chain, error) {
 	return chains, nil
 }
 
+func (c *Chain) GetUserChain() ([]*models.UserChain, error) {
+	var chains []*models.UserChain
+
+	chains, err := models.GetUserChains(c.getMaps())
+	if err != nil {
+		return nil, err
+	}
+
+	return chains, nil
+}
+
 func (c *Chain) Add() error {
 	chain := map[string]interface{}{
 		"platform":    c.Platform,