golang tcp库实现

时间：2023-05-10 17:44

Go语言是一门越来越受欢迎的编程语言，它以其高效、简洁的代码风格、天然的并发性和并发编程模型成为了众人关注的焦点。而针对TCP协议的通信，Go语言提供了一些内置的库来帮助我们实现TCP通信，例如net、net/http等。

这篇文章将从实现的层面上来讲解如何利用Go语言的socket库来实现TCP通信。

TCP协议简介

TCP是传输控制协议，它是一种面向连接的、可靠的、基于字节流的传输协议。在TCP协议中，数据在发送之前需要进行三次握手，保证数据的正确传输。而在数据传输过程中，TCP还提供了一些数据校验、重传等机制来保证数据的可靠性。因此，TCP也被称为可靠传输协议。

Go语言的socket库概述

Go语言提供了原生的 socket 库，主要包括net、net/http、net/rpc三个包。其中，net包提供了最基础的网络通信操作接口，包括IP地址、TCP/UDP协议等；net/http包则提供了HTTP协议相关的网络操作接口；net/rpc则是一个远程过程调用框架。

在net包中，有两个主要的类型：Conn和Listener。其中Conn表示一个面向连接的通信，客户端、服务器端都可以有一个Conn对象，Conn可以实现两台计算机之间的通信;Listener则表示一个网络端口的监听器，可以用来监听并处理客户端的链接请求，返回Conn对象以实现通信。

使用socket库实现TCP通信

在了解了TCP协议和Go语言的socket库后，我们就可以开始使用Go语言的socket库来实现TCP通信了。下面通过一个简单的示例来讲解如何使用socket库实现TCP通信。

服务器端

我们首先来看一下如何实现一个简单的TCP服务器端程序。示例代码如下：

package mainimport (    "fmt"    "net")func main() {    listener, err := net.Listen("tcp", "127.0.0.1:8888")    if err != nil {        fmt.Println("Error: ", err)        return    }    for {        conn, err := listener.Accept()        if err != nil {            fmt.Println("Error: ", err)            continue        }        go handleConn(conn)    }}func handleConn(conn net.Conn) {    defer conn.Close()    buf := make([]byte, 1024)    for {        n, err := conn.Read(buf)        if err != nil {            fmt.Println("Error: ", err)            return        }        fmt.Println("Receive: ", string(buf[:n]))        _, err = conn.Write(buf[:n])        if err != nil {            fmt.Println("Error: ", err)            return        }    }}

在这个示例程序中，我们使用了net包的Listen函数监听了本地的8888端口，并在主函数中使用for循环来不断地等待客户端的链接请求。一旦得到链接请求，就会创建一个新的协程来处理链接请求，从而实现对多个客户端的并发处理。

在handleConn函数中，我们首先使用defer保证程序执行完毕后会关闭当前的链接，然后使用Read函数从客户端读取数据，使用Write函数向客户端发送数据，实现了简单的回显功能。

客户端

接下来，我们来看一下如何实现一个简单的TCP客户端程序。示例代码如下：

package mainimport (    "fmt"    "net")func main() {    conn, err := net.Dial("tcp", "127.0.0.1:8888")    if err != nil {        fmt.Println("Error: ", err)        return    }    defer conn.Close()    _, err = conn.Write([]byte("Hello World!"))    if err != nil {        fmt.Println("Error: ", err)        return    }    buf := make([]byte, 1024)    n, err := conn.Read(buf)    if err != nil {        fmt.Println("Error: ", err)        return    }    fmt.Println("Receive: ", string(buf[:n]))}

在这个示例程序中，我们使用了net包的Dial函数来链接服务器端，向服务器发送了一段数据，并使用Read函数来读取服务器返回的数据。如果没有发生错误，则打印出服务器返回的数据。