Skip to content

Commit

Permalink
docs: Improve ctx.Locals method description, godoc and example
Browse files Browse the repository at this point in the history
  • Loading branch information
sixcolors committed Jun 11, 2024
1 parent 46fffe4 commit cc3dd43
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 9 deletions.
11 changes: 10 additions & 1 deletion ctx.go
Original file line number Diff line number Diff line change
Expand Up @@ -859,6 +859,10 @@ func (c *DefaultCtx) Links(link ...string) {

// Locals makes it possible to pass any values under keys scoped to the request
// and therefore available to all following routes that match the request.
//
// All the values are removed from ctx after returning from the top
// RequestHandler. Additionally, Close method is called on each value
// implementing io.Closer before removing the value from ctx.
func (c *DefaultCtx) Locals(key any, value ...any) any {
if len(value) == 0 {
return c.fasthttp.UserValue(key)
Expand All @@ -868,7 +872,12 @@ func (c *DefaultCtx) Locals(key any, value ...any) any {
}

// Locals function utilizing Go's generics feature.
// This function allows for manipulating and retrieving local values within a request context with a more specific data type.
// This function allows for manipulating and retrieving local values within a
// request context with a more specific data type.
//
// All the values are removed from ctx after returning from the top
// RequestHandler. Additionally, Close method is called on each value
// implementing io.Closer before removing the value from ctx.
func Locals[V any](c Ctx, key any, value ...V) V {
var v V
var ok bool
Expand Down
16 changes: 8 additions & 8 deletions docs/api/ctx.md
Original file line number Diff line number Diff line change
Expand Up @@ -945,10 +945,10 @@ app.Get("/", func(c fiber.Ctx) error {
## Locals
A method that stores variables scoped to the request and, therefore, are available only to the routes that match the request.
A method that stores variables scoped to the request and, therefore, are available only to the routes that match the request. The stored variables are removed after the request is handled. If any of the stored data implements the `io.Closer` interface, its `Close` method will be called before it's removed.
:::tip
This is useful if you want to pass some **specific** data to the next middleware.
This is useful if you want to pass some **specific** data to the next middleware. Remember to perform type assertions when retrieving the data to ensure it is of the expected type. You can also use a non-exported type as a key to avoid collisions.
:::
```go title="Signature"
Expand All @@ -959,24 +959,24 @@ func (c Ctx) Locals(key any, value ...any) any

// key is an unexported type for keys defined in this package.
// This prevents collisions with keys defined in other packages.
type key int
type keyType int

// userKey is the key for user.User values in Contexts. It is
// unexported; clients use user.NewContext and user.FromContext
// instead of using this key directly.
var userKey key

app.Use(func(c fiber.Ctx) error {
c.Locals(userKey, "admin")
app.Use(func(c *fiber.Ctx) error {
c.Locals(userKey, "admin") // Stores the string "admin" under a non-exported type key
return c.Next()
})

app.Get("/admin", func(c fiber.Ctx) error {
if c.Locals(userKey) == "admin" {
app.Get("/admin", func(c *fiber.Ctx) error {
user, ok := c.Locals(userKey).(string) // Retrieves the data stored under the key and performs a type assertion
if ok && user == "admin" {
return c.Status(fiber.StatusOK).SendString("Welcome, admin!")
}
return c.SendStatus(fiber.StatusForbidden)

})
```
Expand Down

0 comments on commit cc3dd43

Please sign in to comment.