正确的代码组织方法在我的具体情况下如何实施？

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《正确的代码组织方法在我的具体情况下如何实施？》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

几乎有两个相同的函数做着大致相同的事情。在这种情况下，组织代码以避免重复的正确方法是什么？ httpgetter() 函数访问云平台 api 并获取 json 响应，然后我在另一个函数中解析该响应，并基于它从模板形成 terraform 清单。 gettoken() 函数执行几乎相同的操作，只是获取一个令牌，然后在 httpgetter() 函数中使用该令牌。

var accessToken = getToken()


func httpGetter(method, url string) (*http.Response, []byte) {
    client := &http.Client{}
    req, err := http.NewRequest(method, url, nil)
    if err != nil {
        fmt.Println(err)
    }
    req.Header.Add("Accept", "application/json;version=35.0")
    req.Header.Add("Authorization", "Bearer "+accessToken)
    res, err := client.Do(req)
    if err != nil {
        fmt.Println(err)
    }
    defer res.Body.Close()
    body, err := ioutil.ReadAll(res.Body)
    if err != nil {
        fmt.Println(err)
    }
    return res, body
}

func getToken() string {
    url := "https://cloud-platform-api.com/api/sessions"
    method := "POST"
    client := &http.Client{}
    req, err := http.NewRequest(method, url, nil)
    if err != nil {
        fmt.Println(err)
    }
    req.Header.Add("Accept", "application/*+xml;version=35.0")
    req.Header.Add("Authorization", "Basic <auth-hash>")
    res, err := client.Do(req)
    if err != nil {
        fmt.Println(err)
    }
    defer res.Body.Close()
    if err != nil {
        fmt.Println(err)
    }
    accessToken := res.Header.Get("x-vmware-vcloud-access-token")
    return accessToken
}

正确答案

首先，如果您知道该方法是 getter，那么您不需要传递方法参数，因此签名将如下所示。另外，您已经将 *http.response 返回给调用者，现在将由调用者决定如何处理响应 + 调用者应该决定在 http 调用失败时做什么，因此返回错误并让调用者决定。

func httpget(url string) (*http.response, error)

现在您还需要带有主体的 post 方法（在某些情况下），因此有另一个功能

func httppost(url string, body []byte) (*http.response, error)

现在，要管理签名并拥有通用代码，您可以拥有一个仅在此文件中使用的私有方法，或者您也可以公开该方法（由您决定）

type headers map[string]string

func http(method, url string, body []byte, headers headers) (*http.response, error) { // we pass headers here so that the caller can pass custom headers for request
    client := &http.client{}
    req, err := http.newrequest(method, url, body)
    if err != nil {
        return nil, err
    }

    req.header.add("accept", "application/json;version=35.0") // common static header you can keep as it is

    for key, value := range headers {
       req.header.add(key, value)
    }
    
    return client.do(req)
}

使用此方法，您从两个方法中进行的调用将如下所示

func httpget(url string, headers headers) (*http.response, error) {
  return http(http.methodget, url, nil, headers)
}

func httppost(url string, body []byte, headers headers) (*http.response, error) {
  return http(http.methodpost, url, body, headers)
}

现在您可以使用它来传递来自调用者的身份验证令牌，例如：

func gettoken() {
  res, err := httppost("https://cloud-platform-api.com/api/sessions", nil, 
    map[string]string{
      "authorization": "basic <auth-hash>",
    }

  if err != nil {
    // do something with error
  }

  if res.statuscode == http.statuscreated {
     // do what you want with the success response like unmarshalling to json
  }
}

对于不需要传递标头的情况，可以这样做

res, err := httpGet("some-url", nil) // you pass header as nil

理论要掌握，实操不能落！以上关于《正确的代码组织方法在我的具体情况下如何实施？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！