1. 程式人生 > 程式設計 >go json資料轉發的實現程式碼

go json資料轉發的實現程式碼

案例

例如,有個 GET 介面,可以批量獲取使用者資訊👇

> curl 'http://localhost:8080/user/1,2,3'
[
  {
    "user_id":1,"other_suff":...
  },{
    "user_id":2,{
    "user_id":3,"other_suff":...
  }
]

同時,我們要將使用者資訊和他們的某些訂單資訊放在一起,組裝成為👇的介面,滿足其他業務需求。

[
  {
    "user_info":{
      "user_id":1,"other_suff":...
    },"order_info":{
      "order_id":1,"user_id":1,"other_suff":...
    }
  },{
    "user_info":{
      "user_id":2,"order_info":{
      "order_id":2,"user_id":2,{
    "user_info":{
      "user_id":3,"order_info":{
      "order_id":3,"user_id":3,"other_suff":...
    }
  }
]

分析

解決這個問題很簡單:把user資訊和order資訊的json用工具解析得到結構體,然後呼叫他們的介面得到資料,根據id關聯和拼裝,最後返回。

這樣的做法存在的一個問題是,程式碼解析了user和order的完整結構。如果user介面返回的使用者資訊增加了欄位,我們這裡的結構體要同步更新,否則我們給出的資料就是不完整的。(這可能是很痛苦的,你要求別的團隊加欄位,得排期...)

其實我們作為資料的“中間商”,只關心user介面json裡的 user_id ,我們使用這個欄位關聯order資料。對於user資訊裡的 other_suff 或者其他資料,我們並不關心,只要保證完整傳出去就好了。

根據 https://golang.org/pkg/encoding/json/#Unmarshal ,可以知道直接丟一個 map[string]interface{} 給 json.Unmarshal 也可以正常解析的,於是我們可以寫出比較通用的透傳程式碼。

type Content []map[string]interface{}

func (c Content) GetByFieldName(name string,defaultVal interface{}) infterface{} {
  for _,item := range c {
    val,ok := item[name]
    if !ok {
      continue
    }
    if val == nil {
      return defaultVal
    }
    return val
  }
  return defaultVal
}

func getUserContentByIDs(ids []int) Content {
  ...
  var c Content
  err := json.Unmarshal(jsonData,&c) 
  ...
  return c
}

func getOrderContentByUserIDs(ids []int) Content {.../*同上*/}

func Handler(userIDs []int) []Combine {

  users := getUserContentByIDs(userIDs)
  orders := getOrderContentByUserIDs(userIDs)

  // 這裡假設使用者和訂單是一對一的關係
  ret := make([]Combine,len(users))
  for _,u := range users {
    for _,o := range orders {
      userID := u.GetByFieldName("user_id",0) 
      orderUserID := o.GetByFieldName("user_id",0)
      if userID != 0 && userID == orderUserID {
        ret = append(ret,Combine{
          UserInfo: u,OrderInfo: o,})
        break
      }
    }
  }
  return ret
}

P.S. 在上面的例子中,每次查詢Content都要遍歷陣列。如果資料量大或者查詢頻繁,可以在初始化Content的時候,根據item的唯一標標識,再給Content根據封裝一個map,提高查詢效率。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。