网络编程（TCP/UDP通信）

TCP通信

TCP server端程序的处理流程：

1. 监听端口
2. 接收客户端请求建立链接
3. 创建goroutine处理链接。

代码示例：

package main

import (
	"bufio"
	"fmt"
	"net"
)

// 处理函数
func process(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("read from client failed, err: ", err)
			break
		}

		recvStr := string(buf[:n])
		fmt.Println("收到 client 端发来的数据: ", recvStr)

		// 发送数据
		conn.Write([]byte(recvStr))
	}
}

// 主函数
func main() {
	// listen 监听
	listen, err := net.Listen("tcp", "127.0.0.1:10010")
	if err != nil {
		fmt.Println("listen failed, err: ", err)
		return
	}

	// 无限循环，在里面一直保持读取状态
	for {
		// 建立连接
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("accept failed, err: ", err)
			continue
		}
		go process(conn)
	}

}

TCP client端进行TCP通信的流程：

1.建立与服务端的链接
2.进行数据收发
3.关闭链接

代码示例

package main

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

func main() {
	conn, err := net.Dial("tcp", "127.0.0.1:10010")
	if err != nil {
		fmt.Println("err: ", err)
		return
	}

	// 关闭连接
	defer conn.Close()

	// 从 os.Stdin里读取数据信息
	inputReader := bufio.NewReader(os.Stdin)

	for {
		// ReadString读取用户输入，直到读到 '\n'。出错的话返回 err，这里没写
		input, _ := inputReader.ReadString('\n')

		// 返回一个string切片，扇区input字符前后的  "\r\n"
		inputInfo := strings.Trim(input, "\r\n")

		// 输入 Q 就退出
		if strings.ToUpper(inputInfo) == "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]))
	}
}

UDP通信

• UDP协议（User Datagram Protocol）中文名称是用户数据报协议，是OSI（Open System Interconnection，开放式系统互联）参考模型中一种无连接的传输层协议，不需要建立连接就能直接进行数据发送和接收，属于不可靠的、没有时序的通信，但是UDP协议的实时性比较好，通常用于视频直播相关领域。

UDP服务端

/*
	UDP 服务端 server
*/

package main

import (
	"fmt"
	"net"
)

func main() {
	listen, err := net.ListenUDP("udp", &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 30000,
	})
	if err != nil {
		fmt.Println("listen failedm err: ", err)
		return
	}

	defer listen.Close()

	// 持续读取数据
	for {
		var data [1024]byte
		n, addr, err := listen.ReadFromUDP(data[:]) // 接收数据
		if err != nil {
			fmt.Println("read udp failed, err: ", err)
			continue
		}

		fmt.Printf("data: %v\taddr: %v\tcount: %v\n", string(data[:n]), addr, n)
		_, err = listen.WriteToUDP(data[:], addr) // 发送数据
		if err != nil {
			fmt.Println("write to udp failed, err: ", err)
			continue
		}
	}
}

UDP客户端

参考TCP lient对代码改了下，可以一直会话（直到用户输入”Q“退出）

package main

import (
	"fmt"
	"net"
	"strings"
)

func main() {
	socket, err := net.DialUDP("udp", nil, &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 30000,
	})
	if err != nil {
		fmt.Println("连接服务器失败, err: ", err)
		return
	}

	defer socket.Close()

	// 我在这里改一下，让它能够一直通信，直到输入 "Q"才退出
	for {
		//sendData := []byte("Hello Server!")
		// 这里改了一下，改成自定义输入
		var infomations string
		fmt.Scan(&infomations)
		sendData := strings.Trim(infomations, "\r\n")
		if strings.ToUpper(sendData) == "Q" {
			fmt.Println("会话结束")
			return
		}

		_, err = socket.Write([]byte(sendData)) // 发送数据
		if err != nil {
			fmt.Println("发送数据失败, err: ", err)
			return
		}

		data := make([]byte, 4096)
		n, remoteAddr, err := socket.ReadFromUDP(data) // 接收数据
		if err != nil {
			fmt.Println("接收数据失败, err: ", err)
			return
		}

		fmt.Printf("recv: %v \taddr: %v \t count: %v\n", string(data[:n]), remoteAddr, n)
	}

}