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

相关推荐
Yeats_Liao2 小时前
Go 语言 TCP 端口扫描器实现与 Goroutine 池原理
开发语言·tcp/ip·golang
我的golang之路果然有问题2 小时前
速成GO访问sql,个人笔记
经验分享·笔记·后端·sql·golang·go·database
ErizJ6 小时前
Golang | 迭代器模式
开发语言·golang·迭代器模式
健康的猪6 小时前
golang的cgo的一点小心得
开发语言·后端·golang
听雨·眠8 小时前
go语言中defer使用指南
开发语言·后端·golang
言之。9 小时前
【Go语言】RPC 使用指南(初学者版)
开发语言·rpc·golang
唐僧洗头爱飘柔952713 小时前
(Go Gin)上手Go Gin 基于Go语言开发的Web框架,本文介绍了各种路由的配置信息;包含各场景下请求参数的基本传入接收
后端·golang·go·restful·gin·goweb开发
是发财不是旺财1 天前
跟着deepseek学golang--认识golang
开发语言·后端·golang
我的golang之路果然有问题1 天前
快速上手GO的net/http包,个人学习笔记
笔记·后端·学习·http·golang·go·net
草海桐1 天前
go 的 net 包
网络·golang·net