增加proto及公共广播

common.go

@@ -0,0 +1,17 @@
+package rabbitmqrpc
+
+const DES_KEY = "mqbitrab"
+
+func rpc_server_conn(url ...string) (*RabbitmqServiceClient, error) {
+
+	var wx_rpc_url string = "127.0.0.1:7955"
+	if len(url) > 0 && url[0] != "" {
+		wx_rpc_url = url[0]
+	}
+	conn, _, err := DialRabbitmqService("tcp", wx_rpc_url)
+	if err != nil {
+		return nil, err
+	}
+
+	return conn, nil
+}

rabbitmq.pb.go

@@ -0,0 +1,194 @@
+// Code generated by protoc-gen-go.
+// source: rabbitmq.proto
+// DO NOT EDIT!
+
+/*
+Package rabbitmqrpc is a generated protocol buffer package.
+
+It is generated from these files:
+	rabbitmq.proto
+
+It has these top-level messages:
+	Request
+	Response
+*/
+package rabbitmqrpc
+
+import proto "github.com/chai2010/protorpc/proto"
+import math "math"
+
+import "io"
+import "log"
+import "net"
+import "net/rpc"
+import "time"
+import protorpc "github.com/chai2010/protorpc"
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ = proto.Marshal
+var _ = math.Inf
+
+// 配置信息请求结构
+type Request struct {
+	Data             *string `protobuf:"bytes,1,opt,name=data" json:"data,omitempty"`
+	Time             *string `protobuf:"bytes,2,opt,name=time" json:"time,omitempty"`
+	Sign             *string `protobuf:"bytes,3,opt,name=sign" json:"sign,omitempty"`
+	XXX_unrecognized []byte  `json:"-"`
+}
+
+func (m *Request) Reset()         { *m = Request{} }
+func (m *Request) String() string { return proto.CompactTextString(m) }
+func (*Request) ProtoMessage()    {}
+
+func (m *Request) GetData() string {
+	if m != nil && m.Data != nil {
+		return *m.Data
+	}
+	return ""
+}
+
+func (m *Request) GetTime() string {
+	if m != nil && m.Time != nil {
+		return *m.Time
+	}
+	return ""
+}
+
+func (m *Request) GetSign() string {
+	if m != nil && m.Sign != nil {
+		return *m.Sign
+	}
+	return ""
+}
+
+// 配置信息响应结构
+type Response struct {
+	Data             *string `protobuf:"bytes,1,opt,name=data" json:"data,omitempty"`
+	Time             *string `protobuf:"bytes,2,opt,name=time" json:"time,omitempty"`
+	Sign             *string `protobuf:"bytes,3,opt,name=sign" json:"sign,omitempty"`
+	XXX_unrecognized []byte  `json:"-"`
+}
+
+func (m *Response) Reset()         { *m = Response{} }
+func (m *Response) String() string { return proto.CompactTextString(m) }
+func (*Response) ProtoMessage()    {}
+
+func (m *Response) GetData() string {
+	if m != nil && m.Data != nil {
+		return *m.Data
+	}
+	return ""
+}
+
+func (m *Response) GetTime() string {
+	if m != nil && m.Time != nil {
+		return *m.Time
+	}
+	return ""
+}
+
+func (m *Response) GetSign() string {
+	if m != nil && m.Sign != nil {
+		return *m.Sign
+	}
+	return ""
+}
+
+func init() {
+}
+
+type RabbitmqService interface {
+	Send(in *Request, out *Response) error
+}
+
+// AcceptRabbitmqServiceClient accepts connections on the listener and serves requests
+// for each incoming connection.  Accept blocks; the caller typically
+// invokes it in a go statement.
+func AcceptRabbitmqServiceClient(lis net.Listener, x RabbitmqService) {
+	srv := rpc.NewServer()
+	if err := srv.RegisterName("RabbitmqService", x); err != nil {
+		log.Fatal(err)
+	}
+
+	for {
+		conn, err := lis.Accept()
+		if err != nil {
+			log.Fatalf("lis.Accept(): %v\n", err)
+		}
+		go srv.ServeCodec(protorpc.NewServerCodec(conn))
+	}
+}
+
+// RegisterRabbitmqService publish the given RabbitmqService implementation on the server.
+func RegisterRabbitmqService(srv *rpc.Server, x RabbitmqService) error {
+	if err := srv.RegisterName("RabbitmqService", x); err != nil {
+		return err
+	}
+	return nil
+}
+
+// NewRabbitmqServiceServer returns a new RabbitmqService Server.
+func NewRabbitmqServiceServer(x RabbitmqService) *rpc.Server {
+	srv := rpc.NewServer()
+	if err := srv.RegisterName("RabbitmqService", x); err != nil {
+		log.Fatal(err)
+	}
+	return srv
+}
+
+// ListenAndServeRabbitmqService listen announces on the local network address laddr
+// and serves the given RabbitmqService implementation.
+func ListenAndServeRabbitmqService(network, addr string, x RabbitmqService) error {
+	lis, err := net.Listen(network, addr)
+	if err != nil {
+		return err
+	}
+	defer lis.Close()
+
+	srv := rpc.NewServer()
+	if err := srv.RegisterName("RabbitmqService", x); err != nil {
+		return err
+	}
+
+	for {
+		conn, err := lis.Accept()
+		if err != nil {
+			log.Fatalf("lis.Accept(): %v\n", err)
+		}
+		go srv.ServeCodec(protorpc.NewServerCodec(conn))
+	}
+}
+
+type RabbitmqServiceClient struct {
+	*rpc.Client
+}
+
+// NewRabbitmqServiceClient returns a RabbitmqService rpc.Client and stub to handle
+// requests to the set of RabbitmqService at the other end of the connection.
+func NewRabbitmqServiceClient(conn io.ReadWriteCloser) (*RabbitmqServiceClient, *rpc.Client) {
+	c := rpc.NewClientWithCodec(protorpc.NewClientCodec(conn))
+	return &RabbitmqServiceClient{c}, c
+}
+
+func (c *RabbitmqServiceClient) Send(in *Request, out *Response) error {
+	return c.Call("RabbitmqService.Send", in, out)
+}
+
+// DialRabbitmqService connects to an RabbitmqService at the specified network address.
+func DialRabbitmqService(network, addr string) (*RabbitmqServiceClient, *rpc.Client, error) {
+	c, err := protorpc.Dial(network, addr)
+	if err != nil {
+		return nil, nil, err
+	}
+	return &RabbitmqServiceClient{c}, c, nil
+}
+
+// DialRabbitmqServiceTimeout connects to an RabbitmqService at the specified network address.
+func DialRabbitmqServiceTimeout(network, addr string,
+	timeout time.Duration) (*RabbitmqServiceClient, *rpc.Client, error) {
+	c, err := protorpc.DialTimeout(network, addr, timeout)
+	if err != nil {
+		return nil, nil, err
+	}
+	return &RabbitmqServiceClient{c}, c, nil
+}

rabbitmq.proto

@@ -0,0 +1,22 @@
+syntax = "proto3";
+package rabbitmqrpc;
+
+// 配置信息请求结构
+message Request {
+	string data = 1;
+	string time = 2;
+    string sign = 3;
+}
+
+// 配置信息响应结构
+message Response {
+	string data = 1;
+    string time = 2;
+    string sign = 3;
+}
+
+
+// rpc方法
+service RabbitmqService {
+    rpc send (Request) returns (Response);    // 发送消息
+}

sign.go

@@ -0,0 +1,77 @@
+package rabbitmqrpc
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+	"errors"
+	"strconv"
+	"strings"
+	"time"
+
+	"git.tetele.net/tgo/crypter"
+)
+
+/**
+ * 签名
+ */
+func Sign(data string, salt string) string {
+
+	var build strings.Builder
+
+	build.WriteString(data)
+	build.WriteString(salt)
+	build.WriteString(DES_KEY)
+
+	data_str := build.String()
+
+	h := md5.New()
+	h.Write([]byte(data_str))             // 需要加密的字符串
+	return hex.EncodeToString(h.Sum(nil)) // 输出加密结果
+
+}
+
+/**
+ * 验证签名
+ */
+func CheckSign(sign_str, data, salt string) bool {
+	sign := Sign(data, salt)
+	if strings.Compare(sign_str, sign) > -1 {
+		return true
+	}
+	return false
+}
+
+/**
+ * 解密
+ */
+func GetOrgData(res *Response) (string, error) {
+
+	res_data := res.GetData()
+
+	if res_data == "" {
+
+		return "", errors.New("未收到收据")
+	}
+
+	time_int64, err := strconv.ParseInt(res.GetTime(), 10, 64)
+	if err != nil {
+		return "", err
+	}
+
+	now_int64 := time.Now().Unix()
+
+	if now_int64-time_int64 > 10 || time_int64-now_int64 > 10 {
+		//时间误差前后10秒，返回
+		return "", errors.New("返回时间错误")
+	}
+
+	check_sign := CheckSign(res.GetSign(), res_data, res.GetTime())
+	if !check_sign {
+		return "", errors.New("返回数据签名错误")
+	}
+
+	//解密
+	res_data_de := crypter.DesDe(res_data, DES_KEY)
+
+	return res_data_de, nil
+}