【GO】TCP交替通信

服务端源码

package main

import (
	"fmt"
	"net"
	"os"
)

/**
处理异常
*/
func JtHandleError(err error, when string) {
	if err != nil {
		fmt.Println(when, "错误原因：", err)
		os.Exit(1)
	}
}
func main() {
	listener, err := net.Listen("tcp", "127.0.0.1:8888")
	JtHandleError(err, "net.Listen")
	fmt.Println("listener...")
	defer listener.Close()
	for {
		// 接入一个客户端
		conn, err := listener.Accept()
		JtHandleError(err, "listener.Accept")
		// 给每一个客户端开启一个线程
		go JtIoWithConn(conn)
	}
}

func JtIoWithConn(conn net.Conn) {
	defer conn.Close()
	// 创建缓存区
	buffer := make([]byte, 1024)
	// 这个获取的是客户端的地址
	clientAddr := conn.RemoteAddr()
	for {
		// 读入消息
		n, err := conn.Read(buffer)
		JtHandleError(err, "conn.Read")
		// 客户端消息
		clientMsg := string(buffer[:n])
		// 打印出客户端发过来的消息
		fmt.Println(clientAddr, clientMsg)
		// 服务端给客户端写的消息
		conn.Write([]byte("服务端以阅：" + clientMsg))
		if clientMsg == "over" {
			fmt.Printf("%v以下线\n", clientAddr)
			break
		}
	}

}

客户端源码

package main

import (
	"fmt"
	"net"
	"os"
)

/**
处理异常
*/
func JcHandleError(err error, when string) {
	if err != nil {
		fmt.Println(when, "错误原因：", err)
		os.Exit(1)
	}
}
func main() {
	// 拨号连接,得到专线conn
	conn, err := net.Dial("tcp", "127.0.0.1:8888")
	JcHandleError(err, "net.Dial")
	defer conn.Close()
	// 在专线中处理io
	ioInteractive(conn)
}

func ioInteractive(conn net.Conn) {
	defer conn.Close()
	// 在一条独立的协程里接收服务端的消息
	go func() {
		// 创建一个缓冲区
		buffer := make([]byte, 1024)
		// 循环读取
		for {
			// 从缓冲区读取消息
			n, err := conn.Read(buffer)
			JcHandleError(err, "conn.Read")
			serverMsg := string(buffer[:n])
			fmt.Println("服务端消息:", serverMsg)
		}
	}()

	// 在主协程里向服务端写消息
	var userInput string
	for {
		// 标准输入与输出
		fmt.Scan(&userInput)
		// 向服务端写出
		conn.Write([]byte(userInput))
		if userInput == "exit" {
			return
		}
	}
}

执行

服务端跑一个，客户端跑俩个即可

博主微信欢迎交流