feat: gorm
parent
1417e654a4
commit
9a4978e1e9
@ -1,20 +0,0 @@
|
|||||||
@echo off
|
|
||||||
@echo ??????????????
|
|
||||||
|
|
||||||
set tables=gift
|
|
||||||
set targetDir=.\model
|
|
||||||
set templateDir=..\..\doc\template
|
|
||||||
|
|
||||||
::set host=127.0.0.1
|
|
||||||
::set port=23306
|
|
||||||
set host=192.168.1.100
|
|
||||||
set port=3306
|
|
||||||
::set dbname=dcg
|
|
||||||
set dbname=dmgame
|
|
||||||
set username=root
|
|
||||||
set password=root
|
|
||||||
|
|
||||||
for %%i in (%tables%) do (
|
|
||||||
echo ????????? %dbname% ??? %%i
|
|
||||||
goctl model mysql datasource --url "%username%:%password%@tcp(%host%:%port%)/%dbname%" -t %%i -d %targetDir% --style go_zero --home %templateDir%
|
|
||||||
)
|
|
||||||
@ -1,12 +0,0 @@
|
|||||||
package user_center
|
|
||||||
|
|
||||||
// protoc 生成
|
|
||||||
//go:generate goctl rpc protoc ./rpc/pb/gift.proto --style=go_zero --go_out=./rpc --go-grpc_out=./rpc --zrpc_out=./rpc
|
|
||||||
|
|
||||||
// api 生成
|
|
||||||
//go:generate goctl api go -api ./api/doc/user_center.api -dir ./api --style go_zero
|
|
||||||
|
|
||||||
// api swagger 文档生成 需要sh
|
|
||||||
//go:generate goctl api plugin -plugin goctl-swagger="swagger -filename user_center.json -host 127.0.0.1" -api ./api/doc/user_center.api -dir ./api/doc
|
|
||||||
|
|
||||||
// model 生成 配合genModel.sh (可能需要复制到terminal执行)
|
|
||||||
@ -1,44 +0,0 @@
|
|||||||
package model
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlc"
|
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ GiftModel = (*customGiftModel)(nil)
|
|
||||||
|
|
||||||
type (
|
|
||||||
// GiftModel is an interface to be customized, add more methods here,
|
|
||||||
// and implement the added methods in customGiftModel.
|
|
||||||
GiftModel interface {
|
|
||||||
giftModel
|
|
||||||
FindOneByGiftId(ctx context.Context, giftId string) (*Gift, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
customGiftModel struct {
|
|
||||||
*defaultGiftModel
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
// NewGiftModel returns a model for the database table.
|
|
||||||
func NewGiftModel(conn sqlx.SqlConn) GiftModel {
|
|
||||||
return &customGiftModel{
|
|
||||||
defaultGiftModel: newGiftModel(conn),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *customGiftModel) FindOneByGiftId(ctx context.Context, giftId string) (*Gift, error) {
|
|
||||||
query := fmt.Sprintf("select %s from %s where `gift_id` = ? limit 1", giftRows, m.table)
|
|
||||||
var resp Gift
|
|
||||||
err := m.conn.QueryRowCtx(ctx, &resp, query, giftId)
|
|
||||||
switch err {
|
|
||||||
case nil:
|
|
||||||
return &resp, nil
|
|
||||||
case sqlc.ErrNotFound:
|
|
||||||
return nil, ErrNotFound
|
|
||||||
default:
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
package model
|
|
||||||
|
|
||||||
import "github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
||||||
|
|
||||||
var ErrNotFound = sqlx.ErrNotFound
|
|
||||||
@ -1,17 +0,0 @@
|
|||||||
Name: gift.rpc
|
|
||||||
ListenOn: 127.0.0.1:10002
|
|
||||||
DB:
|
|
||||||
DataSource: root:root@tcp(127.0.0.1:3306)/dmgame?charset=utf8mb4&loc=Asia%2FShanghai&parseTime=true
|
|
||||||
Etcd:
|
|
||||||
Hosts:
|
|
||||||
- 127.0.0.1:2379
|
|
||||||
Key: gift.rpc
|
|
||||||
Gift:
|
|
||||||
Kafka:
|
|
||||||
Addr: [ "127.0.0.1:9093" ]
|
|
||||||
Topic: "gift"
|
|
||||||
ConsumerGroupId: "msgToDb"
|
|
||||||
Log:
|
|
||||||
Mode: file
|
|
||||||
KeepDays: 7
|
|
||||||
Level: error
|
|
||||||
@ -1,45 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"live-service/app/gift/rpc/msg_transfer"
|
|
||||||
|
|
||||||
"live-service/app/gift/rpc/internal/config"
|
|
||||||
"live-service/app/gift/rpc/internal/server"
|
|
||||||
"live-service/app/gift/rpc/internal/svc"
|
|
||||||
"live-service/app/gift/rpc/pb"
|
|
||||||
|
|
||||||
"github.com/zeromicro/go-zero/core/conf"
|
|
||||||
"github.com/zeromicro/go-zero/core/service"
|
|
||||||
"github.com/zeromicro/go-zero/zrpc"
|
|
||||||
"google.golang.org/grpc"
|
|
||||||
"google.golang.org/grpc/reflection"
|
|
||||||
)
|
|
||||||
|
|
||||||
var configFile = flag.String("f", "etc/gift.yaml", "the config file")
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
var c config.Config
|
|
||||||
conf.MustLoad(*configFile, &c)
|
|
||||||
ctx := svc.NewServiceContext(c)
|
|
||||||
svr := server.NewGiftServer(ctx)
|
|
||||||
|
|
||||||
msg_transfer.Init(context.Background(), ctx, svr)
|
|
||||||
msg_transfer.Run()
|
|
||||||
|
|
||||||
s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
|
|
||||||
pb.RegisterGiftServer(grpcServer, svr)
|
|
||||||
|
|
||||||
if c.Mode == service.DevMode || c.Mode == service.TestMode {
|
|
||||||
reflection.Register(grpcServer)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
defer s.Stop()
|
|
||||||
|
|
||||||
fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
|
|
||||||
s.Start()
|
|
||||||
}
|
|
||||||
@ -1,39 +0,0 @@
|
|||||||
// Code generated by goctl. DO NOT EDIT!
|
|
||||||
// Source: gift.proto
|
|
||||||
|
|
||||||
package gift
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"live-service/app/gift/rpc/pb"
|
|
||||||
|
|
||||||
"github.com/zeromicro/go-zero/zrpc"
|
|
||||||
"google.golang.org/grpc"
|
|
||||||
)
|
|
||||||
|
|
||||||
type (
|
|
||||||
CollectGift = pb.CollectGift
|
|
||||||
Empty = pb.Empty
|
|
||||||
|
|
||||||
Gift interface {
|
|
||||||
// collectGift 收集礼物,仅作为收集使用
|
|
||||||
CollectGift(ctx context.Context, in *CollectGift, opts ...grpc.CallOption) (*Empty, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
defaultGift struct {
|
|
||||||
cli zrpc.Client
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
func NewGift(cli zrpc.Client) Gift {
|
|
||||||
return &defaultGift{
|
|
||||||
cli: cli,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// collectGift 收集礼物,仅作为收集使用
|
|
||||||
func (m *defaultGift) CollectGift(ctx context.Context, in *CollectGift, opts ...grpc.CallOption) (*Empty, error) {
|
|
||||||
client := pb.NewGiftClient(m.cli.Conn())
|
|
||||||
return client.CollectGift(ctx, in, opts...)
|
|
||||||
}
|
|
||||||
@ -1,25 +0,0 @@
|
|||||||
package config
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/zeromicro/go-zero/core/logx"
|
|
||||||
"github.com/zeromicro/go-zero/zrpc"
|
|
||||||
)
|
|
||||||
|
|
||||||
type (
|
|
||||||
Kafka struct {
|
|
||||||
Addr []string
|
|
||||||
Topic string
|
|
||||||
ConsumerGroupId string
|
|
||||||
}
|
|
||||||
Config struct {
|
|
||||||
zrpc.RpcServerConf
|
|
||||||
Log logx.LogConf
|
|
||||||
DB struct {
|
|
||||||
DataSource string
|
|
||||||
}
|
|
||||||
|
|
||||||
Gift struct {
|
|
||||||
Kafka Kafka
|
|
||||||
}
|
|
||||||
}
|
|
||||||
)
|
|
||||||
@ -1,53 +0,0 @@
|
|||||||
package logic
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"git.noahlan.cn/northlan/ntools-go/uuid"
|
|
||||||
"github.com/pkg/errors"
|
|
||||||
"live-service/app/gift/model"
|
|
||||||
"strconv"
|
|
||||||
|
|
||||||
"live-service/app/gift/rpc/internal/svc"
|
|
||||||
"live-service/app/gift/rpc/pb"
|
|
||||||
|
|
||||||
"github.com/zeromicro/go-zero/core/logx"
|
|
||||||
)
|
|
||||||
|
|
||||||
type CollectGiftLogic struct {
|
|
||||||
ctx context.Context
|
|
||||||
svcCtx *svc.ServiceContext
|
|
||||||
logx.Logger
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewCollectGiftLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CollectGiftLogic {
|
|
||||||
return &CollectGiftLogic{
|
|
||||||
ctx: ctx,
|
|
||||||
svcCtx: svcCtx,
|
|
||||||
Logger: logx.WithContext(ctx),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// CollectGift 收集礼物,仅作为收集使用
|
|
||||||
func (l *CollectGiftLogic) CollectGift(in *pb.CollectGift) (*pb.Empty, error) {
|
|
||||||
giftIdStr := strconv.FormatInt(in.GiftId, 10)
|
|
||||||
dbGift, err := l.svcCtx.GiftModel.FindOneByGiftId(l.ctx, giftIdStr)
|
|
||||||
if err == nil {
|
|
||||||
return &pb.Empty{}, nil
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
if !errors.Is(err, model.ErrNotFound) {
|
|
||||||
return nil, errors.Wrap(err, "数据库错误")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
dbGift = &model.Gift{
|
|
||||||
Id: uuid.NextId(),
|
|
||||||
GiftId: giftIdStr,
|
|
||||||
GiftName: in.GiftName,
|
|
||||||
Platform: in.Platform,
|
|
||||||
PPrice: in.TotalCoin,
|
|
||||||
}
|
|
||||||
if _, err := l.svcCtx.GiftModel.Insert(l.ctx, dbGift); err != nil {
|
|
||||||
return nil, errors.Wrap(err, "数据库插入错误")
|
|
||||||
}
|
|
||||||
return &pb.Empty{}, nil
|
|
||||||
}
|
|
||||||
@ -1,29 +0,0 @@
|
|||||||
// Code generated by goctl. DO NOT EDIT!
|
|
||||||
// Source: gift.proto
|
|
||||||
|
|
||||||
package server
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"live-service/app/gift/rpc/internal/logic"
|
|
||||||
"live-service/app/gift/rpc/internal/svc"
|
|
||||||
"live-service/app/gift/rpc/pb"
|
|
||||||
)
|
|
||||||
|
|
||||||
type GiftServer struct {
|
|
||||||
svcCtx *svc.ServiceContext
|
|
||||||
pb.UnimplementedGiftServer
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewGiftServer(svcCtx *svc.ServiceContext) *GiftServer {
|
|
||||||
return &GiftServer{
|
|
||||||
svcCtx: svcCtx,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// collectGift 收集礼物,仅作为收集使用
|
|
||||||
func (s *GiftServer) CollectGift(ctx context.Context, in *pb.CollectGift) (*pb.Empty, error) {
|
|
||||||
l := logic.NewCollectGiftLogic(ctx, s.svcCtx)
|
|
||||||
return l.CollectGift(in)
|
|
||||||
}
|
|
||||||
@ -1,21 +0,0 @@
|
|||||||
package svc
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
||||||
"live-service/app/gift/model"
|
|
||||||
"live-service/app/gift/rpc/internal/config"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ServiceContext struct {
|
|
||||||
Config config.Config
|
|
||||||
|
|
||||||
GiftModel model.GiftModel
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewServiceContext(c config.Config) *ServiceContext {
|
|
||||||
return &ServiceContext{
|
|
||||||
Config: c,
|
|
||||||
// model
|
|
||||||
GiftModel: model.NewGiftModel(sqlx.NewMysql(c.DB.DataSource)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,85 +0,0 @@
|
|||||||
package gift
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"git.noahlan.cn/northlan/ntools-go/kafka"
|
|
||||||
"github.com/Shopify/sarama"
|
|
||||||
"github.com/zeromicro/go-zero/core/logx"
|
|
||||||
"google.golang.org/protobuf/proto"
|
|
||||||
"live-service/app/gift/rpc/internal/server"
|
|
||||||
"live-service/app/gift/rpc/internal/svc"
|
|
||||||
"live-service/app/gift/rpc/pb"
|
|
||||||
pbMq "live-service/app/pb/mq"
|
|
||||||
kfk "live-service/common/kafka"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
type msgHandlerFunc func(data []byte, msgKey string)
|
|
||||||
|
|
||||||
type MsgToDBHandler struct {
|
|
||||||
context context.Context
|
|
||||||
ctx *svc.ServiceContext
|
|
||||||
svr *server.GiftServer
|
|
||||||
|
|
||||||
msgHandle map[string]msgHandlerFunc
|
|
||||||
|
|
||||||
ConsumerGroup *kafka.ConsumerGroup
|
|
||||||
|
|
||||||
mutex sync.Mutex
|
|
||||||
logx.Logger
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *MsgToDBHandler) Init(context context.Context, ctx *svc.ServiceContext, svr *server.GiftServer) {
|
|
||||||
h.context = context
|
|
||||||
h.ctx = ctx
|
|
||||||
h.svr = svr
|
|
||||||
h.Logger = logx.WithContext(h.context)
|
|
||||||
|
|
||||||
cfg := ctx.Config.Gift.Kafka
|
|
||||||
h.msgHandle = make(map[string]msgHandlerFunc)
|
|
||||||
h.msgHandle["gift"] = h.handleGift
|
|
||||||
|
|
||||||
var err error
|
|
||||||
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}, cfg.ConsumerGroupId)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
h.Logger.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *MsgToDBHandler) handleGift(data []byte, msgKey string) {
|
|
||||||
// msg proto
|
|
||||||
var msgFromMq pbMq.MqGift
|
|
||||||
if err := proto.Unmarshal(data, &msgFromMq); err != nil {
|
|
||||||
h.Logger.Error("unmarshal msg err", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
h.mutex.Lock()
|
|
||||||
defer h.mutex.Unlock()
|
|
||||||
if _, err := h.svr.CollectGift(context.Background(), &pb.CollectGift{
|
|
||||||
Platform: msgFromMq.Platform,
|
|
||||||
GiftId: msgFromMq.GiftId,
|
|
||||||
GiftName: msgFromMq.GiftName,
|
|
||||||
TotalCoin: msgFromMq.TotalCoin,
|
|
||||||
}); err != nil {
|
|
||||||
h.Logger.Error("收集礼物错误 ", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
||||||
@ -1,20 +0,0 @@
|
|||||||
package msg_transfer
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"live-service/app/gift/rpc/internal/server"
|
|
||||||
"live-service/app/gift/rpc/internal/svc"
|
|
||||||
"live-service/app/gift/rpc/msg_transfer/gift"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
giftMsgToDb gift.MsgToDBHandler
|
|
||||||
)
|
|
||||||
|
|
||||||
func Init(context context.Context, ctx *svc.ServiceContext, svr *server.GiftServer) {
|
|
||||||
giftMsgToDb.Init(context, ctx, svr)
|
|
||||||
}
|
|
||||||
|
|
||||||
func Run() {
|
|
||||||
go giftMsgToDb.ConsumerGroup.RegisterHandlerAndConsumer(&giftMsgToDb)
|
|
||||||
}
|
|
||||||
@ -1,3 +0,0 @@
|
|||||||
package pb
|
|
||||||
|
|
||||||
//go:generate goctl rpc protoc gift.proto --style=go_zero --go_out=../ --go-grpc_out=../ --zrpc_out=../
|
|
||||||
@ -1,228 +0,0 @@
|
|||||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|
||||||
// versions:
|
|
||||||
// protoc-gen-go v1.27.1
|
|
||||||
// protoc v3.19.4
|
|
||||||
// source: gift.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)
|
|
||||||
)
|
|
||||||
|
|
||||||
// req
|
|
||||||
type CollectGift struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
|
|
||||||
Platform string `protobuf:"bytes,1,opt,name=platform,proto3" json:"platform,omitempty"`
|
|
||||||
GiftId int64 `protobuf:"varint,2,opt,name=giftId,proto3" json:"giftId,omitempty"`
|
|
||||||
GiftName string `protobuf:"bytes,3,opt,name=giftName,proto3" json:"giftName,omitempty"`
|
|
||||||
TotalCoin int64 `protobuf:"varint,4,opt,name=totalCoin,proto3" json:"totalCoin,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *CollectGift) Reset() {
|
|
||||||
*x = CollectGift{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_gift_proto_msgTypes[0]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *CollectGift) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*CollectGift) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *CollectGift) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_gift_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 CollectGift.ProtoReflect.Descriptor instead.
|
|
||||||
func (*CollectGift) Descriptor() ([]byte, []int) {
|
|
||||||
return file_gift_proto_rawDescGZIP(), []int{0}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *CollectGift) GetPlatform() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.Platform
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *CollectGift) GetGiftId() int64 {
|
|
||||||
if x != nil {
|
|
||||||
return x.GiftId
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *CollectGift) GetGiftName() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.GiftName
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *CollectGift) GetTotalCoin() int64 {
|
|
||||||
if x != nil {
|
|
||||||
return x.TotalCoin
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
type Empty struct {
|
|
||||||
state protoimpl.MessageState
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Empty) Reset() {
|
|
||||||
*x = Empty{}
|
|
||||||
if protoimpl.UnsafeEnabled {
|
|
||||||
mi := &file_gift_proto_msgTypes[1]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Empty) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*Empty) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *Empty) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_gift_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 Empty.ProtoReflect.Descriptor instead.
|
|
||||||
func (*Empty) Descriptor() ([]byte, []int) {
|
|
||||||
return file_gift_proto_rawDescGZIP(), []int{1}
|
|
||||||
}
|
|
||||||
|
|
||||||
var File_gift_proto protoreflect.FileDescriptor
|
|
||||||
|
|
||||||
var file_gift_proto_rawDesc = []byte{
|
|
||||||
0x0a, 0x0a, 0x67, 0x69, 0x66, 0x74, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x02, 0x70, 0x62,
|
|
||||||
0x22, 0x7b, 0x0a, 0x0b, 0x43, 0x6f, 0x6c, 0x6c, 0x65, 0x63, 0x74, 0x47, 0x69, 0x66, 0x74, 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, 0x16, 0x0a, 0x06, 0x67,
|
|
||||||
0x69, 0x66, 0x74, 0x49, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x03, 0x52, 0x06, 0x67, 0x69, 0x66,
|
|
||||||
0x74, 0x49, 0x64, 0x12, 0x1a, 0x0a, 0x08, 0x67, 0x69, 0x66, 0x74, 0x4e, 0x61, 0x6d, 0x65, 0x18,
|
|
||||||
0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x67, 0x69, 0x66, 0x74, 0x4e, 0x61, 0x6d, 0x65, 0x12,
|
|
||||||
0x1c, 0x0a, 0x09, 0x74, 0x6f, 0x74, 0x61, 0x6c, 0x43, 0x6f, 0x69, 0x6e, 0x18, 0x04, 0x20, 0x01,
|
|
||||||
0x28, 0x03, 0x52, 0x09, 0x74, 0x6f, 0x74, 0x61, 0x6c, 0x43, 0x6f, 0x69, 0x6e, 0x22, 0x07, 0x0a,
|
|
||||||
0x05, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x32, 0x31, 0x0a, 0x04, 0x67, 0x69, 0x66, 0x74, 0x12, 0x29,
|
|
||||||
0x0a, 0x0b, 0x63, 0x6f, 0x6c, 0x6c, 0x65, 0x63, 0x74, 0x47, 0x69, 0x66, 0x74, 0x12, 0x0f, 0x2e,
|
|
||||||
0x70, 0x62, 0x2e, 0x43, 0x6f, 0x6c, 0x6c, 0x65, 0x63, 0x74, 0x47, 0x69, 0x66, 0x74, 0x1a, 0x09,
|
|
||||||
0x2e, 0x70, 0x62, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x42, 0x06, 0x5a, 0x04, 0x2e, 0x2f, 0x70,
|
|
||||||
0x62, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
file_gift_proto_rawDescOnce sync.Once
|
|
||||||
file_gift_proto_rawDescData = file_gift_proto_rawDesc
|
|
||||||
)
|
|
||||||
|
|
||||||
func file_gift_proto_rawDescGZIP() []byte {
|
|
||||||
file_gift_proto_rawDescOnce.Do(func() {
|
|
||||||
file_gift_proto_rawDescData = protoimpl.X.CompressGZIP(file_gift_proto_rawDescData)
|
|
||||||
})
|
|
||||||
return file_gift_proto_rawDescData
|
|
||||||
}
|
|
||||||
|
|
||||||
var file_gift_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
|
|
||||||
var file_gift_proto_goTypes = []interface{}{
|
|
||||||
(*CollectGift)(nil), // 0: pb.CollectGift
|
|
||||||
(*Empty)(nil), // 1: pb.Empty
|
|
||||||
}
|
|
||||||
var file_gift_proto_depIdxs = []int32{
|
|
||||||
0, // 0: pb.gift.collectGift:input_type -> pb.CollectGift
|
|
||||||
1, // 1: pb.gift.collectGift:output_type -> pb.Empty
|
|
||||||
1, // [1:2] is the sub-list for method output_type
|
|
||||||
0, // [0:1] is the sub-list for method input_type
|
|
||||||
0, // [0:0] is the sub-list for extension type_name
|
|
||||||
0, // [0:0] is the sub-list for extension extendee
|
|
||||||
0, // [0:0] is the sub-list for field type_name
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() { file_gift_proto_init() }
|
|
||||||
func file_gift_proto_init() {
|
|
||||||
if File_gift_proto != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if !protoimpl.UnsafeEnabled {
|
|
||||||
file_gift_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*CollectGift); i {
|
|
||||||
case 0:
|
|
||||||
return &v.state
|
|
||||||
case 1:
|
|
||||||
return &v.sizeCache
|
|
||||||
case 2:
|
|
||||||
return &v.unknownFields
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
file_gift_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
|
|
||||||
switch v := v.(*Empty); 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_gift_proto_rawDesc,
|
|
||||||
NumEnums: 0,
|
|
||||||
NumMessages: 2,
|
|
||||||
NumExtensions: 0,
|
|
||||||
NumServices: 1,
|
|
||||||
},
|
|
||||||
GoTypes: file_gift_proto_goTypes,
|
|
||||||
DependencyIndexes: file_gift_proto_depIdxs,
|
|
||||||
MessageInfos: file_gift_proto_msgTypes,
|
|
||||||
}.Build()
|
|
||||||
File_gift_proto = out.File
|
|
||||||
file_gift_proto_rawDesc = nil
|
|
||||||
file_gift_proto_goTypes = nil
|
|
||||||
file_gift_proto_depIdxs = nil
|
|
||||||
}
|
|
||||||
@ -1,20 +0,0 @@
|
|||||||
syntax = "proto3";
|
|
||||||
|
|
||||||
package pb;
|
|
||||||
|
|
||||||
option go_package = "./pb";
|
|
||||||
|
|
||||||
// req
|
|
||||||
message CollectGift {
|
|
||||||
string platform = 1;
|
|
||||||
int64 giftId = 2;
|
|
||||||
string giftName = 3;
|
|
||||||
int64 totalCoin = 4;
|
|
||||||
}
|
|
||||||
|
|
||||||
message Empty {}
|
|
||||||
|
|
||||||
service gift {
|
|
||||||
// collectGift 收集礼物,仅作为收集使用
|
|
||||||
rpc collectGift(CollectGift) returns (Empty);
|
|
||||||
}
|
|
||||||
@ -1,107 +0,0 @@
|
|||||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
|
||||||
// versions:
|
|
||||||
// - protoc-gen-go-grpc v1.2.0
|
|
||||||
// - protoc v3.19.4
|
|
||||||
// source: gift.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
|
|
||||||
|
|
||||||
// GiftClient is the client API for Gift 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 GiftClient interface {
|
|
||||||
// collectGift 收集礼物,仅作为收集使用
|
|
||||||
CollectGift(ctx context.Context, in *CollectGift, opts ...grpc.CallOption) (*Empty, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
type giftClient struct {
|
|
||||||
cc grpc.ClientConnInterface
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewGiftClient(cc grpc.ClientConnInterface) GiftClient {
|
|
||||||
return &giftClient{cc}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *giftClient) CollectGift(ctx context.Context, in *CollectGift, opts ...grpc.CallOption) (*Empty, error) {
|
|
||||||
out := new(Empty)
|
|
||||||
err := c.cc.Invoke(ctx, "/pb.gift/collectGift", in, out, opts...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return out, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GiftServer is the server API for Gift service.
|
|
||||||
// All implementations must embed UnimplementedGiftServer
|
|
||||||
// for forward compatibility
|
|
||||||
type GiftServer interface {
|
|
||||||
// collectGift 收集礼物,仅作为收集使用
|
|
||||||
CollectGift(context.Context, *CollectGift) (*Empty, error)
|
|
||||||
mustEmbedUnimplementedGiftServer()
|
|
||||||
}
|
|
||||||
|
|
||||||
// UnimplementedGiftServer must be embedded to have forward compatible implementations.
|
|
||||||
type UnimplementedGiftServer struct {
|
|
||||||
}
|
|
||||||
|
|
||||||
func (UnimplementedGiftServer) CollectGift(context.Context, *CollectGift) (*Empty, error) {
|
|
||||||
return nil, status.Errorf(codes.Unimplemented, "method CollectGift not implemented")
|
|
||||||
}
|
|
||||||
func (UnimplementedGiftServer) mustEmbedUnimplementedGiftServer() {}
|
|
||||||
|
|
||||||
// UnsafeGiftServer may be embedded to opt out of forward compatibility for this service.
|
|
||||||
// Use of this interface is not recommended, as added methods to GiftServer will
|
|
||||||
// result in compilation errors.
|
|
||||||
type UnsafeGiftServer interface {
|
|
||||||
mustEmbedUnimplementedGiftServer()
|
|
||||||
}
|
|
||||||
|
|
||||||
func RegisterGiftServer(s grpc.ServiceRegistrar, srv GiftServer) {
|
|
||||||
s.RegisterService(&Gift_ServiceDesc, srv)
|
|
||||||
}
|
|
||||||
|
|
||||||
func _Gift_CollectGift_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
|
||||||
in := new(CollectGift)
|
|
||||||
if err := dec(in); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if interceptor == nil {
|
|
||||||
return srv.(GiftServer).CollectGift(ctx, in)
|
|
||||||
}
|
|
||||||
info := &grpc.UnaryServerInfo{
|
|
||||||
Server: srv,
|
|
||||||
FullMethod: "/pb.gift/collectGift",
|
|
||||||
}
|
|
||||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
|
||||||
return srv.(GiftServer).CollectGift(ctx, req.(*CollectGift))
|
|
||||||
}
|
|
||||||
return interceptor(ctx, in, info, handler)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Gift_ServiceDesc is the grpc.ServiceDesc for Gift service.
|
|
||||||
// It's only intended for direct use with grpc.RegisterService,
|
|
||||||
// and not to be introspected or modified (even as a copy)
|
|
||||||
var Gift_ServiceDesc = grpc.ServiceDesc{
|
|
||||||
ServiceName: "pb.gift",
|
|
||||||
HandlerType: (*GiftServer)(nil),
|
|
||||||
Methods: []grpc.MethodDesc{
|
|
||||||
{
|
|
||||||
MethodName: "collectGift",
|
|
||||||
Handler: _Gift_CollectGift_Handler,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
Streams: []grpc.StreamDesc{},
|
|
||||||
Metadata: "gift.proto",
|
|
||||||
}
|
|
||||||
@ -1,5 +1,5 @@
|
|||||||
package model
|
package model
|
||||||
|
|
||||||
import "github.com/zeromicro/go-zero/core/stores/sqlx"
|
import "gorm.io/gorm"
|
||||||
|
|
||||||
var ErrNotFound = sqlx.ErrNotFound
|
var ErrNotFound = gorm.ErrRecordNotFound
|
||||||
|
|||||||
@ -0,0 +1,45 @@
|
|||||||
|
package platform_user
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestRetrieveBilibili(t *testing.T) {
|
||||||
|
str := `{
|
||||||
|
"error": "",
|
||||||
|
"message": [
|
||||||
|
{
|
||||||
|
"id": 48,
|
||||||
|
"ip": "120.42.46.226",
|
||||||
|
"port": "6666",
|
||||||
|
"scheme_type": 0,
|
||||||
|
"content": "120.42.46.226:6666",
|
||||||
|
"assess_times": 16,
|
||||||
|
"success_times": 14,
|
||||||
|
"avg_response_time": 2.063625,
|
||||||
|
"continuous_failed_times": 0,
|
||||||
|
"score": 3.7924709776481986,
|
||||||
|
"insert_time": 1650852112,
|
||||||
|
"update_time": 1650853064
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}`
|
||||||
|
var respStruct struct {
|
||||||
|
Error string `json:"error"`
|
||||||
|
Message []MyProxy `json:"message"`
|
||||||
|
}
|
||||||
|
err := json.Unmarshal([]byte(str), &respStruct)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Printf("%+v\n", respStruct)
|
||||||
|
|
||||||
|
bilibili, err := RetrieveBilibili("2")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Printf("%+v\n", bilibili)
|
||||||
|
}
|
||||||
@ -1,35 +0,0 @@
|
|||||||
package danmaku
|
|
||||||
|
|
||||||
import pbMq "live-service/app/pb/mq"
|
|
||||||
|
|
||||||
type HandlerFunc func(roomId int64, cmd string, dm *pbMq.MqDanmaku)
|
|
||||||
|
|
||||||
type Manager struct {
|
|
||||||
handlers map[string]HandlerFunc
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewManager() *Manager {
|
|
||||||
return &Manager{
|
|
||||||
handlers: make(map[string]HandlerFunc),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *Manager) Register(h HandlerFunc, cmd string, alias ...string) {
|
|
||||||
if _, ok := m.handlers[cmd]; ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
m.handlers[cmd] = h
|
|
||||||
// alias
|
|
||||||
for _, s := range alias {
|
|
||||||
if _, ok := m.handlers[cmd]; ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
m.handlers[s] = h
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *Manager) Handle(roomId int64, cmd string, user *pbMq.MqDanmaku) {
|
|
||||||
if h, ok := m.handlers[cmd]; ok {
|
|
||||||
h(roomId, cmd, user)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,70 +0,0 @@
|
|||||||
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,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,103 +0,0 @@
|
|||||||
package danmaku
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"git.noahlan.cn/northlan/ntools-go/kafka"
|
|
||||||
"github.com/Shopify/sarama"
|
|
||||||
"github.com/zeromicro/go-zero/core/logx"
|
|
||||||
"google.golang.org/protobuf/proto"
|
|
||||||
pbMq "live-service/app/pb/mq"
|
|
||||||
"live-service/app/user_center/rpc/internal/server"
|
|
||||||
"live-service/app/user_center/rpc/internal/svc"
|
|
||||||
"live-service/app/user_center/rpc/pb"
|
|
||||||
kfk "live-service/common/kafka"
|
|
||||||
"strconv"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
type msgHandlerFunc func(data []byte, msgKey string)
|
|
||||||
|
|
||||||
type MsgToDBHandler struct {
|
|
||||||
context context.Context
|
|
||||||
ctx *svc.ServiceContext
|
|
||||||
svr *server.UserCenterServer
|
|
||||||
|
|
||||||
msgHandle map[string]msgHandlerFunc
|
|
||||||
|
|
||||||
cmdParser *Parser
|
|
||||||
commandManager *Manager
|
|
||||||
|
|
||||||
ConsumerGroup *kafka.ConsumerGroup
|
|
||||||
|
|
||||||
mutex sync.Mutex
|
|
||||||
logx.Logger
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *MsgToDBHandler) Init(context context.Context, ctx *svc.ServiceContext, svr *server.UserCenterServer) {
|
|
||||||
h.context = context
|
|
||||||
h.ctx = ctx
|
|
||||||
h.svr = svr
|
|
||||||
h.Logger = logx.WithContext(h.context)
|
|
||||||
|
|
||||||
cfg := ctx.Config.Danmaku.Kafka
|
|
||||||
h.msgHandle = make(map[string]msgHandlerFunc)
|
|
||||||
h.msgHandle["danmaku"] = h.handleDanmaku
|
|
||||||
|
|
||||||
h.cmdParser = NewCMDParser(ctx.Config.Danmaku.Command.Keys)
|
|
||||||
//
|
|
||||||
h.commandManager = NewManager()
|
|
||||||
h.commandManager.Register(h.handleJoinGame, "j", "J", "加入", "加入游戏")
|
|
||||||
|
|
||||||
var err error
|
|
||||||
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}, cfg.ConsumerGroupId)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
h.Logger.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *MsgToDBHandler) handleJoinGame(_ int64, _ string, danmaku *pbMq.MqDanmaku) {
|
|
||||||
h.mutex.Lock()
|
|
||||||
defer h.mutex.Unlock()
|
|
||||||
_, err := h.svr.RetrievePlatformUser(h.context, &pb.PlatformUserReq{
|
|
||||||
Platform: danmaku.Platform,
|
|
||||||
PUid: strconv.FormatInt(danmaku.Uid, 10),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *MsgToDBHandler) handleDanmaku(data []byte, msgKey string) {
|
|
||||||
// msg proto
|
|
||||||
var msgFromMq pbMq.MqDanmaku
|
|
||||||
if err := proto.Unmarshal(data, &msgFromMq); err != nil {
|
|
||||||
h.Logger.Error("unmarshal msg err", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
cmdStruct := h.cmdParser.Parse(msgFromMq.Content)
|
|
||||||
// 只处理命令
|
|
||||||
if cmdStruct.IsCMD {
|
|
||||||
for _, cmd := range cmdStruct.Arr {
|
|
||||||
h.commandManager.Handle(msgFromMq.LiveRoomId, cmd, &msgFromMq)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
||||||
@ -1,20 +0,0 @@
|
|||||||
package msg_transfer
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"live-service/app/user_center/rpc/internal/server"
|
|
||||||
"live-service/app/user_center/rpc/internal/svc"
|
|
||||||
"live-service/app/user_center/rpc/msg_transfer/danmaku"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
danmakuMsgToDb danmaku.MsgToDBHandler
|
|
||||||
)
|
|
||||||
|
|
||||||
func Init(context context.Context, ctx *svc.ServiceContext, svr *server.UserCenterServer) {
|
|
||||||
danmakuMsgToDb.Init(context, ctx, svr)
|
|
||||||
}
|
|
||||||
|
|
||||||
func Run() {
|
|
||||||
go danmakuMsgToDb.ConsumerGroup.RegisterHandlerAndConsumer(&danmakuMsgToDb)
|
|
||||||
}
|
|
||||||
@ -1,5 +1,5 @@
|
|||||||
package {{.pkg}}
|
package {{.pkg}}
|
||||||
|
|
||||||
import "github.com/zeromicro/go-zero/core/stores/sqlx"
|
import "gorm.io/gorm"
|
||||||
|
|
||||||
var ErrNotFound = sqlx.ErrNotFound
|
var ErrNotFound = gorm.ErrRecordNotFound
|
||||||
|
|||||||
@ -1,12 +1,10 @@
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"git.noahlan.cn/northlan/ntools-go/gorm-zero/gormc"
|
||||||
"fmt"
|
|
||||||
"strings"
|
"strings"
|
||||||
{{if .time}}"time"{{end}}
|
{{if .time}}"time"{{end}}
|
||||||
|
|
||||||
"github.com/zeromicro/go-zero/core/stores/builder"
|
"github.com/zeromicro/go-zero/core/stores/builder"
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlc"
|
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
||||||
"github.com/zeromicro/go-zero/core/stringx"
|
"github.com/zeromicro/go-zero/core/stringx"
|
||||||
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|||||||
@ -1,13 +1,12 @@
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"strings"
|
"strings"
|
||||||
{{if .time}}"time"{{end}}
|
{{if .time}}"time"{{end}}
|
||||||
|
|
||||||
|
"git.noahlan.cn/northlan/ntools-go/gorm-zero/gormc"
|
||||||
"github.com/zeromicro/go-zero/core/stores/builder"
|
"github.com/zeromicro/go-zero/core/stores/builder"
|
||||||
"github.com/zeromicro/go-zero/core/stores/cache"
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlc"
|
|
||||||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
||||||
"github.com/zeromicro/go-zero/core/stringx"
|
"github.com/zeromicro/go-zero/core/stringx"
|
||||||
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|||||||
@ -1,10 +1,8 @@
|
|||||||
|
|
||||||
func (m *default{{.upperStartCamelObject}}Model) Insert(ctx context.Context, data *{{.upperStartCamelObject}}) (sql.Result,error) {
|
func (m *default{{.upperStartCamelObject}}Model) Insert(ctx context.Context, data *{{.upperStartCamelObject}}) error {
|
||||||
{{if .withCache}}{{.keys}}
|
{{if .withCache}}{{.keys}}
|
||||||
ret, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
|
err := m.ExecCtx(ctx, func(conn *gorm.DB) *gorm.DB {
|
||||||
query := fmt.Sprintf("insert into %s (%s) values ({{.expression}})", m.table, {{.lowerStartCamelObject}}RowsExpectAutoSet)
|
return conn.Save(&data)
|
||||||
return conn.ExecCtx(ctx, query, {{.expressionValues}})
|
}, {{.keyValues}}){{else}}err:=m.conn.WithContext(ctx).Save(&data).Error{{end}}
|
||||||
}, {{.keyValues}}){{else}}query := fmt.Sprintf("insert into %s (%s) values ({{.expression}})", m.table, {{.lowerStartCamelObject}}RowsExpectAutoSet)
|
return err
|
||||||
ret,err:=m.conn.ExecCtx(ctx, query, {{.expressionValues}}){{end}}
|
|
||||||
return ret,err
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1 +1 @@
|
|||||||
Insert(ctx context.Context, data *{{.upperStartCamelObject}}) (sql.Result,error)
|
Insert(ctx context.Context, data *{{.upperStartCamelObject}}) error
|
||||||
@ -1,7 +1,7 @@
|
|||||||
|
|
||||||
func new{{.upperStartCamelObject}}Model(conn sqlx.SqlConn{{if .withCache}}, c cache.CacheConf{{end}}) *default{{.upperStartCamelObject}}Model {
|
func new{{.upperStartCamelObject}}Model(conn *gorm.DB{{if .withCache}}, c cache.CacheConf{{end}}) *default{{.upperStartCamelObject}}Model {
|
||||||
return &default{{.upperStartCamelObject}}Model{
|
return &default{{.upperStartCamelObject}}Model{
|
||||||
{{if .withCache}}CachedConn: sqlc.NewConn(conn, c){{else}}conn:conn{{end}},
|
{{if .withCache}}CachedConn: gormc.NewConn(conn, c){{else}}conn:conn{{end}},
|
||||||
table: {{.table}},
|
table: {{.table}},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1 +1 @@
|
|||||||
`db:"{{.field}}"`
|
`gorm:"column:{{.field}}"`
|
||||||
@ -1,10 +1,8 @@
|
|||||||
|
|
||||||
func (m *default{{.upperStartCamelObject}}Model) Update(ctx context.Context, data *{{.upperStartCamelObject}}) error {
|
func (m *default{{.upperStartCamelObject}}Model) Update(ctx context.Context, data *{{.upperStartCamelObject}}) error {
|
||||||
{{if .withCache}}{{.keys}}
|
{{if .withCache}}{{.keys}}
|
||||||
_, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
|
err := m.ExecCtx(ctx, func(conn *gorm.DB) *gorm.DB {
|
||||||
query := fmt.Sprintf("update %s set %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}}", m.table, {{.lowerStartCamelObject}}RowsWithPlaceHolder)
|
return conn.Save(data)
|
||||||
return conn.ExecCtx(ctx, query, {{.expressionValues}})
|
}, {{.keyValues}}){{else}}err:=m.conn.WithContext(ctx).Save(data).Error{{end}}
|
||||||
}, {{.keyValues}}){{else}}query := fmt.Sprintf("update %s set %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}}", m.table, {{.lowerStartCamelObject}}RowsWithPlaceHolder)
|
|
||||||
_,err:=m.conn.ExecCtx(ctx, query, {{.expressionValues}}){{end}}
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue