Skip to content
GitLab
Select Git revision
  • main default
  • gogintemplate-dockerized
Find file Blame History Permalink
grpcservermdl.go 4.69 KiB
1 
package routebuildermdl
2
3 
import (
4 
	"context"
5 
	"encoding/json"
6 
	"net"
7
8 
	"google.golang.org/grpc"
9
10 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/authmdl/jwtmdl"
11 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/authmdl/roleenforcemdl"
12 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/errormdl"
13 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/grpcbuildermdl"
14 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/loggermdl"
15 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/servicebuildermdl"
16 
)
17
18 
// Server server
19 
type Server struct{}
20
21 
// GRPCInit init
22 
func GRPCInit(GRPCPort net.Listener) {
23 
	loggermdl.LogInfo("In GRPCInit")
24 
	maxMsgSize := 50 * 1024 * 1024
25 
	s := grpc.NewServer(grpc.MaxRecvMsgSize(maxMsgSize), grpc.MaxSendMsgSize(maxMsgSize))
26 
	grpcbuildermdl.RegisterGRPCCheckServer(s, &Server{})
27 
	grpcbuildermdl.RegisterGRPCServiceServer(s, &Server{})
28 
	if err := s.Serve(GRPCPort); err != nil {
29 
		loggermdl.LogError("Unable to start GRPC server: %v", err)
30 
	}
31 
	loggermdl.LogError("GRPC server started on :", GRPCPort.Addr().String())
32 
}
33
34 
// GRPCHandler GRPCHandler
35 
func (*Server) GRPCHandler(ctx context.Context, req *grpcbuildermdl.GRPCRequest) (*grpcbuildermdl.GRPCByteResponse, error) {
36 
	loggermdl.LogError("GRPC Handler inoked:")
37 
	principal := servicebuildermdl.Principal{}
38 
	errExecutingActivity := ""
39 
	if req.GetGrpcMessage().GetIsRestricted() {
40 
		claim, _ := jwtmdl.GeneratePricipleObjUsingToken(req.GetGrpcMessage().GetToken(), jwtmdl.GlobalJWTKey)
41 
		groups, grperr := roleenforcemdl.GetGroupNames(claim, "groups")
42 
		if errormdl.CheckErr(grperr) != nil {
43 
			loggermdl.LogError("Error accessing group", grperr)
44 
			return nil, errormdl.CheckErr(grperr)
45 
		}
46 
		userID, ok := claim["userId"].(string)
47 
		if !ok || len(userID) < 2 {
48 
			loggermdl.LogError("Unable to parse UserID from JWT Token")
49 
			return nil, errormdl.Wrap("Unable to parse UserID from JWT Token")
50 
		}
51 
		rawMetadata, ok := claim["metadata"]
52 
		if ok {
53 
			metadata, ok := rawMetadata.(string)
54 
			if !ok {
55 
				loggermdl.LogError("Unable to parse metadata from JWT Token")
56 
				return nil, errormdl.Wrap("Unable to parse metadata from JWT Token")
57 
			}
58 
			principal.Metadata = metadata
59 
		}
60 
		principal.Groups = groups
61 
		principal.UserID = userID
62 
		principal.Token = req.GetGrpcMessage().GetToken()
63 
	}
64
65 
	//TODO: call executeServiveBranchWise
66
67 
	// result, _, _, errorCode, err := executeService(
68 
	// 	req.GetGrpcMessage().GetName(),
69 
	// 	req.GetGrpcMessage().GetData(), req.GetGrpcMessage().GetIsRestricted(),
70 
	// 	req.GetGrpcMessage().GetIsRoleBased(), false, principal)
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
result, _, _, errorCode, err := executeServiceWithBranch( req.GetGrpcMessage().GetName(), req.GetGrpcMessage().GetBranch(), req.GetGrpcMessage().GetData(), req.GetGrpcMessage().GetIsRestricted(), req.GetGrpcMessage().GetIsRoleBased(), false, principal) e, _ := json.Marshal(result) if err != nil { errExecutingActivity = err.Error() } res := &grpcbuildermdl.GRPCByteResponse{ Data: e, ErrorCode: int32(errorCode), Error: errExecutingActivity, } return res, nil } // GRPCCheck GRPCCheck func (*Server) GRPCCheck(ctx context.Context, req *grpcbuildermdl.GRPCRequest) (*grpcbuildermdl.GRPCResponse, error) { claim, _ := jwtmdl.GeneratePricipleObjUsingToken(req.GetGrpcMessage().GetToken(), jwtmdl.GlobalJWTKey) principal := servicebuildermdl.Principal{} groups, grperr := roleenforcemdl.GetGroupNames(claim, "groups") if errormdl.CheckErr(grperr) != nil { loggermdl.LogError(grperr) return nil, errormdl.CheckErr(grperr) } userID, ok := claim["userId"].(string) if !ok || len(userID) < 2 { loggermdl.LogError("Unable to parse UserID from JWT Token") return nil, errormdl.Wrap("Unable to parse UserID from JWT Token") } rawMetadata, ok := claim["metadata"] if ok { metadata, ok := rawMetadata.(string) if !ok { loggermdl.LogError("Unable to parse metadata from JWT Token") return nil, errormdl.Wrap("Unable to parse metadata from JWT Token") } principal.Metadata = metadata } principal.Groups = groups principal.UserID = userID principal.Token = req.GetGrpcMessage().GetToken() //TODO: remove logger loggermdl.LogError("Branch:", req.GetGrpcMessage().GetBranch()) _, _, _, _, err := executeServiceWithBranch( req.GetGrpcMessage().GetName(), req.GetGrpcMessage().GetBranch(), req.GetGrpcMessage().GetData(), req.GetGrpcMessage().GetIsRestricted(), req.GetGrpcMessage().GetIsRoleBased(), false, principal) // _, _, _, _, err := executeService( // req.GetGrpcMessage().GetName(), // req.GetGrpcMessage().GetData(), // req.GetGrpcMessage().GetIsRestricted(), // req.GetGrpcMessage().GetIsRoleBased(), false, principal) res := &grpcbuildermdl.GRPCResponse{ Data: "Response from GRPC Check service", } return res, err }