[网络协议]golang---网络编程--TCP

服务端

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]))
	}
}