一图胜千言

上代码:

• message.proto :

  syntax = "proto3";
  package message;
  option go_package = ".;message";
  
  //订单请求参数
  message OrderRequest{
      string orderId = 1;
      int64 timeStamp = 2;
  }
  
  //订单信息
  message OrderInfo{
      string OrderId = 1;
      string OrderName = 2;
      string OrderStatus = 3;
  }
  
  

• server.go

package main

import (
	"errors"
	"net"
	"net/http"
	"net/rpc"
	"testRPC/rpcAndProto/message"
	"time"
)

type OrderService struct {
}

func (os *OrderService) GetOrderInfo(request message.OrderRequest, response *message.OrderInfo)error{
	//创建订单信息
	orderMap := map[string]message.OrderInfo{
		"1":{OrderId: "1",OrderName: "衣服",OrderStatus: "已付款"},
		"2":{OrderId: "2",OrderName: "手机",OrderStatus: "未付款"},
		"3":{OrderId: "3",OrderName: "电脑",OrderStatus: "已付款"},
	}

	//得到当前时间
	current := time.Now().Unix()
	if (request.TimeStamp>current){
		//返回异常
		*response = message.OrderInfo{OrderId: "0",OrderName: "",OrderStatus: "订单信息异常"}
	}else {
		result := orderMap[request.OrderId]
		if (result.OrderId!=""){
			//找到对应的d订单id，返回正常数据
			*response = orderMap[request.OrderId]
		}else {
			return errors.New("server error")
		}
	}

	return nil
}

func main(){
	//1.初始化结构指针
	orderServer := new(OrderService)

	//2.调用net/rpc包里面的功能进行服务对象注册
	err := rpc.RegisterName("routeName",orderServer)
	if err!=nil{
		panic(err.Error())
	}

	//3.下面的函数可以把 machUtil里面包含的功能函数注册到HTTP协议中,调用者可以使用http方式进行数据传输
	rpc.HandleHTTP()
	//4.监听
	listen,err := net.Listen("tcp",":8081")
	if err!=nil{
		panic(err.Error())
	}

	//5.处理请求
	http.Serve(listen,nil)
}

• client.go

package main

import (
	"fmt"
	"net/rpc"
	"testRPC/rpcAndProto/message"
	"time"
)

func main(){
	//1.链接服务器
	client,err := rpc.DialHTTP("tcp","localhost:8081")
	if err!=nil{
		panic(err.Error())
	}

	//2.封装请求
	timeStamp := time.Now().Unix()
	requset := message.OrderRequest{OrderId: "2",TimeStamp: timeStamp}

	var response *message.OrderInfo		//声明接受参数

	//3.同步调用
	err = client.Call("routeName.GetOrderInfo",requset,&response)
	if err!=nil{
		panic(err.Error())
	}
	fmt.Println("resp.OrderId:",response.OrderId)
	fmt.Println("resp.OrderName:",response.OrderName)
	fmt.Println("resp.OrderStatus:",response.OrderStatus)
}

编译proto，运行server.go 再运行 client.go, 齐活…

• 首先要安装grpc，怎么安装这里不赘述了，直接上代码, 文件结构和上面内置rpc的结构一样
• message.proto:

syntax = "proto3";
package message;
option go_package = ".;message";

//订单请求参数
message OrderRequest{
    string orderId = 1;
    int64 timeStamp = 2;
}

//订单信息
message OrderInfo{
    string OrderId = 1;
    string OrderName = 2;
    string OrderStatus = 3;
}

//声明服务的方法(函数), GetOrderInfo-函数, OrderRequest-请求参数,OrderInfo-返回参数
//OrderService是服务名,在对应的pb.go里面，会有注册这个服务的对应函数, RegisterOrderServiceServer
service OrderService{
    rpc GetOrderInfo(OrderRequest) returns (OrderInfo);
}

• server.go

package main

import (
	"context"
	"errors"
	"fmt"
	"net"
	"net/rpc"
	"test_grpc/message"
	"time"

	"google.golang.org/grpc"
)

type OrderServiceImpl struct {
}

func (os *OrderServiceImpl) GetOrderInfo(ctx context.Context,
	request *message.OrderRequest)(*message.OrderInfo,error){
	//创建订单信息
	orderMap := map[string]message.OrderInfo{
		"1":{OrderId: "1",OrderName: "衣服",OrderStatus: "已付款"},
		"2":{OrderId: "2",OrderName: "手机",OrderStatus: "未付款"},
		"3":{OrderId: "3",OrderName: "电脑",OrderStatus: "已付款"},
	}

	//得到当前时间
	current := time.Now().Unix()
	if (request.TimeStamp>current){
		//返回异常
		return &message.OrderInfo{OrderId: "0",OrderName: "",OrderStatus: "订单信息异常"},errors.New("timeout")
	}else {
		result := orderMap[request.OrderId]
		if (result.OrderId!=""){
			//找到对应的d订单id，返回正常数据
			fmt.Println("result:",result)
			return &result,nil
		}else {
			return nil, errors.New("server error")
		}
	}

	return nil,nil
}

func main(){
	//1.初始化grpc
	server := grpc.NewServer()

	message.RegisterOrderServiceServer(server,new(OrderServiceImpl))

	//3.下面的函数可以把 machUtil里面包含的功能函数注册到HTTP协议中,调用者可以使用http方式进行数据传输
	rpc.HandleHTTP()
	//4.监听
	listen,err := net.Listen("tcp",":8090")
	if err!=nil{
		panic(err.Error())
	}

	//5.处理请求
	_ = server.Serve(listen)
}

• client.go

package main

import (
	"context"
	"fmt"
	"google.golang.org/grpc"
	"test_grpc/message"
	"time"
)

func main(){
	//1.链接服务器
	conn,err := grpc.Dial("localhost:8090",grpc.WithInsecure())
	if err!=nil{
		panic(err.Error())
	}
	defer conn.Close()

	//创建连接
	orderServiceClient := message.NewOrderServiceClient(conn)
	//构造请求参数
	orderRequest := &message.OrderRequest{
		OrderId: "3",
		TimeStamp: time.Now().Unix(),
	}
	//发送请求并且得到回参
	orderInfo,err := orderServiceClient.GetOrderInfo(context.Background(),orderRequest)
	if err!=nil{
		panic(err.Error())
	}
	if orderInfo!=nil{
		fmt.Println("orderID:",orderInfo.OrderId)
		fmt.Println("orderName:",orderInfo.OrderName)
		fmt.Println("orderName:",orderInfo.OrderStatus)
	}
}

编译proto，运行server.go 再运行 client.go, 齐活…
注意，注意，编译grpc可用的proto命令如下:
protoc --go_out=plugins=grpc:. xxx.proto