在Gin框架中如何利用Unix Domain Socket发送请求？

大家好，今天本人给大家带来文章《在Gin框架中如何利用Unix Domain Socket发送请求？》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我正在为鞭炮创建一个包装。

要在命令行上使用 firecracker 启动虚拟机，您必须将套接字文件传递给 firecracker 可执行文件。像这样的事情：

firecracker --api-sock /path/to/file.socket

然后从另一个终端，您可以向该服务器/套接字发出如下请求：

curl --unix-socket /tmp/firecracker.socket -xput 'http://localhost/actions' -d '{"action_type": "sendctrlaltdel"}'

我正在尝试从 gin 服务器内复制相同的内容。

我有一个端点执行第一项工作，即启动服务器。最小代码如下所示：

cmd := exec.command("firecracker", "--api-sock", "/path/to/file.socket")

err := cmd.start()

此端点启动服务器并侦听任何命令。问题是，我不知道如何使用套接字文件向该服务器发出 put 请求。我在网上找到了这个，但对我来说没有多大意义。

这是一个不使用任何套接字文件的起始代码。

func BootSource(c *gin.Context) {
    var body models.BootSource
    c.BindJSON(&body)
    bodyJson, _ := json.Marshal(body)

    // initialize http client
    client := &http.Client{}

    // set the HTTP method, url, and request body
    req, err := http.NewRequest(http.MethodPut, "http://localhost/boot-source", bytes.NewBuffer(bodyJson))
    if err != nil {
        panic(err)
    }

    // set the request header Content-Type for json
    _, err = client.Do(req)
    if err != nil {
        panic(err)
    }
}

如何使此 put 请求使用套接字文件？

另请注意，我正在使用 gin 框架。

正确答案

为此，您需要覆盖 http.client 使用的 transport 来配置一个函数，用于创建与套接字的连接：

client := http.client{
  transport: &http.transport{
    dialcontext: func(ctx context.context, _, _ string) (net.conn, error) {
      return net.dial("unix", "/path/to/socket") 
    },
  },
}

然后您可以使用该客户端，并且它发出的所有请求都将使用该连接。通常，对于通过套接字公开的 http 服务，请求中使用的主机并不重要，因此您可以使用对您有意义的任何值，例如

client.get("http://firecracker/some/api/path")

但是，当您尝试使用 firecracker api 时，为什么不直接使用他们的 sdk：https://github.com/firecracker-microvm/firecracker-go-sdk

这将为您处理连接的设置，并防止您需要手动处理所有请求。

扩展到此 answer，您可以通过克隆默认传输来保留 http 默认值

defaultTransport, ok := http.DefaultTransport.(*http.Transport)
    if !ok {
            panic("http.DefaultTransport is not a *http.Transport")
    }
    unixTransport := defaultTransport.Clone()
    defaultDialContext := unixTransport.DialContext
    unixTransport.DialContext = func(ctx context.Context, _, _ string) (net.Conn, error) {
            return defaultDialContext(ctx, "unix", "/path/to/socket")
    }
    client := http.Client{Transport: unixTransport}
    client.Get("http://example.com")

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。