Go net/http unix 域套接字连接

发表于 2021-12-14 本文字数： 1.6k 阅读时长 ≈ 1 分钟

go package “net/http” 无法直接使用 unix 域套接字侦听的服务，会提示，不支持的协议。

方案

实现自己的 RoundTripper 支持 unix socket 。
伪造拨号功能并将其传递给传输:

func fakeDial(proto, addr string) (conn net.Conn, err error) {
    return net.Dial("unix", sock) // 为 sock path
}

tr := &http.Transport{
    Dial: fakeDial,
}
client := &http.Client{Transport: tr}
resp, err := client.Get("http://unix/test")

package main

import (
	"fmt"
	"io"
	"net"
	"net/http"
	"os"
)

func dieIf(err error) {
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
}

const sock = "/tmp/dummy.sock"

func server() {
	l, err := net.Listen("unix", sock)
	dieIf(err)
	srv := &http.Server{
		Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			fmt.Fprintf(w, "hello: %v\n", r.RequestURI)
		}),
	}
	srv.Serve(l)
}

func fakeDial(proto, addr string) (conn net.Conn, err error) {
	return net.Dial("unix", sock)
}

func main() {
	defer os.Remove(sock)
	go server()
	time.Sleep(time.Second)
	tr := &http.Transport{
		Dial: fakeDial,
	}
	client := &http.Client{Transport: tr}
	resp, err := client.Get("http://unix/test")
	dieIf(err)
	io.Copy(os.Stdout, resp.Body)
	resp.Body.Close()
}