Compare commits

...

3 Commits

Author SHA1 Message Date
56fa3815cb update: hide Param, add SetParam(can't cover value) 2024-04-10 18:03:38 +08:00
9530fa863f feat: ctx add SendStatus 2024-04-10 11:24:17 +08:00
f3fb259eee feat: ctx add MultipartForm func 2024-03-11 16:28:33 +08:00
4 changed files with 69 additions and 6 deletions

3
app.go
View File

@ -165,8 +165,9 @@ func (a *App) handleHTTPRequest(c *Ctx) {
// Find route in tree
value := root.getValue(rPath, c.params, c.skippedNodes, unescape)
if value.params != nil {
c.Params = *value.params
c.params = value.params
}
if value.handlers != nil {
c.handlers = value.handlers
c.fullPath = value.fullPath

22
ctx.go
View File

@ -29,7 +29,7 @@ type Ctx struct {
locals map[string]interface{}
skippedNodes *[]skippedNode
fullPath string
Params Params
//Params Params
}
func newContext(app *App, writer http.ResponseWriter, request *http.Request) *Ctx {
@ -144,7 +144,12 @@ func (c *Ctx) verify() error {
}
func (c *Ctx) Param(key string) string {
return c.Params.ByName(key)
return c.params.ByName(key)
}
func (c *Ctx) SetParam(key, value string) {
params := append(*c.params, Param{Key: key, Value: value})
c.params = &params
}
func (c *Ctx) Form(key string) string {
@ -161,6 +166,14 @@ func (c *Ctx) FormFile(key string) (*multipart.FileHeader, error) {
return fh, err
}
func (c *Ctx) MultipartForm() (*multipart.Form, error) {
if err := c.Request.ParseMultipartForm(c.app.config.BodyLimit); err != nil {
return nil, err
}
return c.Request.MultipartForm, nil
}
func (c *Ctx) Query(key string) string {
return c.Request.URL.Query().Get(key)
}
@ -254,6 +267,11 @@ func (c *Ctx) SetHeader(key string, value string) {
c.writermem.Header().Set(key, value)
}
func (c *Ctx) SendStatus(code int) error {
c.writermem.WriteHeader(code)
return nil
}
func (c *Ctx) SendString(data string) error {
c.SetHeader("Content-Type", "text/plain")
_, err := c.Write([]byte(data))

View File

@ -6,11 +6,22 @@ import (
)
func main() {
app := nf.New(nf.Config{EnableNotImplementHandler: true})
app := nf.New(nf.Config{})
app.Get("/ok", func(c *nf.Ctx) error {
return c.SendStatus(200)
})
api := app.Group("/api")
api.Get("/1", func(c *nf.Ctx) error {
return c.SendString("nice")
api.Use(func(c *nf.Ctx) error {
c.SetParam("age", "18")
return c.Next()
})
api.Get("/:name", func(c *nf.Ctx) error {
name := c.Param("name")
age := c.Param("age")
return c.SendString(name + "@" + age)
})
log.Fatal(app.Run(":80"))

View File

@ -0,0 +1,33 @@
package main
import (
"github.com/loveuer/nf"
"github.com/loveuer/nf/nft/resp"
"log"
)
func main() {
app := nf.New(nf.Config{BodyLimit: 10 * 1024 * 1024})
app.Post("/upload", func(c *nf.Ctx) error {
fs, err := c.MultipartForm()
if err != nil {
return resp.Resp400(c, err.Error())
}
fm := make(map[string][]string)
for key := range fs.File {
if _, exist := fm[key]; !exist {
fm[key] = make([]string, 0)
}
for f := range fs.File[key] {
fm[key] = append(fm[key], fs.File[key][f].Filename)
}
}
return resp.Resp200(c, nf.Map{"value": fs.Value, "files": fm})
})
log.Fatal(app.Run(":13322"))
}