初尝Golang泛型

旧场景

首先让我们看下没有使用泛型的情况下写的代码：

type Resp struct {
	Code int         `json:"code"`
	Msg  string      `json:"msg"`
	Time int         `json:"time"`
	Data interface{} `json:"data"`
}

type Article struct {
	Title   string `json:"title"`
	Content string `json:"content"`
}

func handle() {
	b, err := request("GET", "/api/article", nil)
	if err != nil {
		panic(fmt.Errorf("get article error: %v\n", err))
	}
	// 获取文章
	var res1 Resp
	var article Article
	json.Unmarshal(b, &res1)
	articleByte, err := json.Marshal(res1.Data)
	if err != nil {
		panic(fmt.Errorf("marshal resp data error: %v\n", err))
	}
	json.Unmarshal(articleByte, article)
	fmt.Printf("Title: %s, Content: %s\n", article.Title, article.Content)

}

func request(method, url string, params map[string]interface{}) (b []byte, err error) {
	jsonData, err := json.Marshal(params)
	if err != nil {
		return
	}
	client := &http.Client{}
	req, err := http.NewRequest(method, url, bytes.NewReader(jsonData))
	if err != nil {
		return
	}
	// 处理GET请求的参数
	if strings.ToUpper(method) == "GET" {
		q := req.URL.Query()
		for key, val := range params {
			q.Add(key, val.(string))
		}
		req.URL.RawQuery = q.Encode()
	}

	// do request
	resp, err := client.Do(req)
	if err != nil {
		return nil, fmt.Errorf("request failed: %v", err)
	}

	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)

	b, err = io.ReadAll(resp.Body)
	if err != nil {
		return nil, fmt.Errorf("read response body failed: %v", err)
	}

	return
}

脑袋嗡嗡的，看完调用示例，我是发誓不会调用这个方法的，用一次蛋疼一次。

新方案

好在，我们现在可以使用泛型了，下面是使用泛型改造后的代码：

type Resp[T any] struct {
	Code int    `json:"code"`
	Msg  string `json:"msg"`
	Time int    `json:"time"`
	Data T      `json:"data"`
}

type Article struct {
	Title   string `json:"title"`
	Content string `json:"content"`
}

func handle() {
	b, err := request("GET", "/api/article", nil)
	if err != nil {
		panic(fmt.Errorf("get article error: %v\n", err))
	}
	// 获取文章
	article := b.Data.(Article)
	fmt.Printf("Title: %s, Content: %s\n", article.Title, article.Content)

}

func request(method, url string, params map[string]interface{}) (b Resp[any], err error) {
	jsonData, err := json.Marshal(params)
	if err != nil {
		return
	}
	client := &http.Client{}
	req, err := http.NewRequest(method, url, bytes.NewReader(jsonData))
	if err != nil {
		return
	}
	// 处理GET请求的参数
	if strings.ToUpper(method) == "GET" {
		q := req.URL.Query()
		for key, val := range params {
			q.Add(key, val.(string))
		}
		req.URL.RawQuery = q.Encode()
	}

	// do request
	resp, err := client.Do(req)
	if err != nil {
		return Resp[any]{}, fmt.Errorf("request failed: %v", err)
	}

	defer func(Body io.ReadCloser) {
		_ = Body.Close()
	}(resp.Body)

	respB, err := io.ReadAll(resp.Body)
	if err != nil {
		return Resp[any]{}, fmt.Errorf("read response body failed: %v", err)
	}

	json.Unmarshal(respB, &b)

	return
}

改造后的调用示例看起来心智负担小多了，感谢泛型拯救我的生命。