使用golang操作protobuf

最新推荐文章于 2024-09-15 15:36:34 发布

xlxingff

最新推荐文章于 2024-09-15 15:36:34 发布

阅读量347

点赞数

分类专栏： golang 文章标签： golang protobuf

本文链接：https://blog.youkuaiyun.com/qq_20432379/article/details/86494826

版权

golang 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了如何使用Golang操作protobuf，从Protobuf语法到编写proto文件，再到编译和实际使用，包括proto2和proto3的语法，以及通过chat.proto文件生成chat.pb.go和实现server.go、client.go的应用示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Protobuf语法

proto2语法

proto3语法

编写proto文件

chat.proto

syntax = "proto3";

package chat;

message ReqSendChatMsg
{
    string msg         = 1;
}

message ResSendChatMsg
{
    uint32 ret         = 1;
    string msg         = 2;
    uint32 timestamp   = 3;
}

编译proto文件

protoc --go_out=./  chat.proto

生成chat.pb.go文件

使用proto

server.go

package main

import (
	"fmt"
	"io"
	"net"
	"time"

	"github.com/golang/protobuf/proto"

	chat "XXX/XXX/chat"
)

func handle(conn net.Conn) {
	defer conn.Close()
	ip := conn.RemoteAddr().String()
	fmt.Printf("new client:%s\n", ip)
	for {
		buff := make([]byte, 128)
		bufflen, err := conn.Read(buff)
		if err == io.EOF {
			fmt.Printf("%s close\n", ip)
			return
		}
		if err != nil {
			fmt.Printf("%s read err:%s\n", ip, err.Error())
			continue
		}
		if bufflen > 0 {
			req := &chat.ReqSendChatMsg{}
			proto.Unmarshal(buff[0:bufflen], req)
			fmt.Printf("%s recv:%s\n", ip, req.GetMsg())

			res := &chat.ResSendChatMsg{}
			res.Ret = 200
			res.Msg = "正在处理您的请求"
			res.Timestamp = uint32(time.Now().Unix())
			data, err := proto.Marshal(res)
			if err != nil {
				fmt.Println(err)
				continue
			}
			conn.Write(data)
		}
	}
}

func main() {
	listen, err := net.Listen("tcp", "127.0.0.1:9999")
	if err != nil {
		fmt.Printf("tcp listen err:%s\n", err.Error())
		return
	}
	defer listen.Close()
	fmt.Println("tcp listen 127.0.0.1:9999")
	for {
		conn, err := listen.Accept()
		if err != nil {
			continue
		}
		go handle(conn)
	}
}

client.go

package main

import (
	"fmt"
	"io"
	"net"

	"github.com/golang/protobuf/proto"

	chat "XXX/XXX/chat"
)

func CreatConn(num int) {
	conn, err := net.Dial("tcp", "127.0.0.1:9999")
	if err != nil {
		fmt.Printf("dial err:%s\n", err.Error())
		return
	}
	defer conn.Close()
	for {
		fmt.Print("请输入聊天信息：")

		var msg string
		fmt.Scanf("%s", &msg)
		send := &chat.ReqSendChatMsg{}
		send.Msg = msg

		data, err := proto.Marshal(send)
		if err != nil {
			fmt.Printf("proto marshal err %s\n", err.Error())
			return
		}

		_, err = conn.Write(data)
		if err != nil {
			fmt.Printf("write err:%s\n", err.Error())
		}

		buff := make([]byte, 128)
		datalen, err := conn.Read(buff)
		if err == io.EOF {
			fmt.Println("server close")
			return
		}
		if err == nil && datalen > 0 {
			recv := &chat.ResSendChatMsg{}
			proto.Unmarshal(buff[0:datalen], recv)
			ret := recv.GetRet()
			msg := recv.GetMsg()
			if ret == 200 {
				fmt.Printf("recv server: %s\n", msg)
			}
		}
	}
}

func main() {
	CreatConn(1)
}