grpc代理服务的实现(二)

目录

grpc service 的实现

假设 grpc service 的服务名是 Bar

grpc服务通过unix域监听请求

复制代码
```go
svr := grpc.NewServer()
messages.RegisterBarServer(svr, bar.New())
reflection.Register(svr)
os.Remove(sockPath)
lis, err := net.Listen("unix", sockPath)
if err != nil {
    log.Fatalf("failed to listen: %v", err)
}
go func() {
    if err := svr.Serve(lis); err != nil {
        log.Fatalf("failed to serve: %v", err)
    }
}()
```

建立与代理服务的tcp连接

go 复制代码
   // 与代理服务器建立连接
    conn, err := tls.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", tcp_server.Port), &tls.Config{InsecureSkipVerify: true})
    if err != nil {
        log.Printf("dial tcp failed: %v\n", err)
        return err
    }
    // 与代理服务器通信,告诉它 是提供服务者
    firstWrite := &tcp_server.Device{ID: deviceID, Type: tcp_server.DeviceTypeServer}
    writeData, err := json.Marshal(firstWrite)
    if err != nil {
        log.Printf("marshal first write failed: %v\n", err)
        return err
    }
    _, err = conn.Write(append(writeData, tcp_server.MessageEnd))
    if err != nil {
        log.Printf("write first write failed: %v\n", err)
        return err
    }

    bufBytes, err := tcp_server.ReadData(conn)
    if err != nil {
        log.Printf("read first write failed: %v\n", err)
        return err
    }
    recvData := &tcp_server.OK{}
    err = json.Unmarshal(bufBytes, recvData)
    if err != nil {
        log.Printf("unmarshal first write failed: %v\n", err)
        return err
    }
    if recvData.Code < 0 {
        log.Printf("first write failed: %v\n", recvData.Code)
        return errors.New("first write failed")
    }

请求转发到 unix 上

go 复制代码
   // 与本地的 unix 建立连接
   unixConn, err := net.Dial("unix", sockPath)
	if err != nil {
		log.Fatalf("failed to dial unix: %v", err)
	}
    go func() {
        defer wg.Done()
        _, err = io.Copy(conn, unixConn)
        if err != nil {
            log.Printf("io.Copy failed: %v\n", err)
            return
        }
        fmt.Println("copy conn end")
    }()
    go func() {
        defer wg.Done()
        _, err = io.Copy(unixConn, conn)
        if err != nil {
            log.Printf("io.Copy failed unixConn: %v\n", err)
            return
        }
        fmt.Println("copy unixConn end")
    }()
    return nil

代码地址

https://github.com/wanmei002/websocket-reverse-proxy

相关推荐
LuckyLay9 小时前
LeetCode算法题(Go语言实现)_20
算法·leetcode·职场和发展·golang
宦如云10 小时前
Bash语言的哈希表
开发语言·后端·golang
草海桐11 小时前
golang 的strconv包常用方法
golang·rune·strconv
老赵骑摩托11 小时前
Go语言nil原理深度解析:底层实现与比较规则
开发语言·后端·golang
紧跟先前的步伐15 小时前
【Golang】第十一弹------反射
开发语言·后端·golang
apocelipes18 小时前
golang的条件编译
golang·开发工具和环境
宦如云19 小时前
PHP语言的分布式账本
开发语言·后端·golang
草海桐1 天前
golang 的reflect包的常用方法
golang·reflect
赴前尘1 天前
chromem-go + ollama + bge-m3 进行文档向量嵌入和查询
开发语言·数据库·golang
Golinie1 天前
【Go万字洗髓经】Golang中sync.Mutex的单机锁:实现原理与底层源码
golang·并发·mutex··sync.mutex