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

相关推荐
qq_1728055912 小时前
GIN 反向代理功能
后端·golang·go
__AtYou__12 小时前
Golang | Leetcode Golang题解之第535题TinyURL的加密与解密
leetcode·golang·题解
kevin_tech17 小时前
Go API 多种响应的规范化处理和简化策略
开发语言·后端·golang·状态模式
幺零九零零20 小时前
【Golang】sql.Null* 类型使用(处理空值和零值)
数据库·sql·golang
cookies_s_s20 小时前
Golang--DOS命令、变量、基本数据类型、标识符
golang
__AtYou__21 小时前
Golang | Leetcode Golang题解之第541题反转字符串II
leetcode·golang·题解
flying robot21 小时前
Go的JSON转化
golang
幺零九零零1 天前
【Golang】validator库的使用
开发语言·后端·golang
海绵宝宝de派小星1 天前
Go:接口和反射
开发语言·后端·golang
techdashen1 天前
Go Modules和 雅典项目
开发语言·后端·golang