使用 Socket 连接的 Go 客户端

一分耕耘，一分收获！既然都打开这篇《使用 Socket 连接的 Go 客户端》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

我是 go 新手。我有一个运行 fastapi python 的服务器。我已经在 127.0.0.1:8000/register 中运行了 websocket 请求。

@app.websocket("/register")
async def websocket_endpoint(websocket: websocket):
    await websocket.accept()
    while true:
        data = await websocket.receive_text()
        await websocket.send_text(f"message text was: {data}")

我想通过 go 创建一个作为客户端的应用程序。此应用程序使用套接字连接到此 url 并发送信息来创建帐户。但是当我尝试在谷歌中使用示例代码时：

package main

import "net"
import "fmt"
import "bufio"
import "os"
import "log"

func main() {
  // connect to server
  conn, err := net.Dial("tcp", "127.0.0.1:8000/register")
  if err != nil {
    log.Println(err)
}
  for { 
    // what to send?
    reader := bufio.NewReader(os.Stdin)
    fmt.Print("Text to send: ")
    text, err := reader.ReadString('\n')
    if err != nil {
        log.Println(err)
    }
    // send to server
    fmt.Fprintf(conn, text + "\n")
    // wait for reply
    message, err := bufio.NewReader(conn).ReadString('\n')
    if err != nil {
        log.Println(err)
    }
    fmt.Print("Message from server: "+message)
  }
}

它不起作用，因为 dial tcp:lookup tcp/8000/register:getaddrinfow:未找到指定的类

据我所知，它无法查找此链接，它仅在它是 ip 地址时进行查找。 如何创建对 url 的 socket 请求？

正确答案

使用 github.com/gorilla/websocket 示例：

package main

import (
    "flag"
    "log"
    "net/url"
    "os"
    "os/signal"
    "time"

    "github.com/gorilla/websocket"
)

var addr = flag.string("addr", "127.0.0.1:8080", "http service address")

func main() {
    flag.parse()
    log.setflags(0)

    interrupt := make(chan os.signal, 1)
    signal.notify(interrupt, os.interrupt)

    u := url.url{scheme: "ws", host: *addr, path: "/register"}
    log.printf("connecting to %s", u.string())

    c, _, err := websocket.defaultdialer.dial(u.string(), nil)
    if err != nil {
        log.fatal("dial:", err)
    }
    defer c.close()

    done := make(chan struct{})

    go func() {
        defer close(done)
        for {
            _, message, err := c.readmessage()
            if err != nil {
                log.println("read:", err)
                return
            }
            log.printf("recv: %s", message)
        }
    }()

    ticker := time.newticker(time.second)
    defer ticker.stop()

    for {
        select {
        case <-done:
            return
        case t := <-ticker.c:
            err := c.writemessage(websocket.textmessage, []byte(t.string()))
            if err != nil {
                log.println("write:", err)
                return
            }
        case <-interrupt:
            log.println("interrupt")

            // cleanly close the connection by sending a close message and then
            // waiting (with timeout) for the server to close the connection.
            err := c.writemessage(websocket.closemessage, websocket.formatclosemessage(websocket.closenormalclosure, ""))
            if err != nil {
                log.println("write close:", err)
                return
            }
            select {
            case <-done:
            case <-time.after(time.second):
            }
            return
        }
    }
}

或 nhooyr.io/websocket - 是一个最小且惯用的 go websocket 库 https://github.com/nhooyr/websocket

ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
defer cancel()

c, _, err := websocket.Dial(ctx, "ws://localhost:8080/register", nil)
if err != nil {
    // ...
}
defer c.Close(websocket.StatusInternalError, "the sky is falling")

以上就是《使用 Socket 连接的 Go 客户端》的详细内容，更多关于的资料请关注golang学习网公众号！