golang tcp库实现-Golang-PHP中文网

golang tcp库实现

WBOY

发布： 2023-05-10 13:03:36

原创

594 人浏览过

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 main

import (
    "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 main

import (
    "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函数来读取服务器返回的数据。如果没有发生错误，则打印出服务器返回的数据。