feat: 更新代码以支持使用Gin框架处理HTTP请求,重构加载新功能方式，使用反射动态加载，修改之前硬编码

.gitignore

@@ -1,6 +1,7 @@
 data.db
-test.json
+test*.json
 *.exe
 request.json
 config.toml
 messages.json
+.vscode/settings.json

go.mod

@@ -1,6 +1,6 @@
 module go-bot
 
-go 1.22.2
+go 1.22.5
 
 require github.com/BurntSushi/toml v1.3.2
 
@@ -10,8 +10,36 @@ require (
 	github.com/parnurzeal/gorequest v0.3.0
 )
 
+require (
+	github.com/bytedance/sonic v1.11.6 // indirect
+	github.com/bytedance/sonic/loader v0.1.1 // indirect
+	github.com/cloudwego/base64x v0.1.4 // indirect
+	github.com/cloudwego/iasm v0.2.0 // indirect
+	github.com/gabriel-vasile/mimetype v1.4.3 // indirect
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.14.1 // indirect
+	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-playground/validator/v10 v10.20.0 // indirect
+	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.7 // indirect
+	github.com/leodido/go-urn v1.4.0 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
+	github.com/ugorji/go/codec v1.2.12 // indirect
+	golang.org/x/arch v0.8.0 // indirect
+	golang.org/x/crypto v0.24.0 // indirect
+	golang.org/x/sys v0.21.0 // indirect
+	golang.org/x/text v0.16.0 // indirect
+	google.golang.org/protobuf v1.34.1 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)
+
 require (
 	github.com/elazarl/goproxy v0.0.0-20240618083138-03be62527ccb // indirect
+	github.com/gin-gonic/gin v1.10.0
 	github.com/moul/http2curl v1.0.0 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/smartystreets/goconvey v1.8.1 // indirect

go.sum

@@ -1,26 +1,101 @@
 github.com/BurntSushi/toml v1.3.2 h1:o7IhLm0Msx3BaB+n3Ag7L8EVlByGnpq14C4YWiu/gL8=
 github.com/BurntSushi/toml v1.3.2/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ=
+github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0=
+github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4=
+github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM=
+github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
+github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
+github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
+github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
+github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/elazarl/goproxy v0.0.0-20240618083138-03be62527ccb h1:2SoxRauy2IqekRMggrQk3yNI5X6omSnk6ugVbFywwXs=
 github.com/elazarl/goproxy v0.0.0-20240618083138-03be62527ccb/go.mod h1:Ro8st/ElPeALwNFlcTpWmkr6IoMFfkjXAvTHpevnDsM=
 github.com/elazarl/goproxy/ext v0.0.0-20190711103511-473e67f1d7d2/go.mod h1:gNh8nYJoAm43RfaxurUnxr+N1PwuFV3ZMl/efxlIlY8=
+github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0=
+github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=
+github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
+github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
+github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
+github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
+github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
+github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
+github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
+github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
+github.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8=
+github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
 github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
 github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
 github.com/gopherjs/gopherjs v1.17.2 h1:fQnZVsXk8uxXIStYb0N4bGk7jeyTalG/wsZjQ25dO0g=
 github.com/gopherjs/gopherjs v1.17.2/go.mod h1:pRRIvn/QzFLrKfvEz3qUuEhtE/zLCWfreZ6J5gM2i+k=
+github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
+github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
 github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=
 github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
+github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
+github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM=
+github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
+github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
+github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
+github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
 github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
+github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
 github.com/moul/http2curl v1.0.0 h1:dRMWoAtb+ePxMlLkrCbAqh4TlPHXvoGUSQ323/9Zahs=
 github.com/moul/http2curl v1.0.0/go.mod h1:8UbvGypXm98wA/IqH45anm5Y2Z6ep6O31QGOAZ3H0fQ=
 github.com/parnurzeal/gorequest v0.3.0 h1:SoFyqCDC9COr1xuS6VA8fC8RU7XyrJZN2ona1kEX7FI=
 github.com/parnurzeal/gorequest v0.3.0/go.mod h1:3Kh2QUMJoqw3icWAecsyzkpY7UzRfDhbRdTjtNwNiUE=
+github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
+github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/rogpeppe/go-charset v0.0.0-20180617210344-2471d30d28b4/go.mod h1:qgYeAmZ5ZIpBWTGllZSQnw97Dj+woV0toclVaRGI8pc=
 github.com/smarty/assertions v1.15.0 h1:cR//PqUBUiQRakZWqBiFFQ9wb8emQGDb0HeGdqGByCY=
 github.com/smarty/assertions v1.15.0/go.mod h1:yABtdzeQs6l1brC900WlRNwj6ZR55d7B+E8C6HtKdec=
 github.com/smartystreets/goconvey v1.8.1 h1:qGjIddxOk4grTu9JPOU31tVfq3cNdBlNa5sSznIX1xY=
 github.com/smartystreets/goconvey v1.8.1/go.mod h1:+/u4qLyY6x1jReYOp7GOM2FSt8aP9CzCZL03bI28W60=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
+github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
+github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
+github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
+golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
+golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
+golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
+golang.org/x/crypto v0.24.0 h1:mnl8DM0o513X8fdIkmyFE/5hTYxbwYOjDS/+rK6qpRI=
+golang.org/x/crypto v0.24.0/go.mod h1:Z1PMYSOR5nyMcyAVAIQSKCDwalqy85Aqn1x3Ws4L5DM=
 golang.org/x/net v0.26.0 h1:soB7SVo0PWrY4vPW/+ay0jKDNScG2X9wFeYlXIvJsOQ=
 golang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE=
+golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.21.0 h1:rF+pYz3DAGSQAxAu1CbC7catZg4ebC4UIeIhKxBZvws=
+golang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/text v0.16.0 h1:a94ExnEXNtEwYLGJSIUxnWoxoRz/ZcCsV63ROupILh4=
+golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=
+google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
+google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
+rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=

main.go

@@ -10,24 +10,15 @@ import (
 	"net/http"
 	"time"
 
+	"github.com/gin-gonic/gin"
 	"github.com/goccy/go-json"
-
 	_ "github.com/mattn/go-sqlite3"
 )
 
 func insertMessage(db *sql.DB, data map[string]interface{}) error {
-
-	// Parse JSON data
-	// var data map[string]interface{}
-	// err := json.Unmarshal(jsonData, &data)
-	// if err != nil {
-	// 	log.Fatal(err)
-	// }
 	if data["post_type"] == "message" || data["post_type"] == "message_sent" {
-		// id := data["message_id"].(float64)
 		post_type := data["post_type"].(string)
 		message_type := data["message_type"].(string)
-		// 设置东八区时间
 		loc, err := time.LoadLocation("Asia/Shanghai")
 		if err != nil {
 			log.Fatal(err)
@@ -47,7 +38,6 @@ func insertMessage(db *sql.DB, data map[string]interface{}) error {
 		message_seq := data["message_seq"].(float64)
 
 		fmt.Println(post_type, message_time, int64(group_id), int64(message_id), raw_message, sender_user_id, sender_nickname, sender_card, sender_role, int64(message_seq))
-		// Insert data into database
 		_, err = db.Exec("INSERT INTO messages ( post_type, message_type, time, group_id, message_id, raw_message, sender_user_id, sender_nickname, sender_card, sender_role, message_seq) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
 			post_type, message_type, message_time, group_id, message_id, raw_message, sender_user_id, sender_nickname, sender_card, sender_role, message_seq)
 		fmt.Println("Data inserted successfully!")
@@ -58,33 +48,34 @@ func insertMessage(db *sql.DB, data map[string]interface{}) error {
 	return nil
 }
 
-func handlePost(w http.ResponseWriter, r *http.Request) {
+// handlePost 处理 POST 请求
-	if r.Method != http.MethodPost {
+func handlePost(c *gin.Context) {
-		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+	// 从请求体中读取数据
+	body, err := io.ReadAll(c.Request.Body)
+	if err != nil {
+		// 请求体数据读取失败
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Error reading request body"})
 		return
 	}
 
-	body, err := io.ReadAll(r.Body)
+	// 将 JSON 数据解析为 map 类型
-	if err != nil {
-		http.Error(w, "Error reading request body", http.StatusBadRequest)
-		return
-	}
-	// fmt.Println("handlePost:", string(body))
-	// 解码 JSON 数据到 map[string]interface{} 类型的变量
 	var data map[string]interface{}
 	err = json.Unmarshal(body, &data)
 	if err != nil {
-		fmt.Println("Error decoding JSON:", err)
+		// JSON 数据解析失败
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Error decoding JSON"})
 		return
 	}
 
-	// Open SQLite database
+	// 打开数据库
 	db, err := sql.Open("sqlite3", "./data.db")
 	if err != nil {
+		// 数据库打开失败
 		log.Fatal(err)
 	}
 	defer db.Close()
-	// Create table
+
+	// 创建表
 	_, err = db.Exec(`CREATE TABLE IF NOT EXISTS messages (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
         post_type TEXT,
@@ -100,36 +91,34 @@ func handlePost(w http.ResponseWriter, r *http.Request) {
         message_seq INTEGER
     )`)
 	if err != nil {
+		// 表创建失败
 		log.Fatal(err)
 	}
-	// Insert message into database
+
+	// 插入数据
 	err = insertMessage(db, data)
 	if err != nil {
+		// 数据插入失败
 		log.Fatal(err)
 	}
 
+	// 调用路由处理函数
 	utils.Router(data)
-	w.Write([]byte("JSON data received successfully!"))
+	c.JSON(http.StatusOK, gin.H{"message": "JSON data received successfully!"})
-
 }
 
 func main() {
-
 	cfg := config.GetConfig()
 	APIURL, ok := cfg["APIURL"].(string)
 	if !ok {
 		log.Fatal("加载配置失败！")
 	}
-	// config.PrintConfig(cfg, "")
-	// print(cfg["AllowGroup"].([]interface{})[0].(string))
 
-	http.HandleFunc("/", handlePost)
+	r := gin.Default()
-	// 协程支持
+	r.POST("/", handlePost)
-	// http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-	// 	go handlePost(w, r)
-	// })
 
-	fmt.Println("Server listening on port 5580...")
+	fmt.Println("Server listening on", APIURL, "...")
-	// APIURL 为 go-cqhttp 配置的事件上报地址
+	if err := r.Run(APIURL); err != nil {
-	http.ListenAndServe(APIURL, nil)
+		log.Fatal("服务器启动失败：", err)
+	}
 }

test/testimg.go

@@ -0,0 +1,43 @@
+package main
+
+import (
+	"encoding/base64"
+	"fmt"
+	"go-bot/utils"
+	"log"
+	"os"
+	"strings"
+)
+
+func main() {
+	img := utils.Image2Base64("/home/liyp/aa.jpg")
+	// 保存Base64字符串到文件
+	err := os.WriteFile("output_base64.txt", []byte(img), 0644)
+	if err != nil {
+		log.Fatalf("failed to write base64 string to file: %s", err)
+	}
+	// 将base64转回图片
+	// 去掉格式标志
+
+	parts := strings.Split(img, ",")
+	if len(parts) != 2 {
+		log.Fatalf("invalid data URI")
+	}
+	base64String := parts[1]
+
+	// 解码Base64字符串
+	fileContent, err := base64.StdEncoding.DecodeString(base64String)
+	if err != nil {
+		log.Fatalf("failed to decode base64 string: %s", err)
+	}
+
+	// 将解码后的内容保存为图片文件
+	err = os.WriteFile("output_image.jpg", fileContent, 0644)
+	if err != nil {
+		log.Fatalf("failed to write image file: %s", err)
+	}
+
+	fmt.Println("Image saved as output_image.jpg")
+
+	// println(img)
+}

utils/image2base64.go

@@ -0,0 +1,21 @@
+package utils
+
+import (
+	"encoding/base64"
+	"io"
+	"os"
+)
+
+func Image2Base64(path string) string {
+	file, err := os.Open(path)
+	if err != nil {
+		return ""
+	}
+	defer file.Close()
+
+	if data, err := io.ReadAll(file); err == nil {
+		return "data:image/jpeg;base64," + base64.StdEncoding.EncodeToString(data)
+	}
+	return ""
+
+}

utils/router.go

@@ -7,11 +7,6 @@ import (
 )
 
 func Router(data map[string]interface{}) {
-	// 将map转换为格式化的JSON字符串
-	// jsonString, err := json.MarshalIndent(data, "", "  ")
-	// if err != nil {
-	// 	panic(err)
-	// }
 
 	// 输出格式化后的JSON
 	// fmt.Println(string(jsonString))

workers/ai.go

@@ -10,6 +10,14 @@ import (
 	"github.com/parnurzeal/gorequest"
 )
 
+func init() {
+	RegisterWorkerFactory("ai", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &AI{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type AI struct {
 	*StdAns
 }

workers/core.go

@@ -1,13 +1,18 @@
 package workers
 
 import (
-	"bytes"
 	"encoding/json"
 	"fmt"
 	"go-bot/config"
-	"net/http"
+
+	"github.com/parnurzeal/gorequest"
 )
 
+type Worker interface {
+	CheckPermission() string
+	GetMsg() string
+	SendMsg(msg string) bool
+}
 type StdAns struct {
 	AllowGroup    []interface{}
 	AllowUser     []interface{}
@@ -26,11 +31,34 @@ type StdAns struct {
 	RawMsg        string
 }
 
+// 定义一个map类型的cfg，用于存储配置信息
 var cfg map[string]interface{}
 
-//	func init() {
+// 定义一个WorkerFactory类型的函数，用于创建Worker
-//		cfg = config.GetConfig()
+type WorkerFactory func(parms []string, uid, gid, role, mid, rawMsg string) Worker
-//	}
+
+// 定义一个workerFactories的map，用于存储WorkerFactory
+var workerFactories = make(map[string]WorkerFactory)
+
+// 定义一个RegisterWorkerFactory函数，用于注册WorkerFactory
+func RegisterWorkerFactory(name string, factory WorkerFactory) {
+	workerFactories[name] = factory
+	fmt.Printf("Register WorkerFactory: %s\n", name)
+}
+
+// 定义一个NewWorker函数，用于创建Worker
+func NewWorker(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+	fmt.Printf("NewWorker: %s, %s, %s, %s, %s\n", parms, uid, gid, role, mid)
+	if factory, ok := workerFactories[parms[0]]; ok {
+
+		fmt.Printf("Use WorkerFactory: %s\n", parms[0])
+		return factory(parms, uid, gid, role, mid, rawMsg)
+	}
+	return &Emm{
+		StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+	}
+}
+
 func NewStdAns(parms []string, uid, gid, role, mid, rawMsg string) *StdAns {
 	// var cfg map[string]interface{}
 
@@ -114,13 +142,17 @@ func (s *StdAns) SendMsg(msg string) bool {
 	// println("core:", url)
 	// fmt.Println("请求地址：", url)
 	fmt.Println("响应信息：\n", msg)
-	resp, err := http.Post(url, "application/json", bytes.NewBuffer(requestBody))
+	request := gorequest.New()
-	if err != nil {
+	resp, body, errs := request.Post(url).
+		Set("Content-Type", "application/json").
+		Send(string(requestBody)).End()
+	if len(errs) > 0 {
+
 		fmt.Println("Error sending request:", err)
 		return false
 	}
 	defer resp.Body.Close()
 	// res, _ := io.ReadAll(resp.Body)
-	// fmt.Println("响应返回：", string(res))
+	fmt.Println("响应返回：", body)
 	return true
 }

workers/emm.go

@@ -1,5 +1,13 @@
 package workers
 
+func init() {
+	RegisterWorkerFactory("emm", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &Emm{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type Emm struct {
 	*StdAns
 }

workers/hhsh.go

@@ -9,6 +9,14 @@ import (
 	"github.com/goccy/go-json"
 )
 
+func init() {
+	RegisterWorkerFactory("hhsh", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &Hhsh{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type Hhsh struct {
 	*StdAns
 }

workers/ip.go

@@ -12,6 +12,14 @@ import (
 	"github.com/goccy/go-json"
 )
 
+func init() {
+	RegisterWorkerFactory("ip", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &Ip{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type Ip struct {
 	*StdAns
 }
@@ -36,16 +44,7 @@ func (a *Ip) GetMsg() string {
 		req, _ := http.NewRequest("POST", url, bytes.NewBuffer(json_data))
 		req.Header.Set("Content-Type", "application/json")
 		req.Header.Set("x-api-key", "d84eae04-0654-4173-bc6a-c457c4fc3f4d")
-		// 打印请求信息
+
-		// fmt.Println("请求方法:", req.Method)
-		// fmt.Println("请求URL:", req.URL.String())
-		// fmt.Println("请求头:")
-		// for key, values := range req.Header {
-		// 	for _, value := range values {
-		// 		fmt.Printf("  %s: %s\n", key, value)
-		// 	}
-		// }
-		// fmt.Println("请求体:", string(json_data))
 		client := &http.Client{
 			Timeout: 5 * time.Second, // 设置超时时间为5秒
 		}

workers/lsp.go

@@ -9,6 +9,14 @@ import (
 	"net/http"
 )
 
+func init() {
+	RegisterWorkerFactory("lsp", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &Lsp{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type Lsp struct {
 	*StdAns
 }

workers/newworker.go

@@ -1,38 +0,0 @@
-package workers
-
-import "fmt"
-
-func NewWorker(parms []string, uid, gid, role, mid, rawMsg string) Worker {
-	fmt.Println("NewWorker:", parms)
-	switch parms[0] {
-
-	case "ping":
-		return &Ping{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
-		}
-		// 在这里添加其他 worker 的 case
-	case "pkg":
-		return &Pkg{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
-		}
-	case "hhsh":
-		return &Hhsh{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
-		}
-	case "ip":
-		return &Ip{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
-		}
-	case "lsp":
-		return &Lsp{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
-		}
-	case "ai":
-		return &AI{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
-		}
-	default:
-		return &Emm{
-			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg)}
-	}
-}

workers/ping.go

@@ -1,5 +1,13 @@
 package workers
 
+func init() {
+	RegisterWorkerFactory("ping", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &Ping{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type Ping struct {
 	*StdAns
 }

workers/pkg.go

@@ -10,6 +10,14 @@ import (
 	"github.com/goccy/go-json"
 )
 
+func init() {
+	RegisterWorkerFactory("pkg", func(parms []string, uid, gid, role, mid, rawMsg string) Worker {
+		return &Pkg{
+			StdAns: NewStdAns(parms, uid, gid, role, mid, rawMsg),
+		}
+	})
+}
+
 type Pkg struct {
 	*StdAns
 }

workers/worker.go

@@ -1,8 +0,0 @@
-package workers
-
-// Worker 接口描述了不同类的共同行为
-type Worker interface {
-	CheckPermission() string
-	GetMsg() string
-	SendMsg(msg string) bool
-}