go mock server的簡易實(shí)現(xiàn)示例

更新時(shí)間：2022年07月25日 16:42:03 作者：shinji

這篇文章主要為大家介紹了go mock server的簡易實(shí)現(xiàn)示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

學(xué)習(xí)golang也一段時(shí)間了，看了一些書，上周又看了一本入門級(jí)的《Go語言趣學(xué)指南》，是時(shí)候檢驗(yàn)成果了。

目的：通過讀取本地mock數(shù)據(jù)，發(fā)起http請(qǐng)求，返回給前端，實(shí)現(xiàn)mock功能。

代碼

整體代碼只在50行左右，是一個(gè)極其簡陋的mock server，沒有多余的功能(主要是不會(huì)

步驟1

掃描本地以json結(jié)尾的文件，獲取文件路徑

// 獲取當(dāng)前json文件所在的路徑已經(jīng)文件名，然后進(jìn)行拼接
func ParsePath() (path string, err error) {
   pwd, _ := os.Getwd()
   fileList, err := ioutil.ReadDir(pwd)
   if err != nil {
      log.Fatal(err)
   }
   var curJsonFile string
   for _, v := range fileList {
      name := v.Name()
      if isJson := strings.Contains(name, "json"); isJson {
         curJsonFile = name
      }
   }
   if curJsonFile == "" {
      return curJsonFile, errors.New("dont have json file")
   }
   absPath := pwd + "/" + curJsonFile
   return absPath, nil
}

步驟2

// 根據(jù)傳入的文件名讀取具體的[]byte數(shù)據(jù)
func ReadJSON(path string) ([]byte, error) {
   jsonFile, err := os.Open(path)
   if err != nil {
      log.Fatal("open file err")
   }
   defer jsonFile.Close()
   byteValue, err := ioutil.ReadAll(jsonFile)
   if err != nil {
      return nil, err
   }
   return byteValue, nil
}

步驟3

// 根據(jù)傳入的[]byte數(shù)據(jù)，將數(shù)據(jù)轉(zhuǎn)換反序列化為map類型
// 這里由于不知道json文件的具體數(shù)據(jù)，具體結(jié)構(gòu)，所以用了map[string]interface{}的
// 數(shù)據(jù)類型
func DecodeJSONString(jsonBytes []byte) (map[string]interface{}, error) {
   var m map[string]interface{}
   err := json.Unmarshal(jsonBytes, &amp;m)
   if err != nil {
      log.Fatal("json unmarshal fail", err)
      return nil, err
   }
   return m, nil
}

步驟4

var data map[string]interface{}
func returnResponse(w http.ResponseWriter, r *http.Request) {
   w.Header().Set("content-type", "application/json")
   path := r.URL.Path
   if d := data[path]; d != nil {
      stringData, err := json.Marshal(data[path])
      if err != nil {
         log.Fatal(err)
      }
      w.Write(stringData)
   }
}
func main() {
   path, err := src.ParsePath()
   if err != nil {
      log.Fatal(err)
   }
   bytes, err := src.ReadJSON(path)
   if err != nil {
      log.Fatal(err)
   }
   m, err := src.DecodeJSONString(bytes)
   data = m
   if err != nil {
      log.Fatal("decode fail: ", err)
   }
   for addr, _ := range data {
      http.HandleFunc(addr, returnResponse)
   }
   err = http.ListenAndServe(":3000", nil)
   if err != nil {
      log.Fatal("listenAndServer fail")
   }
}

最終效果

本地訪問localhost:3000/v1/lottery 前端可以拿到返回的數(shù)據(jù)

mock的本地json文件

{
  "/v1/lottery": [{
    "name": "shinji",
    "age": 20
  }, {
    "name": "ayanami",
    "age": 20
  }]
}

最后

這個(gè)是一個(gè)非常簡易的mock server，還不支持多文件格式，多個(gè)文件，多種請(qǐng)求方法，自定義規(guī)則mock數(shù)據(jù)等等。也只是對(duì)當(dāng)前Golang學(xué)習(xí)后的一次小測驗(yàn)，很簡單，更多關(guān)于go mock server實(shí)現(xiàn)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: