### goshop開源專案的更新

> 備註：前面專案中用到的程式碼已經分享到GitHub中去了，並且以後所有專案中會出現的程式碼都會提交上去，歡迎查閱。感興趣的可以點個star哦~  
https://gitee.com/jobhandsome/goshop/

最近在研究微服務框架go-zero，沒有更新，在這裡說聲抱歉，後面會持續更新新的功能。請繼續關注~~~~

今天考慮引數繫結獲取。這裡肯定有些疑惑，之前不是封裝了獲取全部引數的方法嗎？

回顧之前我們們封裝了獲取全部引數的方法，只能是`POST`請求，並且請求方式必須是：`application/x-www-form-urlencoded` 才能正常獲取，侷限性非常大。

* 需求1：使用任意請求方式，任意請求格式都可以獲取到指定的值？

* 需求2：獲取到的`struct`型別引數，轉換成`map`型別用於`gorm`的`DB`查詢？

#### 透過分析上面的兩個需求得到幾個結論：

* `gin`框架中可以使用引數繫結進行任意請求方式和請求型別獲取請求的引數

* `gin`框架整合`gorm`，使用`DB.Where`條件查詢時，需要傳入`map`型別的陣列格式，需要`struct轉map`。

#### 既然知道要做什麼了，那就開始：

首先，先使用繫結的方式獲取引數：
```go
//1. 先定義一下傳入的引數，這裡使用form進行繫結，json進行別名
type ParamsRequest struct {
  Page     int64  `form:"page" binding:"required" json:"page"`
  PageSize int64  `form:"pageSize" binding:"required" json:"pageSize"`
  Name     string `form:"name" json:"name"`
}
//2. 使用繫結來獲取請求的引數 ctx => *gin.Context
var params ParamsRequest
if err := ctx.ShouldBind(&params); err != nil {
  utils.Fail(ctx, "引數錯誤："+err.Error(), nil)
  return
}
```
獲取到引數後，我們們就要使用引數進行`gorm`的`DB.Where`查詢：
1. 首先先對 `struct` 轉換 `map`
```go
// 對應方法 AnyToMap interface 轉 ma
func AnyToMap(v any) (map[string]any, error) {
  dataJson, err := json.Marshal(v)
  if err != nil {
    return nil, err
  }
  var MapData map[string]any
  err = json.Unmarshal(dataJson, &MapData)
  if err != nil {
    return nil, err
  }
  return MapData, nil
}
```
2. 需要 過濾`page`和`pageSize`，不參與查詢的
```go
ParamsFilter := utils.ParamsFilter("page,pageSize", paramsMap)

// 過濾指定陣列中的key
func ParamsFilter(isFilterStr string, params map[string]any) map[string]any {
  var data = make(map[string]any)
  for key, value := range params {
    if value != "" {
      find := strings.Contains(isFilterStr, key)
      if !find {
        data[key] = value
      }
    }
  }
  return data
}
```
3. 前面都是鋪墊，現在才是重點，使用`gorm`進行資料查詢
```go
// 獲取 gorm 的 DB 控制程式碼
DB := config.GetDB()
// 這裡使用自己定義的模型就好
var Result []*model.Category
// 使用 Scopes 傳入要分頁的指標
resErr := DB.Scopes(Paginate.Paginate(strconv.FormatInt(params.Page, 10), strconv.FormatInt(params.PageSize, 10))).Where(ParamsFilter).Find(&Result).Error
// 查詢的返回錯誤，如果有錯誤，就返回
if resErr != nil {
  utils.Fail(ctx, resErr.Error(), nil)
  return
}
// 對查詢的 struct 陣列轉換 樹結構 ToTree 
var category model.Category
ResultLists := category.ToTree(Result)
utils.Success(ctx, "獲取成功", ResultLists)
```

* 對應的`model`檔案：

```go
type Category struct {
  gorm.Model
  Name     string         `form:"name" binding:"required" json:"name" gorm:"varchar(255);not null;default:'';comment:'分類名稱'"` // 分類名稱
  Pid      int64          `form:"pid" json:"pid" gorm:"size:11;not null;default:0;comment:'分類節點'"`                            // 分類節點：0根節點
  Icon     string         `form:"icon" json:"icon" gorm:"varchar(255);not null;default:'';comment:'分類圖示'"`                    // 分類圖示
  State    int64          `form:"state" json:"state" gorm:"size:1;not null;default:0;comment:'分類狀態：0未啟用，1已啟用'"`               // 分類狀態：0未啟用，1已啟用
  Sort     int64          `form:"sort" json:"sort" gorm:"size:11;not null;default:0;comment:'分類排序'"`                          // 分類排序
  Tag      string         `form:"tag" json:"tag" gorm:"varchar(255);not null;default:0;comment:'分類標籤'"`                       // 分類標籤
  Children *CategoryTrees `json:"children"`
}

// CategoryTrees 二叉樹列表
type CategoryTrees []*Category

// ToTree 轉換為樹形結構
func (Category) ToTree(data CategoryTrees) CategoryTrees {
  // 定義 HashMap 的變數，並初始化
  TreeData := make(map[int64]*Category)
  // 先重組資料：以資料的ID作為外層的key編號，以便下面進行子樹的資料組合
  for _, item := range data {
    TreeData[int64(item.ID)] = item
  }
  // 定義 RoleTrees 結構體
  var TreeDataList CategoryTrees
  // 開始生成樹形
  for _, item := range TreeData {
    // 如果沒有根節點樹，則為根節點
    if item.Pid == 0 {
      // 追加到 TreeDataList 結構體中
      TreeDataList = append(TreeDataList, item)
      // 跳過該次迴圈
      continue
    }
    // 透過 上面的 TreeData HashMap的組合，進行判斷是否存在根節點
    // 如果存在根節點，則對應該節點進行處理
    if pItem, ok := TreeData[item.Pid]; ok {
      // 判斷當次迴圈是否存在子節點，如果沒有則作為子節點進行組合
      if pItem.Children == nil {
        // 寫入子節點
        children := CategoryTrees{item}
        // 插入到 當次結構體的子節點欄位中，以指標的方式
        pItem.Children = &children
        // 跳過當前迴圈
        continue
      }
      // 以指標地址的形式進行追加到結構體中
      *pItem.Children = append(*pItem.Children, item)
    }
  }
  return TreeDataList
}
```

到了這一步，我們們就實現了上面需求的功能。 更多功能請持續關注！！！！！

#### 星球地址：https://t.zsxq.com/03MJM7YfI

#### **關注公眾號「程式設計師小喬」**

![程式設計師小喬](https://www.qsjob.fun/zb_users/upload/2022/07/202207041451541741109.jpg "程式設計師小喬")

Go專案實戰—引數繫結，型別轉換

相關文章