Add Renderer

This commit is contained in:
Nise Void 2018-08-19 17:04:29 +02:00
parent 78000acf6c
commit ae68248d25
Signed by untrusted user: NiseVoid
GPG Key ID: FBA14AC83EA602F3
3 changed files with 29 additions and 10 deletions

View File

@ -9,14 +9,15 @@ import (
// Context is passed to handlers and middlewares
type Context struct {
router *Router
Request *http.Request
Response http.ResponseWriter
Param func(string) string
store map[string]interface{}
}
func newContext(res http.ResponseWriter, req *http.Request, param httprouter.Params) *Context {
return &Context{req, res, param.ByName, make(map[string]interface{})}
func newContext(router *Router, res http.ResponseWriter, req *http.Request, param httprouter.Params) *Context {
return &Context{router, req, res, param.ByName, make(map[string]interface{})}
}
// String returns the given status code and writes the bytes to the body
@ -45,6 +46,15 @@ func (c *Context) JSON(code int, data interface{}) error {
return json.NewEncoder(c.Response).Encode(data) // TODO: Encode to buffer first to prevent partial responses on error
}
func (c *Context) Render(code int, template string, data interface{}) error {
if c.router.Renderer == nil {
panic(`Cannot call render without a renderer set`)
}
c.Response.WriteHeader(code)
return c.router.Renderer.Render(c.Response, template, data, c)
}
// Set sets a value in the context. Set is not safe to be used concurrently
func (c *Context) Set(key string, value interface{}) {
c.store[key] = value

View File

@ -20,8 +20,8 @@ type GetHandle func(*Context) error
// Router is the router itself
type Router struct {
routes []route
Port int
routes []route
Renderer Renderer
}
// New returns a new Router
@ -83,11 +83,11 @@ func (r *Router) getHttpr() *httprouter.Router {
for _, v := range r.routes {
if handle, ok := v.Handle.(GetHandle); ok {
httpr.Handle(v.Method, v.Path, handleGET(handle))
httpr.Handle(v.Method, v.Path, handleGET(r, handle))
continue
}
httpr.Handle(v.Method, v.Path, handlePOST(v.Handle))
httpr.Handle(v.Method, v.Path, handlePOST(r, v.Handle))
}
return httpr
@ -119,11 +119,11 @@ func checkInterfaceHandle(f interface{}) {
return
}
func handlePOST(f interface{}) httprouter.Handle {
func handlePOST(r *Router, f interface{}) httprouter.Handle {
funcRv, inputRt := reflect.ValueOf(f), reflect.TypeOf(f).In(1)
return func(res http.ResponseWriter, req *http.Request, param httprouter.Params) {
c := newContext(res, req, param)
c := newContext(r, res, req, param)
data := reflect.New(inputRt)
{
@ -141,9 +141,9 @@ func handlePOST(f interface{}) httprouter.Handle {
}
}
func handleGET(f GetHandle) httprouter.Handle {
func handleGET(r *Router, f GetHandle) httprouter.Handle {
return func(res http.ResponseWriter, req *http.Request, param httprouter.Params) {
c := newContext(res, req, param)
c := newContext(r, res, req, param)
err := f(c)

9
type.go Normal file
View File

@ -0,0 +1,9 @@
package router
import (
"io"
)
type Renderer interface {
Render(w io.Writer, template string, data interface{}, c *Context) error
}