♻️Refactor: remove redundant field `method` in `DefaultCtx` (#3372)

* ♻️Refactor: remove redundant field method in defaultCtx

* ♻️Refactor: rename getMethodINT to getMethodInt
pull/3378/head
Kashiwa 2025-03-26 20:16:53 +08:00 committed by GitHub
parent dab20c9df5
commit e90fe8afbc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 56 additions and 51 deletions

30
app.go
View File

@ -456,17 +456,29 @@ const (
DefaultWriteBufferSize = 4096 DefaultWriteBufferSize = 4096
) )
const (
methodGet = iota
methodHead
methodPost
methodPut
methodDelete
methodConnect
methodOptions
methodTrace
methodPatch
)
// HTTP methods enabled by default // HTTP methods enabled by default
var DefaultMethods = []string{ var DefaultMethods = []string{
MethodGet, methodGet: MethodGet,
MethodHead, methodHead: MethodHead,
MethodPost, methodPost: MethodPost,
MethodPut, methodPut: MethodPut,
MethodDelete, methodDelete: MethodDelete,
MethodConnect, methodConnect: MethodConnect,
MethodOptions, methodOptions: MethodOptions,
MethodTrace, methodTrace: MethodTrace,
MethodPatch, methodPatch: MethodPatch,
} }
// DefaultErrorHandler that process return errors from handlers // DefaultErrorHandler that process return errors from handlers

26
ctx.go
View File

@ -61,7 +61,6 @@ type DefaultCtx struct {
res *DefaultRes // Default response api reference res *DefaultRes // Default response api reference
values [maxParams]string // Route parameter values values [maxParams]string // Route parameter values
viewBindMap sync.Map // Default view map to bind template engine viewBindMap sync.Map // Default view map to bind template engine
method string // HTTP method
baseURI string // HTTP base uri baseURI string // HTTP base uri
pathOriginal string // Original HTTP path pathOriginal string // Original HTTP path
flashMessages redirectionMsgs // Flash messages flashMessages redirectionMsgs // Flash messages
@ -70,7 +69,7 @@ type DefaultCtx struct {
treePathHash int // Hash of the path for the search in the tree treePathHash int // Hash of the path for the search in the tree
indexRoute int // Index of the current route indexRoute int // Index of the current route
indexHandler int // Index of the current handler indexHandler int // Index of the current handler
methodINT int // HTTP method INT equivalent methodInt int // HTTP method INT equivalent
matched bool // Non use route matched matched bool // Non use route matched
} }
@ -1006,19 +1005,17 @@ func (c *DefaultCtx) Location(path string) {
func (c *DefaultCtx) Method(override ...string) string { func (c *DefaultCtx) Method(override ...string) string {
if len(override) == 0 { if len(override) == 0 {
// Nothing to override, just return current method from context // Nothing to override, just return current method from context
return c.method return c.app.method(c.methodInt)
} }
method := utils.ToUpper(override[0]) method := utils.ToUpper(override[0])
mINT := c.app.methodInt(method) methodInt := c.app.methodInt(method)
if mINT == -1 { if methodInt == -1 {
// Provided override does not valid HTTP method, no override, return current method // Provided override does not valid HTTP method, no override, return current method
return c.method return c.app.method(c.methodInt)
} }
c.methodInt = methodInt
c.method = method return method
c.methodINT = mINT
return c.method
} }
// MultipartForm parse form entries from binary. // MultipartForm parse form entries from binary.
@ -1486,7 +1483,7 @@ func (c *DefaultCtx) Route() *Route {
return &Route{ return &Route{
path: c.pathOriginal, path: c.pathOriginal,
Path: c.pathOriginal, Path: c.pathOriginal,
Method: c.method, Method: c.Method(),
Handlers: make([]Handler, 0), Handlers: make([]Handler, 0),
Params: make([]string, 0), Params: make([]string, 0),
} }
@ -1919,8 +1916,7 @@ func (c *DefaultCtx) Reset(fctx *fasthttp.RequestCtx) {
// Set paths // Set paths
c.pathOriginal = c.app.getString(fctx.URI().PathOriginal()) c.pathOriginal = c.app.getString(fctx.URI().PathOriginal())
// Set method // Set method
c.method = c.app.getString(fctx.Request.Header.Method()) c.methodInt = c.app.methodInt(utils.UnsafeString(fctx.Request.Header.Method()))
c.methodINT = c.app.methodInt(c.method)
// Attach *fasthttp.RequestCtx to ctx // Attach *fasthttp.RequestCtx to ctx
c.fasthttp = fctx c.fasthttp = fctx
// reset base uri // reset base uri
@ -1951,8 +1947,8 @@ func (c *DefaultCtx) getBody() []byte {
} }
// Methods to use with next stack. // Methods to use with next stack.
func (c *DefaultCtx) getMethodINT() int { func (c *DefaultCtx) getMethodInt() int {
return c.methodINT return c.methodInt
} }
func (c *DefaultCtx) getIndexRoute() int { func (c *DefaultCtx) getIndexRoute() int {

View File

@ -17,7 +17,7 @@ type CustomCtx interface {
Reset(fctx *fasthttp.RequestCtx) Reset(fctx *fasthttp.RequestCtx)
// Methods to use with next stack. // Methods to use with next stack.
getMethodINT() int getMethodInt() int
getIndexRoute() int getIndexRoute() int
getTreePathHash() int getTreePathHash() int
getDetectionPath() string getDetectionPath() string

View File

@ -345,7 +345,7 @@ type Ctx interface {
release() release()
getBody() []byte getBody() []byte
// Methods to use with next stack. // Methods to use with next stack.
getMethodINT() int getMethodInt() int
getIndexRoute() int getIndexRoute() int
getTreePathHash() int getTreePathHash() int
getDetectionPath() string getDetectionPath() string

View File

@ -14,6 +14,7 @@ import (
"os" "os"
"path/filepath" "path/filepath"
"reflect" "reflect"
"slices"
"strconv" "strconv"
"strings" "strings"
"sync" "sync"
@ -107,7 +108,7 @@ func (app *App) methodExist(c *DefaultCtx) bool {
methods := app.config.RequestMethods methods := app.config.RequestMethods
for i := 0; i < len(methods); i++ { for i := 0; i < len(methods); i++ {
// Skip original method // Skip original method
if c.getMethodINT() == i { if c.getMethodInt() == i {
continue continue
} }
// Reset stack index // Reset stack index
@ -151,7 +152,7 @@ func (app *App) methodExistCustom(c CustomCtx) bool {
methods := app.config.RequestMethods methods := app.config.RequestMethods
for i := 0; i < len(methods); i++ { for i := 0; i < len(methods); i++ {
// Skip original method // Skip original method
if c.getMethodINT() == i { if c.getMethodInt() == i {
continue continue
} }
// Reset stack index // Reset stack index
@ -652,39 +653,35 @@ func getBytesImmutable(s string) []byte {
func (app *App) methodInt(s string) int { func (app *App) methodInt(s string) int {
// For better performance // For better performance
if len(app.configured.RequestMethods) == 0 { if len(app.configured.RequestMethods) == 0 {
// TODO: Use iota instead
switch s { switch s {
case MethodGet: case MethodGet:
return 0 return methodGet
case MethodHead: case MethodHead:
return 1 return methodHead
case MethodPost: case MethodPost:
return 2 return methodPost
case MethodPut: case MethodPut:
return 3 return methodPut
case MethodDelete: case MethodDelete:
return 4 return methodDelete
case MethodConnect: case MethodConnect:
return 5 return methodConnect
case MethodOptions: case MethodOptions:
return 6 return methodOptions
case MethodTrace: case MethodTrace:
return 7 return methodTrace
case MethodPatch: case MethodPatch:
return 8 return methodPatch
default: default:
return -1 return -1
} }
} }
// For method customization // For method customization
for i, v := range app.config.RequestMethods { return slices.Index(app.config.RequestMethods, s)
if s == v { }
return i
}
}
return -1 func (app *App) method(methodInt int) string {
return app.config.RequestMethods[methodInt]
} }
// IsMethodSafe reports whether the HTTP method is considered safe. // IsMethodSafe reports whether the HTTP method is considered safe.

View File

@ -110,9 +110,9 @@ func (r *Route) match(detectionPath, path string, params *[maxParams]string) boo
func (app *App) nextCustom(c CustomCtx) (bool, error) { //nolint:unparam // bool param might be useful for testing func (app *App) nextCustom(c CustomCtx) (bool, error) { //nolint:unparam // bool param might be useful for testing
// Get stack length // Get stack length
tree, ok := app.treeStack[c.getMethodINT()][c.getTreePathHash()] tree, ok := app.treeStack[c.getMethodInt()][c.getTreePathHash()]
if !ok { if !ok {
tree = app.treeStack[c.getMethodINT()][0] tree = app.treeStack[c.getMethodInt()][0]
} }
lenr := len(tree) - 1 lenr := len(tree) - 1
@ -158,9 +158,9 @@ func (app *App) nextCustom(c CustomCtx) (bool, error) { //nolint:unparam // bool
func (app *App) next(c *DefaultCtx) (bool, error) { func (app *App) next(c *DefaultCtx) (bool, error) {
// Get stack length // Get stack length
tree, ok := app.treeStack[c.methodINT][c.treePathHash] tree, ok := app.treeStack[c.methodInt][c.treePathHash]
if !ok { if !ok {
tree = app.treeStack[c.methodINT][0] tree = app.treeStack[c.methodInt][0]
} }
lenTree := len(tree) - 1 lenTree := len(tree) - 1
@ -202,7 +202,7 @@ func (app *App) next(c *DefaultCtx) (bool, error) {
} }
// If c.Next() does not match, return 404 // If c.Next() does not match, return 404
err := NewError(StatusNotFound, "Cannot "+c.method+" "+html.EscapeString(c.pathOriginal)) err := NewError(StatusNotFound, "Cannot "+c.Method()+" "+html.EscapeString(c.pathOriginal))
if !c.matched && app.methodExist(c) { if !c.matched && app.methodExist(c) {
// If no match, scan stack again if other methods match the request // If no match, scan stack again if other methods match the request
// Moved from app.handler because middleware may break the route chain // Moved from app.handler because middleware may break the route chain
@ -221,7 +221,7 @@ func (app *App) defaultRequestHandler(rctx *fasthttp.RequestCtx) {
defer app.ReleaseCtx(ctx) defer app.ReleaseCtx(ctx)
// Check if the HTTP method is valid // Check if the HTTP method is valid
if ctx.methodINT == -1 { if ctx.methodInt == -1 {
_ = ctx.SendStatus(StatusNotImplemented) //nolint:errcheck // Always return nil _ = ctx.SendStatus(StatusNotImplemented) //nolint:errcheck // Always return nil
return return
} }