使用Gin时如何使用unix域套接字发出请求？

Question

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

要在命令行上使用 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 请求。我在网上找到了this，但它对我来说没有多大意义。

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

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 框架。

Answer 1

为此，您需要覆盖

http.Client

使用的传输来配置一个函数，用于创建与套接字的连接：

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 2

扩展到此答案，您可以通过克隆默认传输来保留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")