Skip to content
GitLab
Select Git revision
  • rb_enableResponseHeaders
  • pre-dev
  • staging default
  • devbranch
  • ps_redisOtpFunctionality
  • SG_Conflict_new
  • SG_ConflictResolved
  • rb_conflictsResolved
  • SG_Conflict_resolved
  • ps_redisOtpAndAuth
  • jwtkidfeature
  • jwt-key-id-prajwalp
  • SG_mongoFunctions
  • SG_Mysql_TestCaseMongo
  • SG_Go1.21.10
  • ps_redisAuthCorepkg
  • rb_jwtClaimsChange
  • SG_AggregateOptions
  • coreimmudb-connection-failure-logger
  • SG_Findops
  • v1.2.17
  • v1.2.16
  • v1.2.15
  • v1.2.14
  • v1.2.13
  • v1.2.12
  • v1.2.11
  • v1.2.10
  • v1.2.9
  • v1.2.8
  • v1.2.7
  • v1.2.6
  • v1.2.5
  • v1.2.4
  • v1.2.3
  • v1.2.2
  • v1.2.1
  • v1.1.3
  • v1.2.0
  • v2.0.0-beta.1
Find file Blame History Permalink
routebuilder_fasthttp.go 10.43 KiB
1 
//  +build fasthttp
2
3 
package routebuildermdl
4
5 
import (
6 
	"context"
7 
	"coresls/servers/coresls/app/modules/constantmdl"
8 
	"strings"
9
10 
	"github.com/pquerna/ffjson/ffjson"
11
12 
	routing "github.com/qiangxue/fasthttp-routing"
13
14 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/statemdl"
15
16 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/authmdl/jwtmdl"
17 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/authmdl/roleenforcemdl"
18 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/errormdl"
19 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/loggermdl"
20 
	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/servicebuildermdl"
21 
	version "github.com/hashicorp/go-version"
22 
	"github.com/tidwall/gjson"
23 
)
24
25 
// Init routing init
26 
func Init(o, r, c *routing.RouteGroup, JWTKey string) {
27 
	o.Post("/mql/state", statemdl.StateHandler)
28 
	o.Post("/mql", OpenHandler)
29 
	r.Post("/mql", RestrictedHandler)
30 
	c.Post("/mql", RoleBasedHandler)
31 
	o.Post("/heavymql", HeavyOpenHandler)
32 
	r.Post("/heavymql", HeavyRestrictedHandler)
33 
	c.Post("/heavymql", HeavyRoleBasedHandler)
34 
	jwtmdl.GlobalJWTKey = JWTKey
35 
}
36
37 
func commonHandler(c *routing.Context, isRestricted, isRoleBased, heavyDataActivity bool, principalObj servicebuildermdl.Principal) error {
38 
	serviceHeader := string(c.Request.Header.Peek("Service-Header"))
39 
	services := strings.Split(serviceHeader, ",")
40 
	versionError := appVersioning(c)
41 
	if versionError != nil {
42 
		_, err := c.WriteString(versionError.Error())
43 
		c.SetStatusCode(417)
44 
		return err
45 
	}
46 
	responseMap := make(map[string]responseData)
47 
	var reqBody []byte
48
49 
	reqBody = c.Request.Body()
50 
	requestBody := gjson.ParseBytes(reqBody)
51 
	for i := 0; i < len(services); i++ {
52 
		responseDataObj := responseData{}
53 
		service := services[i]
54 
		result, ab, isCompressed, errorCode, err := executeService(service, []byte(requestBody.Get(service).String()), isRestricted, isRoleBased, heavyDataActivity, principalObj)
55 
		if errormdl.CheckErr1(err) != nil {
56 
			if ab == nil {
57 
				responseDataObj.ErrorCode = errorCode
58 
				responseDataObj.Error = err.Error()
59 
			} else {
60 
				responseDataObj.Error = ab.GetErrorData()
61 
				if responseDataObj.Error == nil {
62 
					responseDataObj.Error = err.Error()
63 
				}
64 
				errorCode := ab.GetErrorCode()
65 
				if errorCode == 0 {
66 
					errorCode = errormdl.EXPECTATIONFAILED
67 
				}
68 
				responseDataObj.ErrorCode = errorCode
69 
				if ab.TransactionEnable {
70 
					loggermdl.LogError("transaction enabled rollback")
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
var err error // database transaction rollback if transaction is enabled switch ab.DatabaseType { case constantmdl.MYSQL: if ab.TXN != nil { loggermdl.LogError("MYSQL Transaction Rollbacked") err = ab.TXN.Rollback() if err != nil { responseDataObj.Error = err.Error() responseDataObj.ErrorCode = errormdl.MYSQLERROR } } case constantmdl.SQLSERVER: if ab.SQLServerTXN != nil { loggermdl.LogError("SQLSERVER Transaction Rollbacked") err = ab.SQLServerTXN.Rollback() if err != nil { responseDataObj.Error = err.Error() responseDataObj.ErrorCode = errormdl.MYSQLERROR } } case constantmdl.GraphDB: if ab.GraphDbTXN != nil { loggermdl.LogError("GRAPHDB Transaction Rollbacked") err = ab.GraphDbTXN.Discard(context.TODO()) if err != nil { responseDataObj.Error = err.Error() responseDataObj.ErrorCode = errormdl.GRAPHDBERROR } } default: loggermdl.LogError("Invalid database type while rollback transaction") } } } } else { if ab != nil { if ab.TransactionEnable { var err error switch ab.DatabaseType { case constantmdl.MYSQL: if ab.TXN != nil { loggermdl.LogError("MYSQL Transaction Commit") err = ab.TXN.Commit() if err != nil { responseDataObj.Error = err.Error() responseDataObj.ErrorCode = errormdl.MYSQLERROR } else { responseDataObj.Result = result responseDataObj.ErrorCode = errormdl.NOERROR } } case constantmdl.SQLSERVER: if ab.SQLServerTXN != nil { loggermdl.LogError("SQLSERVER Transaction Commit") err = ab.SQLServerTXN.Commit() if err != nil { responseDataObj.Error = err.Error() responseDataObj.ErrorCode = errormdl.SQLSERVERERROR } else { responseDataObj.Result = result
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
responseDataObj.ErrorCode = errormdl.NOERROR } } case constantmdl.GraphDB: if ab.SQLServerTXN != nil { loggermdl.LogError("GRAPHDB Transaction Commit") err = ab.GraphDbTXN.Commit(context.TODO()) if err != nil { responseDataObj.Error = err.Error() responseDataObj.ErrorCode = errormdl.GRAPHDBERROR } else { responseDataObj.Result = result responseDataObj.ErrorCode = errormdl.NOERROR } } default: loggermdl.LogError("Invalid database type while commit transaction") } } else { responseDataObj.Result = result responseDataObj.ErrorCode = errormdl.NOERROR } } else { responseDataObj.Result = result responseDataObj.ErrorCode = errormdl.NOERROR } } responseDataObj.IsCompressed = isCompressed responseDataObj = formatResponse(ab, responseDataObj) responseMap[service] = responseDataObj // Token extraction // if ab != nil { // token, ok := ab.GetDataString("MQLToken") // if !ok { // token = string(c.Request.Header.Peek("Authorization")) // token = strings.TrimPrefix(token, "Bearer") // token = strings.TrimSpace(token) // c.Response.Header.Set("Authorization", token) // } // c.Response.Header.Set("Authorization", token) // } else { // token := string(c.Request.Header.Peek("Authorization")) // token = strings.TrimPrefix(token, "Bearer") // token = strings.TrimSpace(token) // c.Response.Header.Set("Authorization", token) // } if ab != nil { token, ok := ab.GetDataString("MQLToken") if ok { c.Response.Header.Set("Authorization", token) } } } ba, _ := ffjson.Marshal(responseMap) _, err := c.Write(ba) c.SetStatusCode(200) return err } // OpenHandler for /o func OpenHandler(c *routing.Context) error { c.Response.Header.Set("content-type", "application/json") principal := servicebuildermdl.Principal{} principal.ClientIP = c.RemoteIP().String() commonHandler(c, false, false, false, principal)
211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280
return nil } // RestrictedHandler for /r func RestrictedHandler(c *routing.Context) error { c.Response.Header.Set("content-type", "application/json") pricipalObj, extractError := extractPricipalObject(c) if extractError != nil { loggermdl.LogError(extractError) _, err := c.WriteString(extractError.Error()) c.SetStatusCode(412) return err } pricipalObj.ClientIP = c.RemoteIP().String() commonHandler(c, true, false, false, pricipalObj) return nil } // RoleBasedHandler for /r/c func RoleBasedHandler(c *routing.Context) error { c.Response.Header.Set("content-type", "application/json") pricipalObj, extractError := extractPricipalObject(c) if extractError != nil { loggermdl.LogError(extractError) _, err := c.WriteString(extractError.Error()) c.SetStatusCode(412) return err } pricipalObj.ClientIP = c.RemoteIP().String() commonHandler(c, true, true, false, pricipalObj) return nil } // HeavyOpenHandler for /o func HeavyOpenHandler(c *routing.Context) error { c.Response.Header.Set("content-type", "application/json") principal := servicebuildermdl.Principal{} principal.ClientIP = c.RemoteIP().String() commonHandler(c, false, false, true, principal) return nil } // HeavyRestrictedHandler for /r func HeavyRestrictedHandler(c *routing.Context) error { c.Response.Header.Set("content-type", "application/json") pricipalObj, extractError := extractPricipalObject(c) if extractError != nil { loggermdl.LogError(extractError) _, err := c.WriteString(extractError.Error()) c.SetStatusCode(412) return err } pricipalObj.ClientIP = c.RemoteIP().String() commonHandler(c, true, false, true, pricipalObj) return nil } // HeavyRoleBasedHandler for /r/c func HeavyRoleBasedHandler(c *routing.Context) error { c.Response.Header.Set("content-type", "application/json") pricipalObj, extractError := extractPricipalObject(c) if extractError != nil { loggermdl.LogError(extractError) _, err := c.WriteString(extractError.Error()) c.SetStatusCode(412) return err } pricipalObj.ClientIP = c.RemoteIP().String() commonHandler(c, true, true, true, pricipalObj)
281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350
return nil } func appVersioning(c *routing.Context) error { if isAppVersionEnabled { appVersion := string(c.Request.Header.Peek("app-version")) if appVersion == "" { return errormdl.Wrap("No App version Found in request header") } ver, err := version.NewVersion(appVersion) if errormdl.CheckErr(err) != nil { return errormdl.CheckErr(err) } if isStrictMode { if !ver.Equal(applicationVersion) { return errormdl.Wrap("Application version mismatched") } } else { if ver.GreaterThan(applicationVersion) { return errormdl.Wrap("Server Version is outdated") } if ver.LessThan(minimumSupportedVersion) { return errormdl.Wrap("Client Version is outdated") } } } return nil } func extractPricipalObject(c *routing.Context) (servicebuildermdl.Principal, error) { principal := servicebuildermdl.Principal{} if jwtmdl.GlobalJWTKey == "" { return principal, errormdl.Wrap("No Global JWT key found") } claim, decodeError := jwtmdl.DecodeToken(&c.Request) if errormdl.CheckErr(decodeError) != nil { // loggermdl.LogError(decodeError) return principal, errormdl.CheckErr(decodeError) } groups, grperr := roleenforcemdl.GetGroupNames(claim, "groups") if errormdl.CheckErr(grperr) != nil { loggermdl.LogError(grperr) return principal, errormdl.CheckErr(grperr) } userID, _ := claim["userId"].(string) // if !ok { // loggermdl.LogError("Unable to parse UserID from JWT Token") // return principal, errormdl.Wrap("Unable to parse UserID from JWT Token") // } if len(userID) < 2 { loggermdl.LogError("UserID length is less than 2") return principal, errormdl.Wrap("UserID length is less than 2") } rawMetadata, ok := claim["metadata"] if ok { metadata, ok := rawMetadata.(string) if !ok { loggermdl.LogError("Unable to parse metadata from JWT Token") return principal, errormdl.Wrap("Unable to parse metadata from JWT Token") } principal.Metadata = metadata } principal.Groups = groups principal.UserID = userID principal.Token = string(c.Request.Header.Peek("Authorization")) return principal, nil
351352
}