英文:
How to use unix domain socket to make a request when using Gin?
问题
我正在为firecracker创建一个包装器。
要在命令行上使用firecracker启动虚拟机,你需要将一个socket文件传递给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()
这个端点启动服务器并监听任何命令。问题是,我不知道如何使用socket文件向该服务器发送PUT请求。我在网上找到了这个,但对我来说不太明白。
下面是一个不使用任何socket文件的起始代码。
func BootSource(c *gin.Context) {
var body models.BootSource
c.BindJSON(&body)
bodyJson, _ := json.Marshal(body)
// 初始化http客户端
client := &http.Client{}
// 设置HTTP方法、URL和请求体
req, err := http.NewRequest(http.MethodPut, "http://localhost/boot-source", bytes.NewBuffer(bodyJson))
if err != nil {
panic(err)
}
// 设置请求头的Content-Type为json
_, err = client.Do(req)
if err != nil {
panic(err)
}
}
我该如何使这个PUT请求使用socket文件?
请注意,我正在使用Gin框架。
英文:
I am creating a wrapper for firecracker.
To start a VM with firecracker on command line, you have to pass a socket file to the firecracker
executable. Something like this:
firecracker --api-sock /path/to/file.socket
Then from another terminal, you can make requests to this server/socket something like this:
curl --unix-socket /tmp/firecracker.socket -XPUT 'http://localhost/actions' -d '{"action_type": "SendCtrlAltDel"}'
I am trying to replicate the same thing from within a Gin server.
I have an endpoint which does the first work, which is to start a server. A minimal code looks like this:
cmd := exec.Command("firecracker", "--api-sock", "/path/to/file.socket")
err := cmd.Start()
This endpoint starts the server and listens for any command. The problem is, I don't know how to use the socket file to make a PUT request to this server. I have found this on the web, but it does not makes much sense to me.
Here is a starter code which does not use any socket file.
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)
}
}
How do I make this PUT request use the socket file?
Please also note that I'm using Gin framework.
答案1
得分: 2
要实现这个功能,你需要重写http.Client
使用的传输方式,以配置一个函数来创建与套接字的连接:
client := http.Client{
Transport: &http.Transport{
DialContext: func(ctx context.Context, _, _ string) (net.Conn, error) {
return net.DialContext(ctx, "unix", "/path/to/socket")
}
}
}
然后你可以使用这个客户端,它发送的所有请求都将使用该连接。通常,对于通过套接字公开的HTTP服务,请求中使用的主机并不重要,所以你可以使用任何对你有意义的值,例如:
client.Get("http://firecracker/some/api/path")
然而,由于你正在尝试使用Firecracker API,为什么不直接使用他们的SDK:https://github.com/firecracker-microvm/firecracker-go-sdk
这将为你处理连接的设置,避免你手动构建所有的请求。
英文:
To do this, you'll need to override the Transport used by your http.Client
to configure a function for it to use to create a connection to the socket:
client := http.Client{
Transport: &http.Transport{
DialContext: func(ctx context.Context, _, _ string) (net.Conn, error) {
return net.DialContext(ctx, "unix", "/path/to/socket")
}
}
}
You can then use that client, and all requests made by it will use that connection. Usually for HTTP services exposed over a socket, the host used in the request is not important, so you can just use any value that makes sense to you e.g
client.Get("http://firecracker/some/api/path")
However, as you are trying to use the Firecracker API, why not just use their SDK: https://github.com/firecracker-microvm/firecracker-go-sdk
This will handle the set up of the connection for you, and prevent you needing to manually craft all of the requests.
答案2
得分: 1
扩展到这个答案,你可以通过克隆默认传输来保留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")
英文:
Extending to this answer, you can keep the http defaults by cloning default transport
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")
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库,让每个人都能够通过互相帮助和分享经验来进步。
评论