tijiao

rpc/userlib/internal/logic/doccollectlogic.go

@@ -4,10 +4,8 @@ import (
 	"app.yhyue.com/moapp/jy_docs/services/model"
 	"context"
 	"log"
-
 	"app.yhyue.com/moapp/jy_docs/rpc/userlib/internal/svc"
 	"app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib"
-
 	userLibService "app.yhyue.com/moapp/jy_docs/services/userlib"
 	"github.com/tal-tech/go-zero/core/logx"
 )

rpc/userlib/internal/logic/docdeletelogic.go

@@ -3,11 +3,10 @@ package logic
 import (
 	userLibService "app.yhyue.com/moapp/jy_docs/services/userlib"
 	"context"
-
 	"app.yhyue.com/moapp/jy_docs/rpc/userlib/internal/svc"
 	"app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib"
-
 	"github.com/tal-tech/go-zero/core/logx"
+	"log"
 )
 
 type DocDeleteLogic struct {
@@ -33,6 +32,7 @@ func (l *DocDeleteLogic) DocDelete(in *userlib.UserRecoveryRequest) (*userlib.Us
 		result.Message = msg
 		return result,nil
 	}
+	log.Printf("用户文档删除,userId:[%s],docId:[%s] 是否成功:[%v]", in.UserId, in.UserDocId, code)
 	result.Code = 0
 	result.Message = msg
 	return result, nil

rpc/userlib/internal/logic/docdownloadlogic.go

@@ -7,7 +7,6 @@ import (
 	userLibService "app.yhyue.com/moapp/jy_docs/services/userlib"
 	"context"
 	"log"
-
 	"github.com/tal-tech/go-zero/core/logx"
 )
 
@@ -33,7 +32,7 @@ func (l *DocDownloadLogic) DocDownload(in *userlib.UserCollectRequest) (*userlib
 			DocId:  in.DocId,
 			AppId:  in.AppId,
 		}, int(in.Cost))
-	log.Printf("用户文档收藏,userId:[%s],docId:[%s] 是否成功:[%v]", in.UserId, in.DocId, b)
+	log.Printf("用户文档转存,userId:[%s],docId:[%s],cost:[%d],是否成功:[%v]", in.UserId, in.DocId,in.Cost, b)
 	if b {
 		return &userlib.UserCollectResponse{
 			Code:    int32(1),

rpc/userlib/internal/logic/docpermanentdeletelogic.go

@@ -5,6 +5,7 @@ import (
 	"app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib"
 	userLibService "app.yhyue.com/moapp/jy_docs/services/userlib"
 	"context"
+	"log"
 
 	"github.com/tal-tech/go-zero/core/logx"
 )
@@ -32,6 +33,7 @@ func (l *DocPermanentDeleteLogic) DocPermanentDelete(in *userlib.UserRecoveryReq
 		result.Message = msg
 		return result,nil
 	}
+	log.Printf("用户文档永久删除,userId:[%s],docId:[%s] 是否成功:[%v]", in.UserId, in.UserDocId, code)
 	result.Code = 0
 	result.Message = msg
 	return result, nil

rpc/userlib/internal/logic/docrestorelogic.go

@@ -5,6 +5,7 @@ import (
 	"app.yhyue.com/moapp/jy_docs/rpc/userlib/userlib"
 	userLibService "app.yhyue.com/moapp/jy_docs/services/userlib"
 	"context"
+	"log"
 
 	"github.com/tal-tech/go-zero/core/logx"
 )
@@ -32,6 +33,7 @@ func (l *DocRestoreLogic) DocRestore(in *userlib.UserRecoveryRequest) (*userlib.
 		result.Message =msg
 		return result, nil
 	}
+	log.Printf("用户文档回收,userId:[%s],docId:[%s] 是否成功:[%v]", in.UserId, in.UserDocId, code)
 	result.Code = 0
 	result.Message = msg
 	return result, nil

rpc/userlib/test/userLib_test.go

@@ -142,7 +142,7 @@ func Test_UserDocDwnload(t *testing.T) {
 	userLib := userlibclient.NewUserLib(client)
 	resp, err := userLib.DocDownload(context.Background(), &userlibclient.UserCollectRequest{
 		DocId:  "123",
-		UserId: "wh5",
+		UserId: "wh8",
 		Cost:   1000,
 		AppId:  "10000",
 	})

rpc/userlib/userlib.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	jyDocsRpcUtil "app.yhyue.com/moapp/jy_docs/services/util"
+	"context"
 	"flag"
 	"fmt"
 	"log"
@@ -27,6 +28,7 @@ func main() {
 		s := zrpc.MustNewServer(config.Configs.RpcServerConf, func(grpcServer *grpc.Server) {
 			userlib.RegisterUserLibServer(grpcServer, srv)
 		})
+		s.AddUnaryInterceptors(rateLimitInterceptor)
 		defer s.Stop()
 		fmt.Printf("Starting rpc server at %s...\n", config.Configs.ListenOn)
 		s.Start()
@@ -34,3 +36,8 @@ func main() {
 		log.Fatal("【jy_docs】 read config error!")
 	}
 }
+func rateLimitInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
+	fmt.Println("拦截器")
+	log.Println("方法名:",info.FullMethod,"  参数：",req)
+	return handler(ctx, req)
+}