go mock server的簡易實現(xiàn)示例
前言
學(xué)習(xí)golang也一段時間了,看了一些書,上周又看了一本入門級的《Go語言趣學(xué)指南》,是時候檢驗成果了。
目的:通過讀取本地mock數(shù)據(jù),發(fā)起http請求,返回給前端,實現(xiàn)mock功能。
代碼
整體代碼只在50行左右,是一個極其簡陋的mock server,沒有多余的功能(主要是不會
步驟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, &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 }] }
最后
這個是一個非常簡易的mock server, 還不支持多文件格式,多個文件,多種請求方法,自定義規(guī)則mock數(shù)據(jù)等等。也只是對當(dāng)前Golang學(xué)習(xí)后的一次小測驗,很簡單,更多關(guān)于go mock server實現(xiàn)的資料請關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
golang?pprof?監(jiān)控goroutine?thread統(tǒng)計原理詳解
這篇文章主要為大家介紹了golang?pprof?監(jiān)控goroutine?thread統(tǒng)計原理詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-04-04