go语言TCP学习

TCP Server端

流程：
1.监听端口
2.接收客户端请求建立连接
3.创建goroutine处理连接

func process(conn net.Conn) { //net.conn可以在多个线程中同时安全读写
	defer conn.Close() //关闭连接
	for ; ; {
		reader := bufio.NewReader(conn)
		var buf [128]byte
		n, err := reader.Read(buf[:]) //读取数据
		if err != nil {
			fmt.Println("连接客户端失败，错误信息：", err)
		}
		recvStr := string(buf[:n])
		fmt.Println("收到客户端信息： ",recvStr)
		conn.Write([]byte(recvStr))  //发送数据
	}
}

func main() {
	listen,err:=net.Listen("tcp","127.0.0.1:8089")
	if err!=nil {
		fmt.Println("监听失败，错误",err)
		return
	}
	for ; ;  {
		conn,err:=listen.Accept()  //建立连接
		if err!=nil {
			fmt.Println("建立连接失败，错误： ",err)
			continue
		}
		go process(conn)      //启动一个goroutine处理连接
	}
}

TCP客户端

**流程：**
1、建立与服务端的连接
2、进行数据收发
3、关闭连接

func main() {
	conn, err := net.Dial("tcp", "127.0.0.1:8089")
	if err != nil {
		fmt.Println("连接失败，错误： ", 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" {
			return //如果输入q就推出
		}

		_, err = conn.Write([]byte(inputInfo)) //发送数据
		if err != nil {
			return
		}
		buf := [512]byte{}
		n, err := conn.Read(buf[:])
		if err != nil {
			fmt.Println("接收失败，错误：", err)
			return
		}
		fmt.Println(string(buf[:n]))
	}
}