【Go语言实战】简简单单的几十行代码实现 TCP 通信

1,457 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

基于net包的小应用

完整代码已经上传到github GitHub-TCP

欢迎starissue

请添加图片描述

首先创建两个目录,一个是client客户端,另一个是server服务端。 在这里插入图片描述

1. 连接

1.1 服务端

  • 监听连接

net 中提供了Listen方法,可以让服务端进行端口监听

ADDRESS := "127.0.0.1:5000"
listener,err := net.Listen("tcp",ADDRESS)
if err != nil {
	fmt.Printf("start tcp server %s failed ,err : %s ",listener,err)
	return
}
defer listener.Close()

1.2 客户端

  • 建立连接

net中提供了Dail方法,让客户端连接服务端

ADDRESS := "127.0.0.1:5000"
conn,err := net.Dial("tcp",ADDRESS) // 主动与服务端建立连接
if err != nil {
	fmt.Printf("dial %s failed; err :%s",ADDRESS,err)
	return
}

2. 通信

2.1 服务端

  • 接受信息

可以通过.Read来读取传输的数据。

	var data [1024]byte
	var msg string
	reader := bufio.NewReader(os.Stdin)
	for {   // 服务端要时刻等待传送过来的数据,所以要用for循环
		//接受信息
		n,err := conn.Read(data[:])
		if err == io.EOF{
			break
		}
		if err != nil {
			fmt.Printf("read from conn failed,err:%s",err)
			return
		}
		fmt.Println("Access Info : ",string(data[:n]))
	}
	defer conn.Close()

2.2 客户端

  • 发送信息

同样可以通过.Write在传输连接中传输数据。

	for{ // 让客户的能一直发送信息,所以就需要一个for循环,保持连接
		fmt.Print("请输入:")
		msg,_ = reader.ReadString('\n')
		msg = strings.TrimSpace(msg)
		if msg == "exit" {
			break
		}
		_, _ = conn.Write([]byte(msg))
	}

3. 回复

当服务端收到信息之后,应该返回信息给客户端。表示已经收到了数据。

3.1 服务端

服务端回复信息

	//回复信息
	fmt.Print("回复信息:")
	msg,_ = reader.ReadString('\n')
	msg = strings.TrimSpace(msg)
	if msg == "exit" {
		break
	}
	_ ,_ = conn.Write([]byte(msg))

3.2 客户端

客户端收到信息

	// 接受信息
	n,err:=conn.Read(data[:])
	if err == io.EOF {
		break
	}
	if err != nil {
		fmt.Println("read from conn failed, err :",err)
		return
	}
	fmt.Println("收到的回复:",string(data[:n]))

最后

小生凡一,期待你的关注。