Skip to content

Latest commit

 

History

History
204 lines (172 loc) · 4.52 KB

17.md

File metadata and controls

204 lines (172 loc) · 4.52 KB

第17节 golang中的socket通信

❤️💕💕计算机网络--TCP/IP 学习。Myblog:http://nsddd.top


[TOC]

TCP 实现

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

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

server 端

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.Printf("read from conn failed, err:%v\n", err)
            break
        }

        recv := string(buf[:n])
        fmt.Printf("收到的数据:%v\n", recv)

        // 将接受到的数据返回给客户端
        _, err = conn.Write([]byte("ok"))
        if err != nil {
            fmt.Printf("write from conn failed, err:%v\n", err)
            break
        }
    }
}

func main() {
    // 建立 tcp 服务
    listen, err := net.Listen("tcp", "127.0.0.1:9090")
    if err != nil {
        fmt.Printf("listen failed, err:%v\n", err)
        return
    }

    for {
        // 等待客户端建立连接
        conn, err := listen.Accept()
        if err != nil {
            fmt.Printf("accept failed, err:%v\n", err)
            continue
        }
        // 启动一个单独的 goroutine 去处理连接
        go process(conn)
    }
}

client 端

package main

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

func main() {
    // 1、与服务端建立连接
    conn, err := net.Dial("tcp", "127.0.0.1:9090")
    if err != nil {
        fmt.Printf("conn server failed, err:%v\n", err)
        return
    }
    // 2、使用 conn 连接进行数据的发送和接收
    input := bufio.NewReader(os.Stdin)
    for {
        s, _ := input.ReadString('\n')
        s = strings.TrimSpace(s)
        if strings.ToUpper(s) == "Q" {
            return
        }

        _, err = conn.Write([]byte(s))
        if err != nil {
            fmt.Printf("send failed, err:%v\n", err)
            return
        }
        // 从服务端接收回复消息
        var buf [1024]byte
        n, err := conn.Read(buf[:])
        if err != nil {
            fmt.Printf("read failed:%v\n", err)
            return
        }
        fmt.Printf("收到服务端回复:%v\n", string(buf[:n]))
    }
}

UDP 实现

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

server 端

package main

import (
    "fmt"
    "net"
)

func main() {
    // 建立 udp 服务器
    listen, err := net.ListenUDP("udp", &net.UDPAddr{
        IP:   net.IPv4(0, 0, 0, 0),
        Port: 9090,
    })
    if err != nil {
        fmt.Printf("listen failed error:%v\n", err)
        return
    }
    defer listen.Close() // 使用完关闭服务

    for {
        // 接收数据
        var data [1024]byte
        n, addr, err := listen.ReadFromUDP(data[:])
        if err != nil {
            fmt.Printf("read data error:%v\n", err)
            return
        }
        fmt.Printf("addr:%v\t count:%v\t data:%v\n", addr, n, string(data[:n]))
        // 发送数据
        _, err = listen.WriteToUDP(data[:n], addr)
        if err != nil {
            fmt.Printf("send data error:%v\n", err)
            return
        }
    }
}

client 端

package main

import (
    "fmt"
    "net"
)

func main() {
    // 建立服务
    listen, err := net.DialUDP("udp", nil, &net.UDPAddr{
        IP:   net.IPv4(0, 0, 0, 0),
        Port: 9090,
    })
    if err != nil {
        fmt.Printf("listen udp server error:%v\n", err)
    }
    defer listen.Close()

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

END 链接