[agent-go] 开始使用GO重构Agent部分

server-go/main.go

@@ -0,0 +1,125 @@
+package main
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin/binding"
+	"github.com/go-playground/validator/v10"
+	"net/http"
+	"server-go/router"
+	"server-go/service"
+)
+
+func myValidate(fl validator.FieldLevel) bool {
+	fmt.Println(fl.Field().Interface().(int))
+	return true
+}
+
+func main() {
+
+	// init the db connection
+	service.InitDB()
+
+	// all the route
+	engine := router.Routers()
+
+	// Get方法
+	engine.GET("/ping", func(c *gin.Context) {
+
+		statusOK := http.StatusOK
+		c.JSON(
+			statusOK,
+			gin.H{})
+	})
+
+	// Post方法，接收request body的形式
+	engine.POST("/jsonPost", func(c *gin.Context) {
+
+		// 自定义 请求规则验证器
+		if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
+			v.RegisterValidation("myValidation", myValidate)
+		}
+
+		var p string
+		err := c.ShouldBindJSON(&p)
+		if err != nil {
+			return
+		}
+	})
+
+	// 上传文件的操作 前端上传文件使用的是form-data的形式
+	// 单个文件上传
+	engine.POST("/upload", func(c *gin.Context) {
+		file, err := c.FormFile("file") // 获取上传的文件
+		if err != nil {
+			c.String(400, fmt.Sprintf("get form err: %s", err.Error()))
+			return
+		}
+		// 如果有其他的表单信息
+		c.DefaultPostForm("name", "default-file-name")
+
+		// 打印文件信息
+		fmt.Printf("File Name: %s\n", file.Filename)
+		fmt.Printf("File Size: %d bytes\n", file.Size)
+		fmt.Printf("MIME Type: %s\n", file.Header.Get("Content-Type"))
+
+		// 保存文件到本地
+		err = c.SaveUploadedFile(file, "./"+file.Filename)
+		if err != nil {
+			c.String(500, fmt.Sprintf("upload file err: %s", err.Error()))
+			return
+		}
+
+		c.String(200, fmt.Sprintf("%s uploaded successfully!", file.Filename))
+	})
+
+	// 多个文件上传
+	engine.POST("/multi-upload", func(c *gin.Context) {
+		form, err := c.MultipartForm() // 获取多个文件上传表单
+		if err != nil {
+			c.String(400, fmt.Sprintf("get form err: %s", err.Error()))
+			return
+		}
+
+		// 遍历所有上传的文件
+		files := form.File["files"]
+		for _, file := range files {
+			// 打印文件信息
+			fmt.Printf("File Name: %s\n", file.Filename)
+			fmt.Printf("File Size: %d bytes\n", file.Size)
+			fmt.Printf("MIME Type: %s\n", file.Header.Get("Content-Type"))
+
+			// 保存文件到本地
+			err = c.SaveUploadedFile(file, "./"+file.Filename)
+			if err != nil {
+				c.String(500, fmt.Sprintf("upload file err: %s", err.Error()))
+				return
+			}
+		}
+
+		c.String(200, fmt.Sprintf("%d files uploaded successfully!", len(files)))
+	})
+
+	// 返回静态文件
+	engine.GET("/file", func(c *gin.Context) {
+		c.File("./example.txt") // 返回指定路径的文件
+	})
+
+	// 返回动态生成的文件
+	engine.GET("/download", func(c *gin.Context) {
+		content := "Hello, world!"
+		fileName := "example.txt"
+
+		// 设置响应头，使浏览器可以下载文件
+		c.Header("Content-Disposition", "attachment; filename="+fileName)
+		c.Header("Content-Type", "application/octet-stream")
+		c.Header("Content-Transfer-Encoding", "binary")
+		c.Header("Cache-Control", "no-cache")
+
+		// 将内容写入响应体中
+		c.String(http.StatusOK, content)
+	})
+
+	engine.Run(":9090")
+
+}