go socket编程

Dear 丶 2022-05-18 02:47 165阅读 0赞

### go的网络应用方面程序的编写都在net包里面, 比如net/http, net/rpc, net/mail, net/smtp 等, 今天我们主要来讲解如何使用net/socket 来编写一个socket 客户端和服务端 ###

> 使用net包来检查 ip 地址是否是合法的ip 地址

`我们可以使用net.ParseIP() 方法来检查, 如果该方法返回了nil 则说明ip 地址是错误的,比如 192.168.1.256 这种就不是合法的地址`

### 使用net/socket 编写socket server(TCP) ###

> 一般的编写socket 服务端都分为几个步骤 1. bind 2. listen 3. accept 4. recv/send data 5. close  
> bind 步骤使用 net.ResolveTCPAddr(), 来实现这个是TCP类型的socket, 这个方法接收两个参数1, ip 协议类型, 一般就是tcp4 和 tcp6, 该方法返回两个结果, \*TCPAddr, error  
> listen 步骤使用net.ListenTCP(),来实现,这个方法接收两个参数, 第一个是协议类型,我们这里是tcp的socket ,因此就写tcp, 第二个参数是 \*TCPAddr 也就是 ResolveTCPAddr()这个方法返回的

### 服务端代码 ###

package growth
    
    import (
        "fmt"
        "net"
        "sync"
    )
    
    // go 学习第十八天 学习go socket
    
    var waitGroup sync.WaitGroup
    
    func Day18Socket()  {
        // net 包的ParseIP 可以检查ip 地址格式是否正确
        checkValidIp := "192.168.1.255"
        result := net.ParseIP(checkValidIp)
        if result == nil {
            fmt.Println("ip 地址错误")
        }else {
            fmt.Println("ip 地址是", result)
        }
        addr := "127.0.0.1:8888"
        Server(addr)
    }
    
    func Server(listen string)  {
        bind, e5 := net.ResolveTCPAddr("tcp4", listen)
    
        if e5 != nil {
            fmt.Println("server ip 地址有误")
            return
        }
        listener, e6 := net.ListenTCP("tcp", bind)
        if e6 != nil {
            fmt.Println("绑定失败,", e6)
        }
    
        for {
            count := 1
            conn, e7 := listener.Accept()
            fmt.Println("accept", conn.RemoteAddr())
            if e7 != nil {
                continue
            }
    
            go handler(conn)
    
            count += 1
            waitGroup.Add(count)
            fmt.Println("wait ....")
        }
    
    
    }
    
    func handler(connect net.Conn)  {
        fmt.Println("处理请求,", connect.RemoteAddr())
        defer waitGroup.Done()
        defer connect.Close()
        var recv []byte
        data, err := connect.Read(recv)
        if err != nil {
            fmt.Println("服务端读取数据错误")
            return
        }
        fmt.Println("接收到客户端数据", string(data))
        connect.Write([]byte("hello client"))
    
    }

### 使用net/socket 编写socket client(TCP) ###

> client 使用 net.DialTCP()来建立个服务端的连接

### 客户端代码 ###

package growth
    
    import (
        "net"
        "fmt"
        "io/ioutil"
    )
    
    // go 学习第十八天 学习go socket
    
    
    func Client()  {
        // go 里面通过ResolveTCPAddr获取连接地址
        // net参数是"tcp4"、"tcp6"、"tcp"中的任意一个 , addr表示域名或者IP地址
        addr := "127.0.0.1:8888"
        server, err := net.ResolveTCPAddr("tcp4", addr)
    
        if err != nil {
            fmt.Println("ip 有误, ", err)
            return
        }
    
        conn, e2 := net.DialTCP("tcp", nil, server)
    
        if e2 != nil {
            fmt.Println("连接失败,", e2)
            return
        }else {
            // 往服务端发送数据
            _, e3 := conn.Write([]byte("hello world!"))
            if e3 != nil {
                fmt.Println("发送失败", e3)
                return
            }
            resp, e4 := ioutil.ReadAll(conn)
            if e4 != nil {
                fmt.Println("读取数据错误", e4)
                return
            }else {
                fmt.Println("收到响应", resp)
            }
    
            fmt.Println("end -=-=-=-=-=-=->")
    
        }
    
    }