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

相关推荐
ifanatic11 分钟前
[面试]-golang基础面试题总结
面试·职场和发展·golang
懒是一种态度18 分钟前
Golang 调用 mongodb 的函数
数据库·mongodb·golang
XINGTECODE1 小时前
海盗王集成网关和商城服务端功能golang版
开发语言·后端·golang
入 梦皆星河1 小时前
在 Ubuntu/Debian 上安装 Go
ubuntu·golang·debian
凡人的AI工具箱1 小时前
15分钟学 Go 第 60 天 :综合项目展示 - 构建微服务电商平台(完整示例25000字)
开发语言·后端·微服务·架构·golang
007php00715 小时前
GoZero 上传文件File到阿里云 OSS 报错及优化方案
服务器·开发语言·数据库·python·阿里云·架构·golang
高 朗16 小时前
【GO基础学习】基础语法(2)切片slice
开发语言·学习·golang·slice
IT书架17 小时前
golang面试题
开发语言·后端·golang
醒过来摸鱼1 天前
【Golang】协程
开发语言·后端·golang
灼华十一1 天前
算法编程题-排序
数据结构·算法·golang·排序算法