服务端
func TcpListen(address string) {
fmt.Println("tcp监听")
listen, err := net.Listen("tcp", address)
if err != nil {
fmt.Errorf("tcp%s监听失败=>%#v", address, err)
return
}
for {
conn, err := listen.Accept()
if err != nil {
fmt.Errorf("建立连接失败=>%#v", err)
continue
}
go netWorkHandler.Process(conn)//每次有新的连接建立就会开启一个协程来处理
}
}
func Process(conn net.Conn) {
fmt.Println("----获取到新消息--执行-")
defer conn.Close()
for {
reader := bufio.NewReader(conn)
var buf [1024]byte
n, err := reader.Read(buf[:])
if err != nil {
break
}
recvStr := string(buf[:n])
fmt.Println("收到客户端的数据:", recvStr)
conn.Write([]byte(recvStr)) //返回给客户端的数据
}
}
func main() {
go listener.TcpListen("0.0.0.0:20001")
select {}//阻塞main函数
}
客户端
可以再控制台输入来测试
import (
"bufio"
"fmt"
"net"
"os"
"strings"
)
func MakeTcpConn(address string) {
conn, err := net.Dial("tcp", address)
if err != nil {
fmt.Println("err :", err)
return
}
defer conn.Close() // 关闭连接
inputReader := bufio.NewReader(os.Stdin)
for {
input, _ := inputReader.ReadString('\n') // 读取用户输入
inputInfo := strings.Trim(input, "\r\n")
if strings.ToUpper(inputInfo) == "Q" { // 如果输入q就退出
return
}
_, err = conn.Write([]byte(inputInfo)) // 发送数据
if err != nil {
return
}
//接受来自服务端的响应数据
buf := [512]byte{}
n, err := conn.Read(buf[:])
if err != nil {
fmt.Println("recv failed, err:", err)
return
}
fmt.Println(string(buf[:n]))
}
}