TCP编程示例

1、TCP协议

TCP/IP(Transmission Control Protocol/Internet Protocol) 即传输控制协议/网间协议，是一种面向连接（连接导向）的、可靠的、基于字节流的传输层（Transport layer）通信协议，因为是面向连接的协议，数据像水流一样传输，会存在黏包问题。

2、TCP服务端

一个TCP服务端可以同时连接很多个客户端。

例如：世界各地的用户使用自己电脑上的浏览器访问淘宝网。

为Go语言中创建多个goroutine实现并发非常方便和高效，所以我们可以每建立一次链接就创建一个goroutine去处理。

TCP服务端程序的处理流程：

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

go的实现代码如下：

func main(){
    // 开启一个 tcp 监听
    listen,err := net.Listen("tcp", ":6666")
    if err != nil {
        fmt.Println(err)
        return
    }

    for {
        // 建立连接，等待并将下一个连接返回给侦听器
        conn,err := listen.Accept() 
        if err != nil {
            fmt.Println("accept failed, err:", err)
            continue
        }
        // 启动一个goroutine处理连接
        go handleConnection(conn)
    }
}

// 处理连接信息
func handleConnection(conn net.Conn){
    defer conn.Close() // 关闭连接
    for {
        // 创建一个有缓冲的读取器 从连接 中读取流
        reader := bufio.NewReader(conn)
        fmt.Println(reader)
        var buf [512]byte
        // 根据buf大小 读取 数据流
        n, err := reader.Read(buf[:]) // 读取数据
        if err != nil {
            fmt.Println("read from client failed, err:\n", err)
            break
        }
        recvStr := string(buf[:n])
        fmt.Println("收到client端发来的数据：\n", recvStr)
        conn.Write([]byte(recvStr)) // 发送数据
    }
}

3、TCP客户端

一个TCP客户端进行TCP通信的流程如下：

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

使用Go语言的net包实现的TCP客户端代码如下：

func main(){
    conn,err := net.Dial("tcp", "127.0.0.1:6666")
    if err != nil {
        fmt.Println("连接服务端失败")
        return
    }
    defer conn.Close()

    inputReader := bufio.NewReader(os.Stdin)

    for{
        // 读取用户输入
        input, _ := inputReader.ReadString('\n')
        inputInfo := strings.Trim(input, "\r\n")
        // 如果输入exit就退出
        if strings.ToLower(inputInfo) == "exit" {
            return
        }
        // 发送数据
        _, err = conn.Write([]byte(inputInfo))
        if err != nil {
            fmt.Println("发送数据失败，Err:", err.Error())
            return
        }

        buf := [512]byte{}
        // 读取连接返回数据
        n, err := conn.Read(buf[:])
        if err != nil {
            fmt.Println("recv failed, err:", err)
            return
        }
        fmt.Println(string(buf[:n]))
    }
}

将上面的代码编译成client或client.exe可执行文件，

先启动server端再启动client端，在client端输入任意内容回车之后就能够在server端看到client端发送的数据，从而实现TCP通信。