1
0
forked from lxh/go-wxhelper
go-wxhelper/main.go
2023-10-11 09:21:07 +08:00

58 lines
1.2 KiB
Go

package main
import (
"bytes"
"go-wechat/client"
"go-wechat/config"
"go-wechat/handler"
"go-wechat/tasks"
"io"
"log"
"net"
)
func init() {
config.InitConfig()
client.InitMySQLClient()
tasks.InitTasks()
}
func process(conn net.Conn) {
// 处理完关闭连接
defer func() {
log.Printf("处理完成: -> %s", conn.RemoteAddr())
_ = conn.Close()
}()
var buf bytes.Buffer
if _, err := io.Copy(&buf, conn); err != nil {
log.Printf("[%s]返回数据失败,错误信息: %v", conn.RemoteAddr(), err)
}
log.Printf("[%s]数据长度: %d", conn.RemoteAddr(), buf.Len())
go handler.Parse(conn.RemoteAddr(), buf.Bytes())
// 将接受到的数据返回给客户端
if _, err := conn.Write([]byte("200 OK")); err != nil {
log.Printf("[%s]返回数据失败,错误信息: %v", conn.RemoteAddr(), err)
}
}
func main() {
// 建立 tcp 服务
listen, err := net.Listen("tcp", "0.0.0.0:19099")
if err != nil {
log.Printf("listen failed, err:%v", err)
return
}
for {
// 等待客户端建立连接
conn, err := listen.Accept()
if err != nil {
log.Printf("accept failed, err:%v", err)
continue
}
// 启动一个单独的 goroutine 去处理连接
go process(conn)
}
}