8、代码结构优化-CSDN博客

本文链接：https://blog.csdn.net/fyx708711/article/details/131775018

本文展示了如何优化Go语言中的main函数，通过使用匿名函数处理HTTP请求，并改进JSON返回结构，利用结构体和omitempty属性减少无用数据传输。主要涉及参数解析、登录逻辑及错误处理。

摘要由CSDN通过智能技术生成

一、优化main函数的匿名函数(对函数结构进行优化)

package main
import (
	"net/http"
)
func userLogin(writer http.ResponseWriter, request *http.Request) {
	//数据库操作
	//逻辑处理
	//restapi json/xml 返回
	//1.获取前端传递的参数
	//mobile,passwd
	//如何获取参数
	//解析参数
	request.ParseForm()
	mobile := request.PostForm.Get("mobile")
	passwd := request.PostForm.Get("passwd")
	loginOk := false
	if mobile == "xxxxxxxxxxx" && passwd == "123456" {
		loginOk = true
	}
	str := `{"code":0,"data":{"id":1,"token":"test"},"msg":"登录成功"}`
	if !loginOk {
		//返回失败的JSON
		str = `{"code":-1,"data":"","msg":"密码不正确"}`
	}
	//设置header 为JSON 默认的text/html,所以特别指出返回的为application/json
	writer.Header().Set("Content-Type", "application/json")
	//设置200状态
	writer.WriteHeader(http.StatusOK)
	//输出
	writer.Write([]byte(str))
}
func main() {
	//把前端请求的格式和封装处理函数进行绑定的标签
	//绑定请求和处理函数
	http.HandleFunc("/user/login", userLogin)
	//启动web服务器
	http.ListenAndServe(":8080", nil)
}

二、优化返回Json代码

package main
import (
	"encoding/json"
	"log"
	"net/http"
)
func userLogin(writer http.ResponseWriter, request *http.Request) {
	//数据库操作
	//逻辑处理
	//restapi json/xml 返回
	//1.获取前端传递的参数
	//mobile,passwd
	//如何获取参数
	//解析参数
	request.ParseForm()
	mobile := request.PostForm.Get("mobile")
	passwd := request.PostForm.Get("passwd")
	loginOk := false
	if mobile == "xxxxxxxxxxx" && passwd == "123456" {
		loginOk = true
	}
	if loginOk {
		//{"id":1,"token":"xx"}
		data := make(map[string]interface{})
		data["id"] = 1
		data["token"] = "test"
		Resp(writer, 0, data, "登录成功")
	} else {
		Resp(writer, -1, nil, "密码不正确")
	}
}
type H struct {
	Code int
	Msg  string
	Data interface{}
}
func Resp(w http.ResponseWriter, code int, data interface{}, msg string) {
	//设置header 为JSON 默认的text/html,所以特别指出返回的为application/json
	w.Header().Set("Content-Type", "application/json")
	//设置200状态
	w.WriteHeader(http.StatusOK)
	//定义一个结构体
	h := H{
		Code: code,
		Msg:  msg,
		Data: data,
	}
	//将结构体转化成JSON字符串
	ret, err := json.Marshal(h)
	if err != nil {
		log.Println(err.Error())
	}
	//输出
	w.Write([]byte(ret))
}
func main() {
	//把前端请求的格式和封装处理函数进行绑定的标签
	//绑定请求和处理函数
	http.HandleFunc("/user/login", userLogin)
	//启动web服务器
	http.ListenAndServe(":8080", nil)
}