golang 创建unix socket http服务端

迷路的羔羊~2024-02-05 15:20

服务端

go 复制代码

package main

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

func main() {

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("hello"))
	})

	http.HandleFunc("/world", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("world"))
	})
	
	// unix socket地址
	addr := "test.sock"
	
	// 如果文件先存在，必须先移除
	if err := os.Remove(addr); err != nil && err == os.ErrNotExist {
		panic(err)
	}
	
	// 使用uinx网络协议
	l, err := net.Listen("unix", addr)
	if err != nil {
		fmt.Println("net.ListenUnix", err)
		return
	}

	err = http.Serve(l, nil)
	if err != nil {
		fmt.Println("http.Serve", err)
		return
	}
}

客户端调用

go 复制代码

package main

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

func main() {

	addr1 := "test.sock"

	// 创建一个基于 Unix 域套接字的 HTTP 客户端
	transport := &http.Transport{
		DialContext: func(ctx context.Context, network string, addr string) (net.Conn, error) {
			return net.Dial("unix", addr1)
		},
	}

	client := &http.Client{Transport: transport}

	// 创建一个 HTTP GET 请求
	// http://unix/world中的unix可以为任何值，建议使用有意义的域名，例如本地服务localhost
	req, err := http.NewRequest("GET", "http://unix/world", nil)
	if err != nil {
		fmt.Println("Error creating request:", err)
		return
	}

	// 发送请求并获取响应
	resp, err := client.Do(req)
	if err != nil {
		fmt.Println("Error sending request:", err)
		return
	}
	defer resp.Body.Close()

	// 读取响应内容
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("Error reading response:", err)
		return
	}

	// 打印响应内容
	fmt.Println("Response from server:", string(body))
}