go tcp编程案例-服务端循环接收客户端消息2

服务端

package main

import (
	"fmt"
	"net"
)

func main() {
	fmt.Println("服务器开始监听...")
	listen, err := net.Listen("tcp", "0.0.0.0:65530")
	if err != nil {
		fmt.Println("listen err=", err)
		return
	}
	defer listen.Close() // 延时关闭listen

	// 循环等待客户端来链接
	for {
		// 等待客户端链接
		fmt.Println("等待客户端来链接...")
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("Accept() err=", err)
		} else {
			fmt.Printf("Accept() success conn=%v, 客户端ip=%v\n", 
			conn, conn.RemoteAddr().String())
		}
		// 这里启动一个协程，为客户端服务
		go process(conn)
	}
	// fmt.Printf("listen success=%v\n", listen)

}


func process(conn net.Conn) {
	// 这里循环的接收客户端发送的数据
	defer conn.Close() // 关闭连接

	for {
		// 创建一个新的切片
		buf := make([]byte, 1024)
		// fmt.Printf("服务器在等待客户端 %s 发送信息\n", conn.RemoteAddr().String())
		// 等待客户端通过conn发送信息，如果客户端没有write[发送]那么协程就堵塞在这里
		n, err := conn.Read(buf)  // 从conn读取
		if err != nil {
			fmt.Println("服务器的Read err=", err)
			return  // 
		}
		// 显示客户端发送的内容到服务器的终端
		fmt.Print(string(buf[:n]))
	}
}

客户端

package main

import (
	"bufio"
	"fmt"
	"net"
	"os"
	"strings"
)

func main() {

	conn, err := net.Dial("tcp", "127.0.0.1:65530")
	if err != nil {
		fmt.Println("client dial err=", err)
		return
	}
	defer conn.Close()
	// fmt.Println("conn 成功=", conn)
	
	// 客户端可以发送单行数据，然后退出
	reader := bufio.NewReader(os.Stdin)  // os.Stdin 代表标准输入[终端]
	// 从终端读取一行用户输入，并准备发送给服务器

	for {
		line, err := reader.ReadString('\n') 
		if err != nil {
			fmt.Println("readString err=", err)
		}
		// 如果用户输入的是 exit 就退出
		line = strings.Trim(line, " \r\n")
		if line == "exit" {
			fmt.Println("客户端退出...")
			break
		}

		// 再将line发送给服务器
		_, err = conn.Write([]byte(line + "\n"))
		if err!= nil {
			fmt.Println("conn.Write err=", err)
		}
	}

//	fmt.Printf("客户端发送了 %d 字节的数据，并退出", n)


}