feat: 修改协议（mq、push），修复问题（礼物乱窜），添加用户rpc
parent
491699a20f
commit
aa7454bba6
@ -0,0 +1,3 @@
|
||||
package pb
|
||||
|
||||
//go:generate goctl rpc protoc user_center.proto --style=go_zero --go_out=../ --go-grpc_out=../ --zrpc_out=../
|
||||
@ -0,0 +1,344 @@
|
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.28.0
|
||||
// protoc v3.19.4
|
||||
// source: user_center.proto
|
||||
|
||||
package pb
|
||||
|
||||
import (
|
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
||||
reflect "reflect"
|
||||
sync "sync"
|
||||
)
|
||||
|
||||
const (
|
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
||||
)
|
||||
|
||||
// model
|
||||
type User struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
|
||||
Id int64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
|
||||
Username string `protobuf:"bytes,2,opt,name=username,proto3" json:"username,omitempty"`
|
||||
// platform
|
||||
Platform string `protobuf:"bytes,3,opt,name=platform,proto3" json:"platform,omitempty"`
|
||||
PUid string `protobuf:"bytes,4,opt,name=pUid,proto3" json:"pUid,omitempty"`
|
||||
PUname string `protobuf:"bytes,5,opt,name=pUname,proto3" json:"pUname,omitempty"`
|
||||
PAvatar string `protobuf:"bytes,6,opt,name=pAvatar,proto3" json:"pAvatar,omitempty"`
|
||||
PInfo string `protobuf:"bytes,7,opt,name=pInfo,proto3" json:"pInfo,omitempty"`
|
||||
}
|
||||
|
||||
func (x *User) Reset() {
|
||||
*x = User{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_user_center_proto_msgTypes[0]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *User) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*User) ProtoMessage() {}
|
||||
|
||||
func (x *User) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_user_center_proto_msgTypes[0]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use User.ProtoReflect.Descriptor instead.
|
||||
func (*User) Descriptor() ([]byte, []int) {
|
||||
return file_user_center_proto_rawDescGZIP(), []int{0}
|
||||
}
|
||||
|
||||
func (x *User) GetId() int64 {
|
||||
if x != nil {
|
||||
return x.Id
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (x *User) GetUsername() string {
|
||||
if x != nil {
|
||||
return x.Username
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *User) GetPlatform() string {
|
||||
if x != nil {
|
||||
return x.Platform
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *User) GetPUid() string {
|
||||
if x != nil {
|
||||
return x.PUid
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *User) GetPUname() string {
|
||||
if x != nil {
|
||||
return x.PUname
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *User) GetPAvatar() string {
|
||||
if x != nil {
|
||||
return x.PAvatar
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *User) GetPInfo() string {
|
||||
if x != nil {
|
||||
return x.PInfo
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// req
|
||||
type PlatformUserReq struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
|
||||
Platform string `protobuf:"bytes,1,opt,name=platform,proto3" json:"platform,omitempty"`
|
||||
PUid string `protobuf:"bytes,2,opt,name=pUid,proto3" json:"pUid,omitempty"`
|
||||
}
|
||||
|
||||
func (x *PlatformUserReq) Reset() {
|
||||
*x = PlatformUserReq{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_user_center_proto_msgTypes[1]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *PlatformUserReq) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*PlatformUserReq) ProtoMessage() {}
|
||||
|
||||
func (x *PlatformUserReq) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_user_center_proto_msgTypes[1]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use PlatformUserReq.ProtoReflect.Descriptor instead.
|
||||
func (*PlatformUserReq) Descriptor() ([]byte, []int) {
|
||||
return file_user_center_proto_rawDescGZIP(), []int{1}
|
||||
}
|
||||
|
||||
func (x *PlatformUserReq) GetPlatform() string {
|
||||
if x != nil {
|
||||
return x.Platform
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *PlatformUserReq) GetPUid() string {
|
||||
if x != nil {
|
||||
return x.PUid
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
type PlatformUserResp struct {
|
||||
state protoimpl.MessageState
|
||||
sizeCache protoimpl.SizeCache
|
||||
unknownFields protoimpl.UnknownFields
|
||||
|
||||
User *User `protobuf:"bytes,1,opt,name=user,proto3" json:"user,omitempty"`
|
||||
}
|
||||
|
||||
func (x *PlatformUserResp) Reset() {
|
||||
*x = PlatformUserResp{}
|
||||
if protoimpl.UnsafeEnabled {
|
||||
mi := &file_user_center_proto_msgTypes[2]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
}
|
||||
|
||||
func (x *PlatformUserResp) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*PlatformUserResp) ProtoMessage() {}
|
||||
|
||||
func (x *PlatformUserResp) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_user_center_proto_msgTypes[2]
|
||||
if protoimpl.UnsafeEnabled && x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use PlatformUserResp.ProtoReflect.Descriptor instead.
|
||||
func (*PlatformUserResp) Descriptor() ([]byte, []int) {
|
||||
return file_user_center_proto_rawDescGZIP(), []int{2}
|
||||
}
|
||||
|
||||
func (x *PlatformUserResp) GetUser() *User {
|
||||
if x != nil {
|
||||
return x.User
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
var File_user_center_proto protoreflect.FileDescriptor
|
||||
|
||||
var file_user_center_proto_rawDesc = []byte{
|
||||
0x0a, 0x11, 0x75, 0x73, 0x65, 0x72, 0x5f, 0x63, 0x65, 0x6e, 0x74, 0x65, 0x72, 0x2e, 0x70, 0x72,
|
||||
0x6f, 0x74, 0x6f, 0x12, 0x02, 0x70, 0x62, 0x22, 0xaa, 0x01, 0x0a, 0x04, 0x55, 0x73, 0x65, 0x72,
|
||||
0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x03, 0x52, 0x02, 0x69, 0x64,
|
||||
0x12, 0x1a, 0x0a, 0x08, 0x75, 0x73, 0x65, 0x72, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x02, 0x20, 0x01,
|
||||
0x28, 0x09, 0x52, 0x08, 0x75, 0x73, 0x65, 0x72, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x1a, 0x0a, 0x08,
|
||||
0x70, 0x6c, 0x61, 0x74, 0x66, 0x6f, 0x72, 0x6d, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08,
|
||||
0x70, 0x6c, 0x61, 0x74, 0x66, 0x6f, 0x72, 0x6d, 0x12, 0x12, 0x0a, 0x04, 0x70, 0x55, 0x69, 0x64,
|
||||
0x18, 0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x70, 0x55, 0x69, 0x64, 0x12, 0x16, 0x0a, 0x06,
|
||||
0x70, 0x55, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x05, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06, 0x70, 0x55,
|
||||
0x6e, 0x61, 0x6d, 0x65, 0x12, 0x18, 0x0a, 0x07, 0x70, 0x41, 0x76, 0x61, 0x74, 0x61, 0x72, 0x18,
|
||||
0x06, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x70, 0x41, 0x76, 0x61, 0x74, 0x61, 0x72, 0x12, 0x14,
|
||||
0x0a, 0x05, 0x70, 0x49, 0x6e, 0x66, 0x6f, 0x18, 0x07, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x70,
|
||||
0x49, 0x6e, 0x66, 0x6f, 0x22, 0x41, 0x0a, 0x0f, 0x50, 0x6c, 0x61, 0x74, 0x66, 0x6f, 0x72, 0x6d,
|
||||
0x55, 0x73, 0x65, 0x72, 0x52, 0x65, 0x71, 0x12, 0x1a, 0x0a, 0x08, 0x70, 0x6c, 0x61, 0x74, 0x66,
|
||||
0x6f, 0x72, 0x6d, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x70, 0x6c, 0x61, 0x74, 0x66,
|
||||
0x6f, 0x72, 0x6d, 0x12, 0x12, 0x0a, 0x04, 0x70, 0x55, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28,
|
||||
0x09, 0x52, 0x04, 0x70, 0x55, 0x69, 0x64, 0x22, 0x30, 0x0a, 0x10, 0x50, 0x6c, 0x61, 0x74, 0x66,
|
||||
0x6f, 0x72, 0x6d, 0x55, 0x73, 0x65, 0x72, 0x52, 0x65, 0x73, 0x70, 0x12, 0x1c, 0x0a, 0x04, 0x75,
|
||||
0x73, 0x65, 0x72, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x08, 0x2e, 0x70, 0x62, 0x2e, 0x55,
|
||||
0x73, 0x65, 0x72, 0x52, 0x04, 0x75, 0x73, 0x65, 0x72, 0x32, 0x4f, 0x0a, 0x0a, 0x75, 0x73, 0x65,
|
||||
0x72, 0x43, 0x65, 0x6e, 0x74, 0x65, 0x72, 0x12, 0x41, 0x0a, 0x14, 0x72, 0x65, 0x74, 0x72, 0x69,
|
||||
0x65, 0x76, 0x65, 0x50, 0x6c, 0x61, 0x74, 0x66, 0x6f, 0x72, 0x6d, 0x55, 0x73, 0x65, 0x72, 0x12,
|
||||
0x13, 0x2e, 0x70, 0x62, 0x2e, 0x50, 0x6c, 0x61, 0x74, 0x66, 0x6f, 0x72, 0x6d, 0x55, 0x73, 0x65,
|
||||
0x72, 0x52, 0x65, 0x71, 0x1a, 0x14, 0x2e, 0x70, 0x62, 0x2e, 0x50, 0x6c, 0x61, 0x74, 0x66, 0x6f,
|
||||
0x72, 0x6d, 0x55, 0x73, 0x65, 0x72, 0x52, 0x65, 0x73, 0x70, 0x42, 0x06, 0x5a, 0x04, 0x2e, 0x2f,
|
||||
0x70, 0x62, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
|
||||
}
|
||||
|
||||
var (
|
||||
file_user_center_proto_rawDescOnce sync.Once
|
||||
file_user_center_proto_rawDescData = file_user_center_proto_rawDesc
|
||||
)
|
||||
|
||||
func file_user_center_proto_rawDescGZIP() []byte {
|
||||
file_user_center_proto_rawDescOnce.Do(func() {
|
||||
file_user_center_proto_rawDescData = protoimpl.X.CompressGZIP(file_user_center_proto_rawDescData)
|
||||
})
|
||||
return file_user_center_proto_rawDescData
|
||||
}
|
||||
|
||||
var file_user_center_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
|
||||
var file_user_center_proto_goTypes = []interface{}{
|
||||
(*User)(nil), // 0: pb.User
|
||||
(*PlatformUserReq)(nil), // 1: pb.PlatformUserReq
|
||||
(*PlatformUserResp)(nil), // 2: pb.PlatformUserResp
|
||||
}
|
||||
var file_user_center_proto_depIdxs = []int32{
|
||||
0, // 0: pb.PlatformUserResp.user:type_name -> pb.User
|
||||
1, // 1: pb.userCenter.retrievePlatformUser:input_type -> pb.PlatformUserReq
|
||||
2, // 2: pb.userCenter.retrievePlatformUser:output_type -> pb.PlatformUserResp
|
||||
2, // [2:3] is the sub-list for method output_type
|
||||
1, // [1:2] is the sub-list for method input_type
|
||||
1, // [1:1] is the sub-list for extension type_name
|
||||
1, // [1:1] is the sub-list for extension extendee
|
||||
0, // [0:1] is the sub-list for field type_name
|
||||
}
|
||||
|
||||
func init() { file_user_center_proto_init() }
|
||||
func file_user_center_proto_init() {
|
||||
if File_user_center_proto != nil {
|
||||
return
|
||||
}
|
||||
if !protoimpl.UnsafeEnabled {
|
||||
file_user_center_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*User); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
file_user_center_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*PlatformUserReq); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
file_user_center_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
|
||||
switch v := v.(*PlatformUserResp); i {
|
||||
case 0:
|
||||
return &v.state
|
||||
case 1:
|
||||
return &v.sizeCache
|
||||
case 2:
|
||||
return &v.unknownFields
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
}
|
||||
type x struct{}
|
||||
out := protoimpl.TypeBuilder{
|
||||
File: protoimpl.DescBuilder{
|
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||
RawDescriptor: file_user_center_proto_rawDesc,
|
||||
NumEnums: 0,
|
||||
NumMessages: 3,
|
||||
NumExtensions: 0,
|
||||
NumServices: 1,
|
||||
},
|
||||
GoTypes: file_user_center_proto_goTypes,
|
||||
DependencyIndexes: file_user_center_proto_depIdxs,
|
||||
MessageInfos: file_user_center_proto_msgTypes,
|
||||
}.Build()
|
||||
File_user_center_proto = out.File
|
||||
file_user_center_proto_rawDesc = nil
|
||||
file_user_center_proto_goTypes = nil
|
||||
file_user_center_proto_depIdxs = nil
|
||||
}
|
||||
@ -0,0 +1,33 @@
|
||||
syntax = "proto3";
|
||||
|
||||
package pb;
|
||||
|
||||
option go_package = "./pb";
|
||||
|
||||
// model
|
||||
message User {
|
||||
int64 id = 1;
|
||||
string username = 2;
|
||||
// platform
|
||||
string platform = 3;
|
||||
string pUid = 4;
|
||||
string pUname = 5;
|
||||
string pAvatar = 6;
|
||||
string pInfo = 7;
|
||||
}
|
||||
|
||||
// req
|
||||
message PlatformUserReq {
|
||||
string platform = 1;
|
||||
string pUid = 2;
|
||||
}
|
||||
|
||||
message PlatformUserResp {
|
||||
User user = 1;
|
||||
}
|
||||
|
||||
|
||||
service userCenter {
|
||||
// user
|
||||
rpc retrievePlatformUser(PlatformUserReq) returns (PlatformUserResp);
|
||||
}
|
||||
@ -0,0 +1,107 @@
|
||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
||||
// versions:
|
||||
// - protoc-gen-go-grpc v1.2.0
|
||||
// - protoc v3.19.4
|
||||
// source: user_center.proto
|
||||
|
||||
package pb
|
||||
|
||||
import (
|
||||
context "context"
|
||||
grpc "google.golang.org/grpc"
|
||||
codes "google.golang.org/grpc/codes"
|
||||
status "google.golang.org/grpc/status"
|
||||
)
|
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the grpc package it is being compiled against.
|
||||
// Requires gRPC-Go v1.32.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion7
|
||||
|
||||
// UserCenterClient is the client API for UserCenter service.
|
||||
//
|
||||
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
||||
type UserCenterClient interface {
|
||||
// user
|
||||
RetrievePlatformUser(ctx context.Context, in *PlatformUserReq, opts ...grpc.CallOption) (*PlatformUserResp, error)
|
||||
}
|
||||
|
||||
type userCenterClient struct {
|
||||
cc grpc.ClientConnInterface
|
||||
}
|
||||
|
||||
func NewUserCenterClient(cc grpc.ClientConnInterface) UserCenterClient {
|
||||
return &userCenterClient{cc}
|
||||
}
|
||||
|
||||
func (c *userCenterClient) RetrievePlatformUser(ctx context.Context, in *PlatformUserReq, opts ...grpc.CallOption) (*PlatformUserResp, error) {
|
||||
out := new(PlatformUserResp)
|
||||
err := c.cc.Invoke(ctx, "/pb.userCenter/retrievePlatformUser", in, out, opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// UserCenterServer is the server API for UserCenter service.
|
||||
// All implementations must embed UnimplementedUserCenterServer
|
||||
// for forward compatibility
|
||||
type UserCenterServer interface {
|
||||
// user
|
||||
RetrievePlatformUser(context.Context, *PlatformUserReq) (*PlatformUserResp, error)
|
||||
mustEmbedUnimplementedUserCenterServer()
|
||||
}
|
||||
|
||||
// UnimplementedUserCenterServer must be embedded to have forward compatible implementations.
|
||||
type UnimplementedUserCenterServer struct {
|
||||
}
|
||||
|
||||
func (UnimplementedUserCenterServer) RetrievePlatformUser(context.Context, *PlatformUserReq) (*PlatformUserResp, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method RetrievePlatformUser not implemented")
|
||||
}
|
||||
func (UnimplementedUserCenterServer) mustEmbedUnimplementedUserCenterServer() {}
|
||||
|
||||
// UnsafeUserCenterServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to UserCenterServer will
|
||||
// result in compilation errors.
|
||||
type UnsafeUserCenterServer interface {
|
||||
mustEmbedUnimplementedUserCenterServer()
|
||||
}
|
||||
|
||||
func RegisterUserCenterServer(s grpc.ServiceRegistrar, srv UserCenterServer) {
|
||||
s.RegisterService(&UserCenter_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
func _UserCenter_RetrievePlatformUser_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(PlatformUserReq)
|
||||
if err := dec(in); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(UserCenterServer).RetrievePlatformUser(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: "/pb.userCenter/retrievePlatformUser",
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(UserCenterServer).RetrievePlatformUser(ctx, req.(*PlatformUserReq))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
}
|
||||
|
||||
// UserCenter_ServiceDesc is the grpc.ServiceDesc for UserCenter service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
// and not to be introspected or modified (even as a copy)
|
||||
var UserCenter_ServiceDesc = grpc.ServiceDesc{
|
||||
ServiceName: "pb.userCenter",
|
||||
HandlerType: (*UserCenterServer)(nil),
|
||||
Methods: []grpc.MethodDesc{
|
||||
{
|
||||
MethodName: "retrievePlatformUser",
|
||||
Handler: _UserCenter_RetrievePlatformUser_Handler,
|
||||
},
|
||||
},
|
||||
Streams: []grpc.StreamDesc{},
|
||||
Metadata: "user_center.proto",
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
// Code generated by goctl. DO NOT EDIT!
|
||||
// Source: user_center.proto
|
||||
|
||||
package usercenter
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"dcg/app/user_center/pb"
|
||||
|
||||
"github.com/zeromicro/go-zero/zrpc"
|
||||
"google.golang.org/grpc"
|
||||
)
|
||||
|
||||
type (
|
||||
PlatformUserReq = pb.PlatformUserReq
|
||||
PlatformUserResp = pb.PlatformUserResp
|
||||
User = pb.User
|
||||
|
||||
UserCenter interface {
|
||||
// user
|
||||
RetrievePlatformUser(ctx context.Context, in *PlatformUserReq, opts ...grpc.CallOption) (*PlatformUserResp, error)
|
||||
}
|
||||
|
||||
defaultUserCenter struct {
|
||||
cli zrpc.Client
|
||||
}
|
||||
)
|
||||
|
||||
func NewUserCenter(cli zrpc.Client) UserCenter {
|
||||
return &defaultUserCenter{
|
||||
cli: cli,
|
||||
}
|
||||
}
|
||||
|
||||
// user
|
||||
func (m *defaultUserCenter) RetrievePlatformUser(ctx context.Context, in *PlatformUserReq, opts ...grpc.CallOption) (*PlatformUserResp, error) {
|
||||
client := pb.NewUserCenterClient(m.cli.Conn())
|
||||
return client.RetrievePlatformUser(ctx, in, opts...)
|
||||
}
|
||||
@ -0,0 +1,70 @@
|
||||
package danmaku
|
||||
|
||||
import (
|
||||
"git.noahlan.cn/northlan/ntools-go/stringn/ac"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type (
|
||||
CMD struct {
|
||||
IsCMD bool // 是否CMD
|
||||
Arr []string // 具体CMD []string
|
||||
}
|
||||
Parser struct {
|
||||
trie *ac.Trie
|
||||
allKeyArr []string
|
||||
keywordMap map[string]struct{}
|
||||
}
|
||||
ParserBuilder struct {
|
||||
}
|
||||
)
|
||||
|
||||
func NewCMDParser(keys []string) *Parser {
|
||||
p := &Parser{
|
||||
keywordMap: make(map[string]struct{}),
|
||||
allKeyArr: make([]string, len(keys)),
|
||||
}
|
||||
for _, keyword := range keys {
|
||||
p.keywordMap[keyword] = struct{}{}
|
||||
p.allKeyArr = append(p.allKeyArr, keyword)
|
||||
}
|
||||
p.trie = ac.NewTrieBuilder().AddStrings(p.allKeyArr).Build()
|
||||
return p
|
||||
}
|
||||
|
||||
func (p *Parser) ParseTest(content string) {
|
||||
p.trie.MatchString(content)
|
||||
}
|
||||
|
||||
func (p *Parser) Parse(content string) *CMD {
|
||||
// 移除多余空格,小写
|
||||
tmpContent := strings.ToLower(strings.TrimSpace(content))
|
||||
matches := p.trie.MatchString(tmpContent)
|
||||
|
||||
allKeyLen := 0
|
||||
matchedKeyMap := make(map[string]struct{})
|
||||
for _, match := range matches {
|
||||
tmp := p.allKeyArr[match.Pattern()]
|
||||
matchedKeyMap[tmp] = struct{}{}
|
||||
allKeyLen += len(tmp)
|
||||
}
|
||||
isCMD := len(tmpContent) <= allKeyLen
|
||||
|
||||
// 避免同类型指令重复
|
||||
arrMap := make(map[rune]struct{})
|
||||
var matchedCmdArr []string
|
||||
if isCMD {
|
||||
matchedCmdArr = make([]string, 0, len(matchedKeyMap))
|
||||
for s := range matchedKeyMap {
|
||||
sRune := []rune(s)
|
||||
if _, ok := arrMap[sRune[0]]; !ok {
|
||||
arrMap[sRune[0]] = struct{}{}
|
||||
matchedCmdArr = append(matchedCmdArr, s)
|
||||
}
|
||||
}
|
||||
}
|
||||
return &CMD{
|
||||
IsCMD: len(tmpContent) <= allKeyLen,
|
||||
Arr: matchedCmdArr,
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,46 @@
|
||||
package danmaku
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParse(t *testing.T) {
|
||||
contents := []string{
|
||||
"jc2m2b1s",
|
||||
"jjjjjjjjjjjj",
|
||||
"c1c1c1c1c1c1c1c1c1c1",
|
||||
"c5c6c7c8c9c2c3c4c1c2c3c4c5c6c2c3c4c5c6c1",
|
||||
"j2jjjjjjjj",
|
||||
"j",
|
||||
"加入游戏",
|
||||
"加入",
|
||||
"没有什么意义的弹幕",
|
||||
"92813182798dsjks8923kjhsddfh892h3jkl214",
|
||||
"昵称为什么可以这么长",
|
||||
"六十九的覅哦我就法拉盛就发链接我i苏联空军弗拉放假 ",
|
||||
"一堆乱七八糟的弹幕来袭",
|
||||
"不服你咬我啊？",
|
||||
"红方前排速度m2b2",
|
||||
"c1c2c3c4c1c2c3c4",
|
||||
}
|
||||
|
||||
p := NewCMDParser([]string{"j", "c1", "c2", "c3", "c4", "b1", "b2", "b3", "s", "b2", "b3", "w", "m1", "m2", "m3", "加入", "加入游戏"})
|
||||
|
||||
for _, content := range contents {
|
||||
cmdStruct := p.Parse(content)
|
||||
fmt.Printf("弹幕: [%s] 是命令否？ [%v]\n", content, cmdStruct.IsCMD)
|
||||
fmt.Printf("清洗后的真实命令为：%s len:%d\n\n", cmdStruct.Arr, len(cmdStruct.Arr))
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkCmd(b *testing.B) {
|
||||
p := NewCMDParser([]string{"j", "c1", "c2", "c3", "c4", "b1", "b2", "b3", "s", "b2", "b3", "w", "m1", "m2", "m3", "加入", "加入游戏"})
|
||||
content := "jc2m2b1s"
|
||||
|
||||
b.ResetTimer()
|
||||
b.ReportAllocs()
|
||||
for i := 0; i < b.N; i++ {
|
||||
p.ParseTest(content)
|
||||
}
|
||||
}
|
||||
@ -1,97 +0,0 @@
|
||||
package danmaku
|
||||
|
||||
import (
|
||||
"dcg/config"
|
||||
"dcg/game/command"
|
||||
pbMq "dcg/game/pb/mq"
|
||||
pbPush "dcg/game/pb/push"
|
||||
"dcg/game/svc"
|
||||
kfk "dcg/pkg/kafka"
|
||||
"git.noahlan.cn/northlan/ntools-go/kafka"
|
||||
"git.noahlan.cn/northlan/ntools-go/logger"
|
||||
"github.com/Shopify/sarama"
|
||||
"github.com/golang/protobuf/proto"
|
||||
"regexp"
|
||||
)
|
||||
|
||||
type MsgToDbHandler struct {
|
||||
ctx *svc.ServiceContext
|
||||
msgHandle map[string]msgHandlerFunc
|
||||
regex *regexp.Regexp
|
||||
commandManager *command.Manager
|
||||
ConsumerGroup *kafka.ConsumerGroup
|
||||
}
|
||||
|
||||
func (h *MsgToDbHandler) Init(ctx *svc.ServiceContext) {
|
||||
h.ctx = ctx
|
||||
|
||||
cfg := config.Config.Kafka.Danmaku
|
||||
h.msgHandle = make(map[string]msgHandlerFunc)
|
||||
h.msgHandle["danmaku"] = h.handleDanmaku
|
||||
|
||||
h.commandManager = command.NewManager()
|
||||
h.commandManager.Register("j", h.handleAllCmd)
|
||||
h.commandManager.Register("c", h.handleAllCmd)
|
||||
h.commandManager.Register("s", h.handleAllCmd)
|
||||
h.commandManager.Register("m", h.handleAllCmd)
|
||||
h.commandManager.Register("w", h.handleAllCmd)
|
||||
|
||||
var err error
|
||||
h.regex, err = regexp.Compile(config.Config.Command.Regex)
|
||||
if err != nil {
|
||||
logger.SLog.Error(err)
|
||||
}
|
||||
|
||||
h.ConsumerGroup, err = kafka.NewConsumerGroup(&kafka.ConsumerGroupConfig{
|
||||
KafkaVersion: sarama.V3_1_0_0,
|
||||
OffsetsInitial: sarama.OffsetNewest,
|
||||
IsReturnErr: false,
|
||||
UnMarshaler: kfk.ProtobufMarshaler,
|
||||
}, cfg.Addr, []string{cfg.Topic}, config.Config.ConsumerGroupId.MsgToDb)
|
||||
|
||||
if err != nil {
|
||||
logger.SLog.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func (h *MsgToDbHandler) handleAllCmd(_ int64, _ string, user *pbPush.User) {
|
||||
|
||||
}
|
||||
|
||||
func (h *MsgToDbHandler) handleDanmaku(data []byte, msgKey string) {
|
||||
// danmaku msg proto
|
||||
var msgFromMq pbMq.MqDanmaku
|
||||
err := proto.Unmarshal(data, &msgFromMq)
|
||||
if err != nil {
|
||||
logger.SLog.Error("unmarshal msg err", err)
|
||||
return
|
||||
}
|
||||
cmdArr := h.parseCommands(msgFromMq.Content)
|
||||
for _, cmd := range cmdArr {
|
||||
h.commandManager.Handle(msgFromMq.LiveRoomId, cmd, &pbPush.User{
|
||||
UId: msgFromMq.Uid,
|
||||
Uname: msgFromMq.Uname,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func (h *MsgToDbHandler) parseCommands(msg string) []string {
|
||||
if h.regex == nil {
|
||||
logger.SLog.Error("regex is null")
|
||||
return []string{}
|
||||
}
|
||||
return h.regex.FindAllString(msg, -1)
|
||||
}
|
||||
|
||||
func (MsgToDbHandler) Setup(_ sarama.ConsumerGroupSession) error { return nil }
|
||||
func (MsgToDbHandler) Cleanup(_ sarama.ConsumerGroupSession) error { return nil }
|
||||
func (h *MsgToDbHandler) ConsumeClaim(sess sarama.ConsumerGroupSession,
|
||||
claim sarama.ConsumerGroupClaim) error {
|
||||
for msg := range claim.Messages() {
|
||||
if hFunc, ok := h.msgHandle[msg.Topic]; ok {
|
||||
hFunc(msg.Value, string(msg.Key))
|
||||
}
|
||||
sess.MarkMessage(msg, "")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue