Compare commits
	
		
			29 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 190584e0e6 | |||
| b7003b9ec9 | |||
| 4f871271e8 | |||
| 91f4793678 | |||
| 3b30bb049e | |||
| f0c5b36ea9 | |||
| 647ec64c3b | |||
| b5f9b6b638 | |||
| c7949febf2 | |||
| 15a4b2a713 | |||
| 493c6ebae8 | |||
| fb847b03af | |||
| f826633e6e | |||
| edeae23bf1 | |||
| a038b86147 | |||
| ede0b99d3a | |||
| d04ce18eb0 | |||
| 8ae9a0f107 | |||
| a259bb6dbc | |||
| adf32568ee | |||
| 0cfa159cb1 | |||
| 0ead99608a | |||
| 7fe3e66cad | |||
| a73d7d1654 | |||
| bbd7a7bc2c | |||
| f5151eb214 | |||
| eefb9ac9f5 | |||
| 468a7d212d | |||
| a4def75d06 | 
| @@ -31,13 +31,13 @@ type EnumDef struct { | |||||||
| 	Values       []EnumDefVal | 	Values       []EnumDefVal | ||||||
| } | } | ||||||
|  |  | ||||||
| var rexPackage = rext.W(regexp.MustCompile("^package\\s+(?P<name>[A-Za-z0-9_]+)\\s*$")) | var rexPackage = rext.W(regexp.MustCompile(`^package\s+(?P<name>[A-Za-z0-9_]+)\s*$`)) | ||||||
|  |  | ||||||
| var rexEnumDef = rext.W(regexp.MustCompile("^\\s*type\\s+(?P<name>[A-Za-z0-9_]+)\\s+(?P<type>[A-Za-z0-9_]+)\\s*//\\s*(@enum:type).*$")) | var rexEnumDef = rext.W(regexp.MustCompile(`^\s*type\s+(?P<name>[A-Za-z0-9_]+)\s+(?P<type>[A-Za-z0-9_]+)\s*//\s*(@enum:type).*$`)) | ||||||
|  |  | ||||||
| var rexValueDef = rext.W(regexp.MustCompile("^\\s*(?P<name>[A-Za-z0-9_]+)\\s+(?P<type>[A-Za-z0-9_]+)\\s*=\\s*(?P<value>(\"[A-Za-z0-9_:]+\"|[0-9]+))\\s*(//(?P<descr>.*))?.*$")) | var rexValueDef = rext.W(regexp.MustCompile(`^\s*(?P<name>[A-Za-z0-9_]+)\s+(?P<type>[A-Za-z0-9_]+)\s*=\s*(?P<value>("[A-Za-z0-9_:]+"|[0-9]+))\s*(//(?P<descr>.*))?.*$`)) | ||||||
|  |  | ||||||
| var rexChecksumConst = rext.W(regexp.MustCompile("const ChecksumGenerator = \"(?P<cs>[A-Za-z0-9_]*)\"")) | var rexChecksumConst = rext.W(regexp.MustCompile(`const ChecksumGenerator = "(?P<cs>[A-Za-z0-9_]*)"`)) | ||||||
|  |  | ||||||
| func GenerateEnumSpecs(sourceDir string, destFile string) error { | func GenerateEnumSpecs(sourceDir string, destFile string) error { | ||||||
|  |  | ||||||
|   | |||||||
| @@ -13,6 +13,7 @@ var ( | |||||||
| 	CatForeign = ErrorCategory{"Foreign"} // A foreign error that some component threw (e.g. an unknown mongodb error), happens if we call Wrap(..) on an non-bmerror value | 	CatForeign = ErrorCategory{"Foreign"} // A foreign error that some component threw (e.g. an unknown mongodb error), happens if we call Wrap(..) on an non-bmerror value | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | //goland:noinspection GoUnusedGlobalVariable | ||||||
| var AllCategories = []ErrorCategory{CatWrap, CatSystem, CatUser, CatForeign} | var AllCategories = []ErrorCategory{CatWrap, CatSystem, CatUser, CatForeign} | ||||||
|  |  | ||||||
| type ErrorSeverity struct{ Severity string } | type ErrorSeverity struct{ Severity string } | ||||||
| @@ -26,6 +27,7 @@ var ( | |||||||
| 	SevFatal = ErrorSeverity{"Fatal"} | 	SevFatal = ErrorSeverity{"Fatal"} | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | //goland:noinspection GoUnusedGlobalVariable | ||||||
| var AllSeverities = []ErrorSeverity{SevTrace, SevDebug, SevInfo, SevWarn, SevErr, SevFatal} | var AllSeverities = []ErrorSeverity{SevTrace, SevDebug, SevInfo, SevWarn, SevErr, SevFatal} | ||||||
|  |  | ||||||
| type ErrorType struct { | type ErrorType struct { | ||||||
| @@ -33,6 +35,7 @@ type ErrorType struct { | |||||||
| 	DefaultStatusCode *int | 	DefaultStatusCode *int | ||||||
| } | } | ||||||
|  |  | ||||||
|  | //goland:noinspection GoUnusedGlobalVariable | ||||||
| var ( | var ( | ||||||
| 	TypeInternal       = ErrorType{"INTERNAL_ERROR", langext.Ptr(500)} | 	TypeInternal       = ErrorType{"INTERNAL_ERROR", langext.Ptr(500)} | ||||||
| 	TypePanic          = ErrorType{"PANIC", langext.Ptr(500)} | 	TypePanic          = ErrorType{"PANIC", langext.Ptr(500)} | ||||||
| @@ -44,6 +47,7 @@ var ( | |||||||
| 	TypeBindFailQuery    = ErrorType{"BINDFAIL_QUERY", langext.Ptr(400)} | 	TypeBindFailQuery    = ErrorType{"BINDFAIL_QUERY", langext.Ptr(400)} | ||||||
| 	TypeBindFailJSON     = ErrorType{"BINDFAIL_JSON", langext.Ptr(400)} | 	TypeBindFailJSON     = ErrorType{"BINDFAIL_JSON", langext.Ptr(400)} | ||||||
| 	TypeBindFailFormData = ErrorType{"BINDFAIL_FORMDATA", langext.Ptr(400)} | 	TypeBindFailFormData = ErrorType{"BINDFAIL_FORMDATA", langext.Ptr(400)} | ||||||
|  | 	TypeBindFailHeader   = ErrorType{"BINDFAIL_HEADER", langext.Ptr(400)} | ||||||
|  |  | ||||||
| 	TypeUnauthorized = ErrorType{"UNAUTHORIZED", langext.Ptr(401)} | 	TypeUnauthorized = ErrorType{"UNAUTHORIZED", langext.Ptr(401)} | ||||||
| 	TypeAuthFailed   = ErrorType{"AUTH_FAILED", langext.Ptr(401)} | 	TypeAuthFailed   = ErrorType{"AUTH_FAILED", langext.Ptr(401)} | ||||||
| @@ -51,6 +55,10 @@ var ( | |||||||
| 	// other values come from pkgconfig | 	// other values come from pkgconfig | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | func NewType(key string, defStatusCode *int) ErrorType { | ||||||
|  | 	return ErrorType{key, defStatusCode} | ||||||
|  | } | ||||||
|  |  | ||||||
| type LogPrintLevel string | type LogPrintLevel string | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
|   | |||||||
| @@ -10,8 +10,8 @@ type ErrorPackageConfig struct { | |||||||
| 	ZeroLogAllTraces    bool                                             // autom print zerolog logs on .Build()  (for all Severities) | 	ZeroLogAllTraces    bool                                             // autom print zerolog logs on .Build()  (for all Severities) | ||||||
| 	RecursiveErrors     bool                                             // errors contains their Origin-Error | 	RecursiveErrors     bool                                             // errors contains their Origin-Error | ||||||
| 	ExtendedGinOutput   bool                                             // Log extended data (trace, meta, ...) to gin in err.Output() | 	ExtendedGinOutput   bool                                             // Log extended data (trace, meta, ...) to gin in err.Output() | ||||||
| 	ExtendGinOutput     func(json map[string]any) // (Optionally) extend the gin output with more fields | 	ExtendGinOutput     func(err *ExErr, json map[string]any)            // (Optionally) extend the gin output with more fields | ||||||
| 	ExtendGinDataOutput func(json map[string]any) // (Optionally) extend the gin `__data` output with more fields | 	ExtendGinDataOutput func(err *ExErr, depth int, json map[string]any) // (Optionally) extend the gin `__data` output with more fields | ||||||
| } | } | ||||||
|  |  | ||||||
| type ErrorPackageConfigInit struct { | type ErrorPackageConfigInit struct { | ||||||
| @@ -19,8 +19,8 @@ type ErrorPackageConfigInit struct { | |||||||
| 	ZeroLogAllTraces    bool | 	ZeroLogAllTraces    bool | ||||||
| 	RecursiveErrors     bool | 	RecursiveErrors     bool | ||||||
| 	ExtendedGinOutput   bool | 	ExtendedGinOutput   bool | ||||||
| 	ExtendGinOutput     *func(json map[string]any) | 	ExtendGinOutput     func(err *ExErr, json map[string]any) | ||||||
| 	ExtendGinDataOutput *func(json map[string]any) | 	ExtendGinDataOutput func(err *ExErr, depth int, json map[string]any) | ||||||
| } | } | ||||||
|  |  | ||||||
| var initialized = false | var initialized = false | ||||||
| @@ -30,8 +30,8 @@ var pkgconfig = ErrorPackageConfig{ | |||||||
| 	ZeroLogAllTraces:    false, | 	ZeroLogAllTraces:    false, | ||||||
| 	RecursiveErrors:     true, | 	RecursiveErrors:     true, | ||||||
| 	ExtendedGinOutput:   false, | 	ExtendedGinOutput:   false, | ||||||
| 	ExtendGinOutput:     func(json map[string]any) {}, | 	ExtendGinOutput:     func(err *ExErr, json map[string]any) {}, | ||||||
| 	ExtendGinDataOutput: func(json map[string]any) {}, | 	ExtendGinDataOutput: func(err *ExErr, depth int, json map[string]any) {}, | ||||||
| } | } | ||||||
|  |  | ||||||
| // Init initializes the exerr packages | // Init initializes the exerr packages | ||||||
| @@ -42,13 +42,23 @@ func Init(cfg ErrorPackageConfigInit) { | |||||||
| 		panic("Cannot re-init error package") | 		panic("Cannot re-init error package") | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	ego := func(err *ExErr, json map[string]any) {} | ||||||
|  | 	egdo := func(err *ExErr, depth int, json map[string]any) {} | ||||||
|  |  | ||||||
|  | 	if cfg.ExtendGinOutput != nil { | ||||||
|  | 		ego = cfg.ExtendGinOutput | ||||||
|  | 	} | ||||||
|  | 	if cfg.ExtendGinDataOutput != nil { | ||||||
|  | 		egdo = cfg.ExtendGinDataOutput | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	pkgconfig = ErrorPackageConfig{ | 	pkgconfig = ErrorPackageConfig{ | ||||||
| 		ZeroLogErrTraces:    cfg.ZeroLogErrTraces, | 		ZeroLogErrTraces:    cfg.ZeroLogErrTraces, | ||||||
| 		ZeroLogAllTraces:    cfg.ZeroLogAllTraces, | 		ZeroLogAllTraces:    cfg.ZeroLogAllTraces, | ||||||
| 		RecursiveErrors:     cfg.RecursiveErrors, | 		RecursiveErrors:     cfg.RecursiveErrors, | ||||||
| 		ExtendedGinOutput:   cfg.ExtendedGinOutput, | 		ExtendedGinOutput:   cfg.ExtendedGinOutput, | ||||||
| 		ExtendGinOutput:     langext.Coalesce(cfg.ExtendGinOutput, func(json map[string]any) {}), | 		ExtendGinOutput:     ego, | ||||||
| 		ExtendGinDataOutput: langext.Coalesce(cfg.ExtendGinDataOutput, func(json map[string]any) {}), | 		ExtendGinDataOutput: egdo, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	initialized = true | 	initialized = true | ||||||
|   | |||||||
| @@ -162,7 +162,6 @@ func (ee *ExErr) RecursiveStatuscode() *int { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// fallback to <empty> |  | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -179,6 +178,18 @@ func (ee *ExErr) RecursiveCategory() ErrorCategory { | |||||||
| 	return ee.Category | 	return ee.Category | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // RecursiveMeta searches (top-down) for teh first error that has a meta value with teh specified key | ||||||
|  | // and returns its value (or nil) | ||||||
|  | func (ee *ExErr) RecursiveMeta(key string) *MetaValue { | ||||||
|  | 	for curr := ee; curr != nil; curr = curr.OriginalError { | ||||||
|  | 		if metaval, ok := curr.Meta[key]; ok { | ||||||
|  | 			return langext.Ptr(metaval) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| func (ee *ExErr) Depth() int { | func (ee *ExErr) Depth() int { | ||||||
| 	if ee.OriginalError == nil { | 	if ee.OriginalError == nil { | ||||||
| 		return 1 | 		return 1 | ||||||
|   | |||||||
							
								
								
									
										75
									
								
								exerr/gin.go
									
									
									
									
									
								
							
							
						
						
									
										75
									
								
								exerr/gin.go
									
									
									
									
									
								
							| @@ -3,50 +3,82 @@ package exerr | |||||||
| import ( | import ( | ||||||
| 	"github.com/gin-gonic/gin" | 	"github.com/gin-gonic/gin" | ||||||
| 	json "gogs.mikescher.com/BlackForestBytes/goext/gojson" | 	json "gogs.mikescher.com/BlackForestBytes/goext/gojson" | ||||||
|  | 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||||
| 	"net/http" | 	"net/http" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func (ee *ExErr) toJson() gin.H { | func (ee *ExErr) toJson(depth int, applyExtendListener bool) langext.H { | ||||||
| 	json := gin.H{} | 	ginJson := langext.H{} | ||||||
|  |  | ||||||
| 	if ee.UniqueID != "" { | 	if ee.UniqueID != "" { | ||||||
| 		json["id"] = ee.UniqueID | 		ginJson["id"] = ee.UniqueID | ||||||
| 	} | 	} | ||||||
| 	if ee.Category != CatWrap { | 	if ee.Category != CatWrap { | ||||||
| 		json["category"] = ee.Category | 		ginJson["category"] = ee.Category | ||||||
| 	} | 	} | ||||||
| 	if ee.Type != TypeWrap { | 	if ee.Type != TypeWrap { | ||||||
| 		json["type"] = ee.Type | 		ginJson["type"] = ee.Type | ||||||
| 	} | 	} | ||||||
| 	if ee.StatusCode != nil { | 	if ee.StatusCode != nil { | ||||||
| 		json["statuscode"] = ee.StatusCode | 		ginJson["statuscode"] = ee.StatusCode | ||||||
| 	} | 	} | ||||||
| 	if ee.Message != "" { | 	if ee.Message != "" { | ||||||
| 		json["message"] = ee.Message | 		ginJson["message"] = ee.Message | ||||||
| 	} | 	} | ||||||
| 	if ee.Caller != "" { | 	if ee.Caller != "" { | ||||||
| 		json["caller"] = ee.Caller | 		ginJson["caller"] = ee.Caller | ||||||
| 	} | 	} | ||||||
| 	if ee.Severity != SevErr { | 	if ee.Severity != SevErr { | ||||||
| 		json["severity"] = ee.Severity | 		ginJson["severity"] = ee.Severity | ||||||
| 	} | 	} | ||||||
| 	if ee.Timestamp != (time.Time{}) { | 	if ee.Timestamp != (time.Time{}) { | ||||||
| 		json["time"] = ee.Timestamp.Format(time.RFC3339) | 		ginJson["time"] = ee.Timestamp.Format(time.RFC3339) | ||||||
| 	} | 	} | ||||||
| 	if ee.WrappedErrType != "" { | 	if ee.WrappedErrType != "" { | ||||||
| 		json["wrappedErrType"] = ee.WrappedErrType | 		ginJson["wrappedErrType"] = ee.WrappedErrType | ||||||
| 	} | 	} | ||||||
| 	if ee.OriginalError != nil { | 	if ee.OriginalError != nil { | ||||||
| 		json["original"] = ee.OriginalError.toJson() | 		ginJson["original"] = ee.OriginalError.toJson(depth+1, applyExtendListener) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	pkgconfig.ExtendGinDataOutput(json) | 	if applyExtendListener { | ||||||
|  | 		pkgconfig.ExtendGinDataOutput(ee, depth, ginJson) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	return json | 	return ginJson | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // ToAPIJson converts the ExError to a json object | ||||||
|  | // (the same object as used in the Output(gin) method) | ||||||
|  | // | ||||||
|  | // Parameters: | ||||||
|  | //   - [applyExtendListener]:  if false the pkgconfig.ExtendGinOutput / pkgconfig.ExtendGinDataOutput will not be applied | ||||||
|  | //   - [includeWrappedErrors]: if false we do not include the recursive/wrapped errors in `__data` | ||||||
|  | func (ee *ExErr) ToAPIJson(applyExtendListener bool, includeWrappedErrors bool) langext.H { | ||||||
|  |  | ||||||
|  | 	apiOutput := langext.H{ | ||||||
|  | 		"errorid":   ee.UniqueID, | ||||||
|  | 		"message":   ee.RecursiveMessage(), | ||||||
|  | 		"errorcode": ee.RecursiveType().Key, | ||||||
|  | 		"category":  ee.RecursiveCategory().Category, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if includeWrappedErrors { | ||||||
|  | 		apiOutput["__data"] = ee.toJson(0, applyExtendListener) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if applyExtendListener { | ||||||
|  | 		pkgconfig.ExtendGinOutput(ee, apiOutput) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return apiOutput | ||||||
| } | } | ||||||
|  |  | ||||||
| func (ee *ExErr) Output(g *gin.Context) { | func (ee *ExErr) Output(g *gin.Context) { | ||||||
|  |  | ||||||
|  | 	warnOnPkgConfigNotInitialized() | ||||||
|  |  | ||||||
| 	var statuscode = http.StatusInternalServerError | 	var statuscode = http.StatusInternalServerError | ||||||
|  |  | ||||||
| 	var baseCat = ee.RecursiveCategory() | 	var baseCat = ee.RecursiveCategory() | ||||||
| @@ -65,20 +97,7 @@ func (ee *ExErr) Output(g *gin.Context) { | |||||||
| 		statuscode = *baseType.DefaultStatusCode | 		statuscode = *baseType.DefaultStatusCode | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	warnOnPkgConfigNotInitialized() | 	ginOutput := ee.ToAPIJson(true, pkgconfig.ExtendedGinOutput) | ||||||
|  |  | ||||||
| 	ginOutput := gin.H{ |  | ||||||
| 		"errorid":   ee.UniqueID, |  | ||||||
| 		"message":   ee.RecursiveMessage(), |  | ||||||
| 		"errorcode": ee.RecursiveType(), |  | ||||||
| 		"category":  ee.RecursiveCategory(), |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if pkgconfig.ExtendedGinOutput { |  | ||||||
| 		ginOutput["__data"] = ee.toJson() |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	pkgconfig.ExtendGinOutput(ginOutput) |  | ||||||
|  |  | ||||||
| 	g.Render(statuscode, json.GoJsonRender{Data: ginOutput, NilSafeSlices: true, NilSafeMaps: true}) | 	g.Render(statuscode, json.GoJsonRender{Data: ginOutput, NilSafeSlices: true, NilSafeMaps: true}) | ||||||
| } | } | ||||||
|   | |||||||
| @@ -14,6 +14,9 @@ type AppContext struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| func CreateAppContext(g *gin.Context, innerCtx context.Context, cancelFn context.CancelFunc) *AppContext { | func CreateAppContext(g *gin.Context, innerCtx context.Context, cancelFn context.CancelFunc) *AppContext { | ||||||
|  | 	for key, value := range g.Keys { | ||||||
|  | 		innerCtx = context.WithValue(innerCtx, key, value) | ||||||
|  | 	} | ||||||
| 	return &AppContext{ | 	return &AppContext{ | ||||||
| 		inner:      innerCtx, | 		inner:      innerCtx, | ||||||
| 		cancelFunc: cancelFn, | 		cancelFunc: cancelFn, | ||||||
| @@ -38,6 +41,10 @@ func (ac *AppContext) Value(key any) any { | |||||||
| 	return ac.inner.Value(key) | 	return ac.inner.Value(key) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (ac *AppContext) Set(key, value any) { | ||||||
|  | 	ac.inner = context.WithValue(ac.inner, key, value) | ||||||
|  | } | ||||||
|  |  | ||||||
| func (ac *AppContext) Cancel() { | func (ac *AppContext) Cancel() { | ||||||
| 	ac.cancelled = true | 	ac.cancelled = true | ||||||
| 	ac.cancelFunc() | 	ac.cancelFunc() | ||||||
| @@ -50,10 +57,3 @@ func (ac *AppContext) RequestURI() string { | |||||||
| 		return "" | 		return "" | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (ac *AppContext) FinishSuccess(res HTTPResponse) HTTPResponse { |  | ||||||
| 	if ac.cancelled { |  | ||||||
| 		panic("Cannot finish a cancelled request") |  | ||||||
| 	} |  | ||||||
| 	return res |  | ||||||
| } |  | ||||||
|   | |||||||
| @@ -1,8 +1,14 @@ | |||||||
| package ginext | package ginext | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"fmt" | ||||||
| 	"github.com/gin-gonic/gin" | 	"github.com/gin-gonic/gin" | ||||||
|  | 	"github.com/rs/zerolog/log" | ||||||
|  | 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||||
|  | 	"gogs.mikescher.com/BlackForestBytes/goext/mathext" | ||||||
|  | 	"net" | ||||||
| 	"net/http" | 	"net/http" | ||||||
|  | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| @@ -13,6 +19,15 @@ type GinWrapper struct { | |||||||
| 	allowCors      bool | 	allowCors      bool | ||||||
| 	ginDebug       bool | 	ginDebug       bool | ||||||
| 	requestTimeout time.Duration | 	requestTimeout time.Duration | ||||||
|  |  | ||||||
|  | 	routeSpecs []ginRouteSpec | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type ginRouteSpec struct { | ||||||
|  | 	Method      string | ||||||
|  | 	URL         string | ||||||
|  | 	Middlewares []string | ||||||
|  | 	Handler     string | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewEngine(allowCors bool, ginDebug bool, timeout time.Duration) *GinWrapper { | func NewEngine(allowCors bool, ginDebug bool, timeout time.Duration) *GinWrapper { | ||||||
| @@ -33,18 +48,94 @@ func NewEngine(allowCors bool, ginDebug bool, timeout time.Duration) *GinWrapper | |||||||
| 		engine.Use(CorsMiddleware()) | 		engine.Use(CorsMiddleware()) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	// do not debug-print routes | ||||||
|  | 	gin.DebugPrintRouteFunc = func(_, _, _ string, _ int) {} | ||||||
|  |  | ||||||
| 	if ginDebug { | 	if ginDebug { | ||||||
|  | 		gin.SetMode(gin.ReleaseMode) | ||||||
|  |  | ||||||
| 		ginlogger := gin.Logger() | 		ginlogger := gin.Logger() | ||||||
| 		engine.Use(func(context *gin.Context) { | 		engine.Use(func(context *gin.Context) { | ||||||
| 			if !wrapper.SuppressGinLogs { | 			if !wrapper.SuppressGinLogs { | ||||||
| 				ginlogger(context) | 				ginlogger(context) | ||||||
| 			} | 			} | ||||||
| 		}) | 		}) | ||||||
|  | 	} else { | ||||||
|  | 		gin.SetMode(gin.DebugMode) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return wrapper | 	return wrapper | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinWrapper) ServeHTTP(writer http.ResponseWriter, request *http.Request) { | func (w *GinWrapper) ListenAndServeHTTP(addr string, postInit func(port string)) (chan error, *http.Server) { | ||||||
| 	w.engine.ServeHTTP(writer, request) |  | ||||||
|  | 	w.DebugPrintRoutes() | ||||||
|  |  | ||||||
|  | 	httpserver := &http.Server{ | ||||||
|  | 		Addr:    addr, | ||||||
|  | 		Handler: w.engine, | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	errChan := make(chan error) | ||||||
|  |  | ||||||
|  | 	go func() { | ||||||
|  |  | ||||||
|  | 		ln, err := net.Listen("tcp", httpserver.Addr) | ||||||
|  | 		if err != nil { | ||||||
|  | 			errChan <- err | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		_, port, err := net.SplitHostPort(ln.Addr().String()) | ||||||
|  | 		if err != nil { | ||||||
|  | 			errChan <- err | ||||||
|  | 			return | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		log.Info().Str("address", httpserver.Addr).Msg("HTTP-Server started on http://localhost:" + port) | ||||||
|  |  | ||||||
|  | 		if postInit != nil { | ||||||
|  | 			postInit(port) // the net.Listener a few lines above is at this point actually already buffering requests | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		errChan <- httpserver.Serve(ln) | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	return errChan, httpserver | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *GinWrapper) DebugPrintRoutes() { | ||||||
|  | 	if !w.ginDebug { | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	lines := make([][4]string, 0) | ||||||
|  |  | ||||||
|  | 	pad := [4]int{0, 0, 0, 0} | ||||||
|  |  | ||||||
|  | 	for _, spec := range w.routeSpecs { | ||||||
|  |  | ||||||
|  | 		line := [4]string{ | ||||||
|  | 			spec.Method, | ||||||
|  | 			spec.URL, | ||||||
|  | 			strings.Join(spec.Middlewares, " -> "), | ||||||
|  | 			spec.Handler, | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		lines = append(lines, line) | ||||||
|  |  | ||||||
|  | 		pad[0] = mathext.Max(pad[0], len(line[0])) | ||||||
|  | 		pad[1] = mathext.Max(pad[1], len(line[1])) | ||||||
|  | 		pad[2] = mathext.Max(pad[2], len(line[2])) | ||||||
|  | 		pad[3] = mathext.Max(pad[3], len(line[3])) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	for _, line := range lines { | ||||||
|  |  | ||||||
|  | 		fmt.Printf("Gin-Route: %s  %s  -->  %s  -->  %s\n", | ||||||
|  | 			langext.StrPadRight("["+line[0]+"]", " ", pad[0]+2), | ||||||
|  | 			langext.StrPadRight(line[1], " ", pad[1]), | ||||||
|  | 			langext.StrPadRight(line[2], " ", pad[2]), | ||||||
|  | 			langext.StrPadRight(line[3], " ", pad[3])) | ||||||
|  | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -25,7 +25,7 @@ func Wrap(w *GinWrapper, fn WHandlerFunc) gin.HandlerFunc { | |||||||
| 				Str("trace", stackTrace). | 				Str("trace", stackTrace). | ||||||
| 				Build() | 				Build() | ||||||
|  |  | ||||||
| 			wrap = APIError(g, err) | 			wrap = Error(err) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		if g.Writer.Written() { | 		if g.Writer.Written() { | ||||||
|   | |||||||
| @@ -2,8 +2,10 @@ package ginext | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
|  | 	"fmt" | ||||||
| 	"github.com/gin-gonic/gin" | 	"github.com/gin-gonic/gin" | ||||||
| 	"github.com/gin-gonic/gin/binding" | 	"github.com/gin-gonic/gin/binding" | ||||||
|  | 	"gogs.mikescher.com/BlackForestBytes/goext/exerr" | ||||||
| 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||||
| 	"runtime/debug" | 	"runtime/debug" | ||||||
| ) | ) | ||||||
| @@ -15,6 +17,7 @@ type PreContext struct { | |||||||
| 	query   any | 	query   any | ||||||
| 	body    any | 	body    any | ||||||
| 	form    any | 	form    any | ||||||
|  | 	header  any | ||||||
| } | } | ||||||
|  |  | ||||||
| func (pctx *PreContext) URI(uri any) *PreContext { | func (pctx *PreContext) URI(uri any) *PreContext { | ||||||
| @@ -37,36 +40,73 @@ func (pctx *PreContext) Form(form any) *PreContext { | |||||||
| 	return pctx | 	return pctx | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (pctx *PreContext) Header(header any) *PreContext { | ||||||
|  | 	pctx.header = header | ||||||
|  | 	return pctx | ||||||
|  | } | ||||||
|  |  | ||||||
| func (pctx PreContext) Start() (*AppContext, *gin.Context, *HTTPResponse) { | func (pctx PreContext) Start() (*AppContext, *gin.Context, *HTTPResponse) { | ||||||
| 	if pctx.uri != nil { | 	if pctx.uri != nil { | ||||||
| 		if err := pctx.ginCtx.ShouldBindUri(pctx.uri); err != nil { | 		if err := pctx.ginCtx.ShouldBindUri(pctx.uri); err != nil { | ||||||
| 			return nil, nil, langext.Ptr(APIError(pctx.ginCtx, commonApiErr.BindFailURI, "Failed to read uri", err)) | 			err = exerr.Wrap(err, "Failed to read uri"). | ||||||
|  | 				WithType(exerr.TypeBindFailURI). | ||||||
|  | 				Str("struct_type", fmt.Sprintf("%T", pctx.uri)). | ||||||
|  | 				Build() | ||||||
|  | 			return nil, nil, langext.Ptr(Error(err)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if pctx.query != nil { | 	if pctx.query != nil { | ||||||
| 		if err := pctx.ginCtx.ShouldBindQuery(pctx.query); err != nil { | 		if err := pctx.ginCtx.ShouldBindQuery(pctx.query); err != nil { | ||||||
| 			return nil, nil, langext.Ptr(APIError(pctx.ginCtx, commonApiErr.BindFailQuery, "Failed to read query", err)) | 			err = exerr.Wrap(err, "Failed to read query"). | ||||||
|  | 				WithType(exerr.TypeBindFailQuery). | ||||||
|  | 				Str("struct_type", fmt.Sprintf("%T", pctx.query)). | ||||||
|  | 				Build() | ||||||
|  | 			return nil, nil, langext.Ptr(Error(err)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if pctx.body != nil { | 	if pctx.body != nil { | ||||||
| 		if pctx.ginCtx.ContentType() == "application/json" { | 		if pctx.ginCtx.ContentType() == "application/json" { | ||||||
| 			if err := pctx.ginCtx.ShouldBindJSON(pctx.body); err != nil { | 			if err := pctx.ginCtx.ShouldBindJSON(pctx.body); err != nil { | ||||||
| 				return nil, nil, langext.Ptr(APIError(pctx.ginCtx, commonApiErr.BindFailJSON, "Failed to read body", err)) | 				err = exerr.Wrap(err, "Failed to read json-body"). | ||||||
|  | 					WithType(exerr.TypeBindFailJSON). | ||||||
|  | 					Str("struct_type", fmt.Sprintf("%T", pctx.body)). | ||||||
|  | 					Build() | ||||||
|  | 				return nil, nil, langext.Ptr(Error(err)) | ||||||
| 			} | 			} | ||||||
| 		} else { | 		} else { | ||||||
| 			return nil, nil, langext.Ptr(APIError(pctx.ginCtx, commonApiErr.BindFailJSON, "missing JSON body", nil)) | 			err := exerr.New(exerr.TypeBindFailJSON, "missing JSON body"). | ||||||
|  | 				Str("struct_type", fmt.Sprintf("%T", pctx.body)). | ||||||
|  | 				Build() | ||||||
|  | 			return nil, nil, langext.Ptr(Error(err)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if pctx.form != nil { | 	if pctx.form != nil { | ||||||
| 		if pctx.ginCtx.ContentType() == "multipart/form-data" { | 		if pctx.ginCtx.ContentType() == "multipart/form-data" { | ||||||
| 			if err := pctx.ginCtx.ShouldBindWith(pctx.form, binding.Form); err != nil { | 			if err := pctx.ginCtx.ShouldBindWith(pctx.form, binding.Form); err != nil { | ||||||
| 				return nil, nil, langext.Ptr(APIError(pctx.ginCtx, commonApiErr.BindFailFormData, "Failed to read multipart-form", err)) | 				err = exerr.Wrap(err, "Failed to read multipart-form"). | ||||||
|  | 					WithType(exerr.TypeBindFailFormData). | ||||||
|  | 					Str("struct_type", fmt.Sprintf("%T", pctx.form)). | ||||||
|  | 					Build() | ||||||
|  | 				return nil, nil, langext.Ptr(Error(err)) | ||||||
| 			} | 			} | ||||||
| 		} else { | 		} else { | ||||||
| 			return nil, nil, langext.Ptr(APIError(pctx.ginCtx, commonApiErr.BindFailJSON, "missing form body", nil)) | 			err := exerr.New(exerr.TypeBindFailFormData, "missing form body"). | ||||||
|  | 				Str("struct_type", fmt.Sprintf("%T", pctx.form)). | ||||||
|  | 				Build() | ||||||
|  | 			return nil, nil, langext.Ptr(Error(err)) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if pctx.header != nil { | ||||||
|  | 		if err := pctx.ginCtx.ShouldBindHeader(pctx.query); err != nil { | ||||||
|  | 			err = exerr.Wrap(err, "Failed to read header"). | ||||||
|  | 				WithType(exerr.TypeBindFailHeader). | ||||||
|  | 				Str("struct_type", fmt.Sprintf("%T", pctx.query)). | ||||||
|  | 				Build() | ||||||
|  | 			return nil, nil, langext.Ptr(Error(err)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -7,50 +7,93 @@ import ( | |||||||
| 	json "gogs.mikescher.com/BlackForestBytes/goext/gojson" | 	json "gogs.mikescher.com/BlackForestBytes/goext/gojson" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | type headerval struct { | ||||||
|  | 	Key string | ||||||
|  | 	Val string | ||||||
|  | } | ||||||
|  |  | ||||||
| type HTTPResponse interface { | type HTTPResponse interface { | ||||||
| 	Write(g *gin.Context) | 	Write(g *gin.Context) | ||||||
|  | 	WithHeader(k string, v string) HTTPResponse | ||||||
| } | } | ||||||
|  |  | ||||||
| type jsonHTTPResponse struct { | type jsonHTTPResponse struct { | ||||||
| 	statusCode int | 	statusCode int | ||||||
| 	data       any | 	data       any | ||||||
|  | 	headers    []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j jsonHTTPResponse) Write(g *gin.Context) { | func (j jsonHTTPResponse) Write(g *gin.Context) { | ||||||
|  | 	for _, v := range j.headers { | ||||||
|  | 		g.Header(v.Key, v.Val) | ||||||
|  | 	} | ||||||
| 	g.Render(j.statusCode, json.GoJsonRender{Data: j.data, NilSafeSlices: true, NilSafeMaps: true}) | 	g.Render(j.statusCode, json.GoJsonRender{Data: j.data, NilSafeSlices: true, NilSafeMaps: true}) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j jsonHTTPResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| type emptyHTTPResponse struct { | type emptyHTTPResponse struct { | ||||||
| 	statusCode int | 	statusCode int | ||||||
|  | 	headers    []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j emptyHTTPResponse) Write(g *gin.Context) { | func (j emptyHTTPResponse) Write(g *gin.Context) { | ||||||
|  | 	for _, v := range j.headers { | ||||||
|  | 		g.Header(v.Key, v.Val) | ||||||
|  | 	} | ||||||
| 	g.Status(j.statusCode) | 	g.Status(j.statusCode) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j emptyHTTPResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| type textHTTPResponse struct { | type textHTTPResponse struct { | ||||||
| 	statusCode int | 	statusCode int | ||||||
| 	data       string | 	data       string | ||||||
|  | 	headers    []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j textHTTPResponse) Write(g *gin.Context) { | func (j textHTTPResponse) Write(g *gin.Context) { | ||||||
|  | 	for _, v := range j.headers { | ||||||
|  | 		g.Header(v.Key, v.Val) | ||||||
|  | 	} | ||||||
| 	g.String(j.statusCode, "%s", j.data) | 	g.String(j.statusCode, "%s", j.data) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j textHTTPResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| type dataHTTPResponse struct { | type dataHTTPResponse struct { | ||||||
| 	statusCode  int | 	statusCode  int | ||||||
| 	data        []byte | 	data        []byte | ||||||
| 	contentType string | 	contentType string | ||||||
|  | 	headers     []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j dataHTTPResponse) Write(g *gin.Context) { | func (j dataHTTPResponse) Write(g *gin.Context) { | ||||||
|  | 	for _, v := range j.headers { | ||||||
|  | 		g.Header(v.Key, v.Val) | ||||||
|  | 	} | ||||||
| 	g.Data(j.statusCode, j.contentType, j.data) | 	g.Data(j.statusCode, j.contentType, j.data) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j dataHTTPResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| type fileHTTPResponse struct { | type fileHTTPResponse struct { | ||||||
| 	mimetype string | 	mimetype string | ||||||
| 	filepath string | 	filepath string | ||||||
| 	filename *string | 	filename *string | ||||||
|  | 	headers  []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j fileHTTPResponse) Write(g *gin.Context) { | func (j fileHTTPResponse) Write(g *gin.Context) { | ||||||
| @@ -59,26 +102,46 @@ func (j fileHTTPResponse) Write(g *gin.Context) { | |||||||
| 		g.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", *j.filename)) | 		g.Header("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"", *j.filename)) | ||||||
|  |  | ||||||
| 	} | 	} | ||||||
|  | 	for _, v := range j.headers { | ||||||
|  | 		g.Header(v.Key, v.Val) | ||||||
|  | 	} | ||||||
| 	g.File(j.filepath) | 	g.File(j.filepath) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j fileHTTPResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| type redirectHTTPResponse struct { | type redirectHTTPResponse struct { | ||||||
| 	statusCode int | 	statusCode int | ||||||
| 	url        string | 	url        string | ||||||
|  | 	headers    []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j redirectHTTPResponse) Write(g *gin.Context) { | func (j redirectHTTPResponse) Write(g *gin.Context) { | ||||||
| 	g.Redirect(j.statusCode, j.url) | 	g.Redirect(j.statusCode, j.url) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j redirectHTTPResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| type jsonAPIErrResponse struct { | type jsonAPIErrResponse struct { | ||||||
| 	err     *exerr.ExErr | 	err     *exerr.ExErr | ||||||
|  | 	headers []headerval | ||||||
| } | } | ||||||
|  |  | ||||||
| func (j jsonAPIErrResponse) Write(g *gin.Context) { | func (j jsonAPIErrResponse) Write(g *gin.Context) { | ||||||
| 	j.err.Output(g) | 	j.err.Output(g) | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (j jsonAPIErrResponse) WithHeader(k string, v string) HTTPResponse { | ||||||
|  | 	j.headers = append(j.headers, headerval{k, v}) | ||||||
|  | 	return j | ||||||
|  | } | ||||||
|  |  | ||||||
| func Status(sc int) HTTPResponse { | func Status(sc int) HTTPResponse { | ||||||
| 	return &emptyHTTPResponse{statusCode: sc} | 	return &emptyHTTPResponse{statusCode: sc} | ||||||
| } | } | ||||||
| @@ -107,12 +170,18 @@ func Redirect(sc int, newURL string) HTTPResponse { | |||||||
| 	return &redirectHTTPResponse{statusCode: sc, url: newURL} | 	return &redirectHTTPResponse{statusCode: sc, url: newURL} | ||||||
| } | } | ||||||
|  |  | ||||||
| func APIError(g *gin.Context, e error) HTTPResponse { | func Error(e error) HTTPResponse { | ||||||
| 	return &jsonAPIErrResponse{ | 	return &jsonAPIErrResponse{ | ||||||
| 		err: exerr.FromError(e), | 		err: exerr.FromError(e), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func NotImplemented(g *gin.Context) HTTPResponse { | func ErrWrap(e error, errorType exerr.ErrorType, msg string) HTTPResponse { | ||||||
| 	return APIError(g, exerr.New(exerr.TypeNotImplemented, "").Build()) | 	return &jsonAPIErrResponse{ | ||||||
|  | 		err: exerr.FromError(exerr.Wrap(e, msg).WithType(errorType).Build()), | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func NotImplemented() HTTPResponse { | ||||||
|  | 	return Error(exerr.New(exerr.TypeNotImplemented, "").Build()) | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										135
									
								
								ginext/routes.go
									
									
									
									
									
								
							
							
						
						
									
										135
									
								
								ginext/routes.go
									
									
									
									
									
								
							| @@ -2,7 +2,14 @@ package ginext | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/gin-gonic/gin" | 	"github.com/gin-gonic/gin" | ||||||
|  | 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||||
|  | 	"gogs.mikescher.com/BlackForestBytes/goext/rext" | ||||||
| 	"net/http" | 	"net/http" | ||||||
|  | 	"path" | ||||||
|  | 	"reflect" | ||||||
|  | 	"regexp" | ||||||
|  | 	"runtime" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| var anyMethods = []string{ | var anyMethods = []string{ | ||||||
| @@ -14,58 +21,87 @@ var anyMethods = []string{ | |||||||
| type GinRoutesWrapper struct { | type GinRoutesWrapper struct { | ||||||
| 	wrapper        *GinWrapper | 	wrapper        *GinWrapper | ||||||
| 	routes         gin.IRouter | 	routes         gin.IRouter | ||||||
|  | 	absPath        string | ||||||
|  | 	defaultHandler []gin.HandlerFunc | ||||||
| } | } | ||||||
|  |  | ||||||
| type GinRouteBuilder struct { | type GinRouteBuilder struct { | ||||||
| 	routes *GinRoutesWrapper | 	routes *GinRoutesWrapper | ||||||
|  |  | ||||||
| 	methods  []string | 	method   string | ||||||
| 	relPath  string | 	relPath  string | ||||||
|  | 	absPath  string | ||||||
| 	handlers []gin.HandlerFunc | 	handlers []gin.HandlerFunc | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinWrapper) Routes() *GinRoutesWrapper { | func (w *GinWrapper) Routes() *GinRoutesWrapper { | ||||||
| 	return &GinRoutesWrapper{wrapper: w, routes: w.engine} | 	return &GinRoutesWrapper{ | ||||||
|  | 		wrapper:        w, | ||||||
|  | 		routes:         w.engine, | ||||||
|  | 		absPath:        "", | ||||||
|  | 		defaultHandler: make([]gin.HandlerFunc, 0), | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) Group(relativePath string) *GinRoutesWrapper { | func (w *GinRoutesWrapper) Group(relativePath string) *GinRoutesWrapper { | ||||||
| 	return &GinRoutesWrapper{wrapper: w.wrapper, routes: w.routes.Group(relativePath)} | 	return &GinRoutesWrapper{ | ||||||
|  | 		wrapper:        w.wrapper, | ||||||
|  | 		routes:         w.routes.Group(relativePath), | ||||||
|  | 		defaultHandler: langext.ArrCopy(w.defaultHandler), | ||||||
|  | 		absPath:        joinPaths(w.absPath, relativePath), | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *GinRoutesWrapper) Use(middleware ...gin.HandlerFunc) *GinRoutesWrapper { | ||||||
|  | 	defHandler := langext.ArrCopy(w.defaultHandler) | ||||||
|  | 	defHandler = append(defHandler, middleware...) | ||||||
|  | 	return &GinRoutesWrapper{wrapper: w.wrapper, routes: w.routes, defaultHandler: defHandler} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) GET(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) GET(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodGet}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodGet, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) POST(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) POST(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodPost}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodPost, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) DELETE(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) DELETE(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodDelete}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodDelete, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) PATCH(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) PATCH(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodPatch}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodPatch, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) PUT(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) PUT(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodPut}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodPut, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) OPTIONS(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) OPTIONS(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodOptions}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodOptions, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) HEAD(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) HEAD(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{http.MethodHead}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route(http.MethodHead, relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) COUNT(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) COUNT(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: []string{"COUNT"}, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route("COUNT", relativePath) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRoutesWrapper) Any(relativePath string) *GinRouteBuilder { | func (w *GinRoutesWrapper) Any(relativePath string) *GinRouteBuilder { | ||||||
| 	return &GinRouteBuilder{routes: w, methods: anyMethods, relPath: relativePath, handlers: make([]gin.HandlerFunc, 0)} | 	return w._route("*", relativePath) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (w *GinRoutesWrapper) _route(method string, relativePath string) *GinRouteBuilder { | ||||||
|  | 	return &GinRouteBuilder{ | ||||||
|  | 		routes:   w, | ||||||
|  | 		method:   method, | ||||||
|  | 		relPath:  relativePath, | ||||||
|  | 		absPath:  joinPaths(w.absPath, relativePath), | ||||||
|  | 		handlers: langext.ArrCopy(w.defaultHandler), | ||||||
|  | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRouteBuilder) Use(middleware ...gin.HandlerFunc) *GinRouteBuilder { | func (w *GinRouteBuilder) Use(middleware ...gin.HandlerFunc) *GinRouteBuilder { | ||||||
| @@ -74,12 +110,83 @@ func (w *GinRouteBuilder) Use(middleware ...gin.HandlerFunc) *GinRouteBuilder { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinRouteBuilder) Handle(handler WHandlerFunc) { | func (w *GinRouteBuilder) Handle(handler WHandlerFunc) { | ||||||
|  |  | ||||||
|  | 	middlewareNames := langext.ArrMap(w.handlers, func(v gin.HandlerFunc) string { return nameOfFunction(v) }) | ||||||
|  | 	handlerName := nameOfFunction(handler) | ||||||
|  |  | ||||||
| 	w.handlers = append(w.handlers, Wrap(w.routes.wrapper, handler)) | 	w.handlers = append(w.handlers, Wrap(w.routes.wrapper, handler)) | ||||||
| 	for _, m := range w.methods { |  | ||||||
| 		w.routes.routes.Handle(m, w.relPath, w.handlers...) | 	methodName := w.method | ||||||
|  |  | ||||||
|  | 	if w.method == "*" { | ||||||
|  | 		methodName = "ANY" | ||||||
|  | 		for _, method := range anyMethods { | ||||||
|  | 			w.routes.routes.Handle(method, w.relPath, w.handlers...) | ||||||
| 		} | 		} | ||||||
|  | 	} else { | ||||||
|  | 		w.routes.routes.Handle(w.method, w.relPath, w.handlers...) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	w.routes.wrapper.routeSpecs = append(w.routes.wrapper.routeSpecs, ginRouteSpec{ | ||||||
|  | 		Method:      methodName, | ||||||
|  | 		URL:         w.absPath, | ||||||
|  | 		Middlewares: middlewareNames, | ||||||
|  | 		Handler:     handlerName, | ||||||
|  | 	}) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (w *GinWrapper) NoRoute(handler WHandlerFunc) { | func (w *GinWrapper) NoRoute(handler WHandlerFunc) { | ||||||
| 	w.engine.NoRoute(Wrap(w, handler)) | 	w.engine.NoRoute(Wrap(w, handler)) | ||||||
|  |  | ||||||
|  | 	w.routeSpecs = append(w.routeSpecs, ginRouteSpec{ | ||||||
|  | 		Method:      "ANY", | ||||||
|  | 		URL:         "[NO_ROUTE]", | ||||||
|  | 		Middlewares: nil, | ||||||
|  | 		Handler:     nameOfFunction(handler), | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func nameOfFunction(f any) string { | ||||||
|  |  | ||||||
|  | 	fname := runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name() | ||||||
|  |  | ||||||
|  | 	split := strings.Split(fname, "/") | ||||||
|  | 	if len(split) == 0 { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	fname = split[len(split)-1] | ||||||
|  |  | ||||||
|  | 	// https://stackoverflow.com/a/32925345/1761622 | ||||||
|  | 	if strings.HasSuffix(fname, "-fm") { | ||||||
|  | 		fname = fname[:len(fname)-len("-fm")] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	suffix := rext.W(regexp.MustCompile(`\.func[0-9]+(?:\.[0-9]+)*$`)) | ||||||
|  |  | ||||||
|  | 	if match, ok := suffix.MatchFirst(fname); ok { | ||||||
|  | 		fname = fname[:len(fname)-match.FullMatch().Length()] | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return fname | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // joinPaths is copied verbatim from gin@v1.9.1/gin.go | ||||||
|  | func joinPaths(absolutePath, relativePath string) string { | ||||||
|  | 	if relativePath == "" { | ||||||
|  | 		return absolutePath | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	finalPath := path.Join(absolutePath, relativePath) | ||||||
|  | 	if lastChar(relativePath) == '/' && lastChar(finalPath) != '/' { | ||||||
|  | 		return finalPath + "/" | ||||||
|  | 	} | ||||||
|  | 	return finalPath | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func lastChar(str string) uint8 { | ||||||
|  | 	if str == "" { | ||||||
|  | 		panic("The length of the string can't be 0") | ||||||
|  | 	} | ||||||
|  | 	return str[len(str)-1] | ||||||
| } | } | ||||||
|   | |||||||
							
								
								
									
										2
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								go.mod
									
									
									
									
									
								
							| @@ -14,7 +14,7 @@ require ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| require ( | require ( | ||||||
| 	github.com/bytedance/sonic v1.10.0-rc2 // indirect | 	github.com/bytedance/sonic v1.10.0-rc3 // indirect | ||||||
| 	github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect | 	github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect | ||||||
| 	github.com/chenzhuoyu/iasm v0.9.0 // indirect | 	github.com/chenzhuoyu/iasm v0.9.0 // indirect | ||||||
| 	github.com/gabriel-vasile/mimetype v1.4.2 // indirect | 	github.com/gabriel-vasile/mimetype v1.4.2 // indirect | ||||||
|   | |||||||
							
								
								
									
										2
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								go.sum
									
									
									
									
									
								
							| @@ -2,6 +2,8 @@ github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1 | |||||||
| github.com/bytedance/sonic v1.10.0-rc/go.mod h1:ElCzW+ufi8qKqNW0FY314xriJhyJhuoJ3gFZdAHF7NM= | github.com/bytedance/sonic v1.10.0-rc/go.mod h1:ElCzW+ufi8qKqNW0FY314xriJhyJhuoJ3gFZdAHF7NM= | ||||||
| github.com/bytedance/sonic v1.10.0-rc2 h1:oDfRZ+4m6AYCOC0GFeOCeYqvBmucy1isvouS2K0cPzo= | github.com/bytedance/sonic v1.10.0-rc2 h1:oDfRZ+4m6AYCOC0GFeOCeYqvBmucy1isvouS2K0cPzo= | ||||||
| github.com/bytedance/sonic v1.10.0-rc2/go.mod h1:iZcSUejdk5aukTND/Eu/ivjQuEL0Cu9/rf50Hi0u/g4= | github.com/bytedance/sonic v1.10.0-rc2/go.mod h1:iZcSUejdk5aukTND/Eu/ivjQuEL0Cu9/rf50Hi0u/g4= | ||||||
|  | github.com/bytedance/sonic v1.10.0-rc3 h1:uNSnscRapXTwUgTyOF0GVljYD08p9X/Lbr9MweSV3V0= | ||||||
|  | github.com/bytedance/sonic v1.10.0-rc3/go.mod h1:iZcSUejdk5aukTND/Eu/ivjQuEL0Cu9/rf50Hi0u/g4= | ||||||
| github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= | github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY= | ||||||
| github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= | github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= | ||||||
| github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d h1:77cEq6EriyTZ0g/qfRdp61a3Uu/AWrgIq2s0ClJV1g0= | github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d h1:77cEq6EriyTZ0g/qfRdp61a3Uu/AWrgIq2s0ClJV1g0= | ||||||
|   | |||||||
| @@ -1,5 +1,5 @@ | |||||||
| package goext | package goext | ||||||
|  |  | ||||||
| const GoextVersion = "0.0.189" | const GoextVersion = "0.0.218" | ||||||
|  |  | ||||||
| const GoextVersionTimestamp = "2023-07-24T11:11:15+0200" | const GoextVersionTimestamp = "2023-07-27T17:16:30+0200" | ||||||
|   | |||||||
| @@ -8,7 +8,7 @@ import ( | |||||||
|  |  | ||||||
| func TestGroupByNameOrEmpty1(t *testing.T) { | func TestGroupByNameOrEmpty1(t *testing.T) { | ||||||
|  |  | ||||||
| 	regex1 := W(regexp.MustCompile("0(?P<group1>A+)B(?P<group2>C+)0")) | 	regex1 := W(regexp.MustCompile(`0(?P<group1>A+)B(?P<group2>C+)0`)) | ||||||
|  |  | ||||||
| 	match1, ok1 := regex1.MatchFirst("Hello 0AAAABCCC0 Bye.") | 	match1, ok1 := regex1.MatchFirst("Hello 0AAAABCCC0 Bye.") | ||||||
|  |  | ||||||
| @@ -26,7 +26,7 @@ func TestGroupByNameOrEmpty1(t *testing.T) { | |||||||
|  |  | ||||||
| func TestGroupByNameOrEmpty2(t *testing.T) { | func TestGroupByNameOrEmpty2(t *testing.T) { | ||||||
|  |  | ||||||
| 	regex1 := W(regexp.MustCompile("0(?P<group1>A+)B(?P<group2>C+)(?P<group3>C+)?0")) | 	regex1 := W(regexp.MustCompile(`0(?P<group1>A+)B(?P<group2>C+)(?P<group3>C+)?0`)) | ||||||
|  |  | ||||||
| 	match1, ok1 := regex1.MatchFirst("Hello 0AAAABCCC0 Bye.") | 	match1, ok1 := regex1.MatchFirst("Hello 0AAAABCCC0 Bye.") | ||||||
|  |  | ||||||
|   | |||||||
| @@ -39,7 +39,7 @@ func HashSqliteSchema(ctx context.Context, schemaStr string) (string, error) { | |||||||
| 	return HashSqliteDatabase(ctx, db) | 	return HashSqliteDatabase(ctx, db) | ||||||
| } | } | ||||||
|  |  | ||||||
| func HashSqliteDatabase(ctx context.Context, db DB) (string, error) { | func HashSqliteDatabase(ctx context.Context, db Queryable) (string, error) { | ||||||
| 	ss, err := CreateSqliteDatabaseSchemaString(ctx, db) | 	ss, err := CreateSqliteDatabaseSchemaString(ctx, db) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return "", err | 		return "", err | ||||||
| @@ -50,7 +50,7 @@ func HashSqliteDatabase(ctx context.Context, db DB) (string, error) { | |||||||
| 	return hex.EncodeToString(cs[:]), nil | 	return hex.EncodeToString(cs[:]), nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func CreateSqliteDatabaseSchemaString(ctx context.Context, db DB) (string, error) { | func CreateSqliteDatabaseSchemaString(ctx context.Context, db Queryable) (string, error) { | ||||||
|  |  | ||||||
| 	type colInfo struct { | 	type colInfo struct { | ||||||
| 		Name       string  `db:"name"` | 		Name       string  `db:"name"` | ||||||
|   | |||||||
| @@ -7,9 +7,19 @@ import ( | |||||||
| 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | 	"gogs.mikescher.com/BlackForestBytes/goext/langext" | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | type TxStatus string | ||||||
|  |  | ||||||
|  | const ( | ||||||
|  | 	TxStatusInitial  TxStatus = "INITIAL" | ||||||
|  | 	TxStatusActive   TxStatus = "ACTIVE" | ||||||
|  | 	TxStatusComitted TxStatus = "COMMITTED" | ||||||
|  | 	TxStatusRollback TxStatus = "ROLLBACK" | ||||||
|  | ) | ||||||
|  |  | ||||||
| type Tx interface { | type Tx interface { | ||||||
| 	Rollback() error | 	Rollback() error | ||||||
| 	Commit() error | 	Commit() error | ||||||
|  | 	Status() TxStatus | ||||||
| 	Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) | 	Exec(ctx context.Context, sql string, prep PP) (sql.Result, error) | ||||||
| 	Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) | 	Query(ctx context.Context, sql string, prep PP) (*sqlx.Rows, error) | ||||||
| } | } | ||||||
| @@ -18,6 +28,9 @@ type transaction struct { | |||||||
| 	tx       *sqlx.Tx | 	tx       *sqlx.Tx | ||||||
| 	id       uint16 | 	id       uint16 | ||||||
| 	lstr     []Listener | 	lstr     []Listener | ||||||
|  | 	status   TxStatus | ||||||
|  | 	execCtr  int | ||||||
|  | 	queryCtr int | ||||||
| } | } | ||||||
|  |  | ||||||
| func NewTransaction(xtx *sqlx.Tx, txid uint16, lstr []Listener) Tx { | func NewTransaction(xtx *sqlx.Tx, txid uint16, lstr []Listener) Tx { | ||||||
| @@ -25,6 +38,9 @@ func NewTransaction(xtx *sqlx.Tx, txid uint16, lstr []Listener) Tx { | |||||||
| 		tx:       xtx, | 		tx:       xtx, | ||||||
| 		id:       txid, | 		id:       txid, | ||||||
| 		lstr:     lstr, | 		lstr:     lstr, | ||||||
|  | 		status:   TxStatusInitial, | ||||||
|  | 		execCtr:  0, | ||||||
|  | 		queryCtr: 0, | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -38,6 +54,10 @@ func (tx *transaction) Rollback() error { | |||||||
|  |  | ||||||
| 	result := tx.tx.Rollback() | 	result := tx.tx.Rollback() | ||||||
|  |  | ||||||
|  | 	if result == nil { | ||||||
|  | 		tx.status = TxStatusRollback | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, v := range tx.lstr { | 	for _, v := range tx.lstr { | ||||||
| 		v.PostTxRollback(tx.id, result) | 		v.PostTxRollback(tx.id, result) | ||||||
| 	} | 	} | ||||||
| @@ -55,6 +75,10 @@ func (tx *transaction) Commit() error { | |||||||
|  |  | ||||||
| 	result := tx.tx.Commit() | 	result := tx.tx.Commit() | ||||||
|  |  | ||||||
|  | 	if result == nil { | ||||||
|  | 		tx.status = TxStatusComitted | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, v := range tx.lstr { | 	for _, v := range tx.lstr { | ||||||
| 		v.PostTxRollback(tx.id, result) | 		v.PostTxRollback(tx.id, result) | ||||||
| 	} | 	} | ||||||
| @@ -73,6 +97,10 @@ func (tx *transaction) Exec(ctx context.Context, sqlstr string, prep PP) (sql.Re | |||||||
|  |  | ||||||
| 	res, err := tx.tx.NamedExecContext(ctx, sqlstr, prep) | 	res, err := tx.tx.NamedExecContext(ctx, sqlstr, prep) | ||||||
|  |  | ||||||
|  | 	if tx.status == TxStatusInitial && err == nil { | ||||||
|  | 		tx.status = TxStatusActive | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, v := range tx.lstr { | 	for _, v := range tx.lstr { | ||||||
| 		v.PostExec(langext.Ptr(tx.id), origsql, sqlstr, prep) | 		v.PostExec(langext.Ptr(tx.id), origsql, sqlstr, prep) | ||||||
| 	} | 	} | ||||||
| @@ -94,6 +122,10 @@ func (tx *transaction) Query(ctx context.Context, sqlstr string, prep PP) (*sqlx | |||||||
|  |  | ||||||
| 	rows, err := sqlx.NamedQueryContext(ctx, tx.tx, sqlstr, prep) | 	rows, err := sqlx.NamedQueryContext(ctx, tx.tx, sqlstr, prep) | ||||||
|  |  | ||||||
|  | 	if tx.status == TxStatusInitial && err == nil { | ||||||
|  | 		tx.status = TxStatusActive | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, v := range tx.lstr { | 	for _, v := range tx.lstr { | ||||||
| 		v.PostQuery(langext.Ptr(tx.id), origsql, sqlstr, prep) | 		v.PostQuery(langext.Ptr(tx.id), origsql, sqlstr, prep) | ||||||
| 	} | 	} | ||||||
| @@ -103,3 +135,11 @@ func (tx *transaction) Query(ctx context.Context, sqlstr string, prep PP) (*sqlx | |||||||
| 	} | 	} | ||||||
| 	return rows, nil | 	return rows, nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (tx *transaction) Status() TxStatus { | ||||||
|  | 	return tx.status | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (tx *transaction) Traffic() (int, int) { | ||||||
|  | 	return tx.execCtr, tx.queryCtr | ||||||
|  | } | ||||||
|   | |||||||
| @@ -71,12 +71,12 @@ func SupportsColors() bool { | |||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var term256Regex = regexp.MustCompile("(?i)-256(color)?$") | 	var term256Regex = regexp.MustCompile(`(?i)-256(color)?$`) | ||||||
| 	if term256Regex.MatchString(termenv) { | 	if term256Regex.MatchString(termenv) { | ||||||
| 		return true | 		return true | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	var termBasicRegex = regexp.MustCompile("(?i)^screen|^xterm|^vt100|^vt220|^rxvt|color|ansi|cygwin|linux") | 	var termBasicRegex = regexp.MustCompile(`(?i)^screen|^xterm|^vt100|^vt220|^rxvt|color|ansi|cygwin|linux`) | ||||||
|  |  | ||||||
| 	if termBasicRegex.MatchString(termenv) { | 	if termBasicRegex.MatchString(termenv) { | ||||||
| 		return true | 		return true | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user