登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-03-13 15:27:30 412浏览 收藏

大家好,今天本人给大家带来文章《在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学习网公众号。

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>