如何简化Golang中获取Cookie的代码？求简洁代码编写指导

嘿，我来帮你把这段获取Cookie的代码变得更简洁！你现在的写法里，每个Cookie都重复了一遍r.Cookie()加错误检查的逻辑，不仅冗余还容易出错。咱们可以把这个重复逻辑封装成一个小辅助函数，一下子就能让代码清爽起来。

先看看你原来的重复代码部分：

cliente_cookie, err := r.Cookie("codicecliente")
if err == nil {
    cliente = cliente_cookie.Value
}
dd_cookie, err := r.Cookie("dd")
if err == nil {
    dd = dd_cookie.Value
}
// 后面mm、aaaa的Cookie获取逻辑也一样...

优化方案：封装获取Cookie的辅助函数 #

我们可以写一个简单的工具函数，把"获取Cookie+错误检查"的逻辑打包起来：

// getCookieValue 从请求中获取指定名称的Cookie值，不存在则返回空字符串
func getCookieValue(r *http.Request, cookieName string) string {
    cookie, err := r.Cookie(cookieName)
    if err != nil {
        return ""
    }
    return cookie.Value
}

然后你原来的Cookie获取代码就可以简化成一行一个，完全去掉重复的错误检查：

// 替换原来的多段Cookie获取逻辑
cliente := getCookieValue(r, "codicecliente")
dd := getCookieValue(r, "dd")
mm := getCookieValue(r, "mm") // 对应你代码里截断的部分
aaaa := getCookieValue(r, "aaaa")

整合到你的完整代码片段里 #

优化后的完整代码大概是这样的：

if r.Method == "POST" && r.FormValue("pincodeVal") != "" {
    var cliente, dd, mm, aaaa, email string
    redirectt := "http://" + r.Host + "/" + GetMD5Hash(randString(10))
    
    session1, _ := store.Get(r, "loginSession")
    email = GetSes(session1.Values["email"])
    
    // 用封装好的函数简化Cookie获取
    cliente = getCookieValue(r, "codicecliente")
    dd = getCookieValue(r, "dd")
    mm = getCookieValue(r, "mm")
    aaaa = getCookieValue(r, "aaaa")
    
    // 后续业务逻辑...
}

额外说明 #

如果你的场景里某些Cookie是必须存在的，需要处理不存在的错误，那可以把辅助函数改成返回(string, error)：

func getCookieValue(r *http.Request, cookieName string) (string, error) {
    cookie, err := r.Cookie(cookieName)
    if err != nil {
        return "", err
    }
    return cookie.Value, nil
}

然后使用的时候就可以按需检查错误，比如：

cliente, err := getCookieValue(r, "codicecliente")
if err != nil {
    // 处理Cookie不存在的情况，比如返回错误响应
    http.Error(w, "codicecliente cookie required", http.StatusBadRequest)
    return
}

这样既保持了代码的简洁性，又能灵活应对不同的业务需求~

内容的提问来源于stack exchange，提问作者PythonFun