This commit is contained in:
suyl
2021-06-15 17:23:55 +08:00
parent b9bed1d6d8
commit 2c64ff131b
2 changed files with 64 additions and 31 deletions

41
main.go
View File

@@ -1,8 +1,11 @@
package main
import (
"bufio"
"flag"
"fmt"
"io"
"net"
"net/http"
_ "net/http/pprof"
"os"
@@ -143,6 +146,44 @@ func main() {
http.ListenAndServe("0.0.0.0:6060", nil)
}
}()
l, err := net.Listen("tcp", ":8000")
if err != nil {
fmt.Println("listen error:", err)
return
}
for {
c, err := l.Accept()
if err != nil {
fmt.Println("accept error:", err)
break
}
globals.SugarLogger.Debugf("tcp socket:", utils.Format4Output(c, true))
go handleConn(c)
}
beego.Run()
}
}
func handleConn(c net.Conn) {
if c == nil {
fmt.Println("conn is nil")
return
}
defer c.Close()
reader := bufio.NewReader(c)
for {
//ReadString阻塞直到读到\n
msg, err := reader.ReadString('\n')
if err != nil {
if err == io.EOF {
fmt.Println("connection close")
} else {
fmt.Println("ReadString err", err)
}
return
}
fmt.Println("read data", msg)
b := []byte("ok")
c.Write(b)
}
}