controller.go

package hfw

//手动匹配路由
import (
	"net/http"
	"strings"

	"github.com/gorilla/websocket"
	"github.com/hsyan2008/hfw/configs"
	"github.com/hsyan2008/hfw/redis"
	"github.com/hsyan2008/hfw/session"
)

//ControllerInterface ..
//init和finish必定会执行，而且不允许被修改
// Before和After之间是业务逻辑，所有Before也是必定会执行
//用户手动StopRun()后，中止业务逻辑，跳过After，继续Finish
type ControllerInterface interface {
	Init(*HTTPContext)
	Before(*HTTPContext)
	After(*HTTPContext)
	Finish(*HTTPContext)
	NotFound(*HTTPContext)
	ServerError(*HTTPContext)
}

//确认Controller实现了接口 ControllerInterface
var _ ControllerInterface = &Controller{}

//Controller ..
type Controller struct {
}

//Init 请不要实现Init方法
func (ctl *Controller) Init(httpCtx *HTTPContext) {

	var err error

	// logger.Debug("Controller init")

	if strings.Contains(httpCtx.Request.URL.RawQuery, "format=json") {
		httpCtx.IsJSON = true
	} else if strings.Contains(httpCtx.Request.Header.Get("Accept"), "application/json") {
		httpCtx.IsJSON = true
	}

	if strings.Contains(httpCtx.Request.Header.Get("Accept-Encoding"), "gzip") {
		httpCtx.IsZip = true
	}

	// _ = httpCtx.Request.ParseMultipartForm(2 * 1024 * 1024)

	//开启session，暂时只支持redis
	if configs.Config.EnableSession || configs.Config.Session.IsEnable {
		if redis.DefaultIns != nil {
			store := session.NewSessRedisStore(redis.DefaultIns, configs.Config.Redis)
			httpCtx.Session = session.NewSession(httpCtx.Request, store, configs.Config.Session)
		} else {
			httpCtx.Error("session enable faild: redis instance is nil")
		}
	}
	httpCtx.ThrowCheck(500, err)
}

//Before ..
func (ctl *Controller) Before(httpCtx *HTTPContext) {
	// logger.Debug("Controller Before")
}

//After ..
func (ctl *Controller) After(httpCtx *HTTPContext) {
	// logger.Debug("Controller After")
	if websocket.IsWebSocketUpgrade(httpCtx.Request) || httpCtx.IsCloseRender {
		return
	}
}

//Finish 请不要实现Finish方法
func (ctl *Controller) Finish(httpCtx *HTTPContext) {
	if websocket.IsWebSocketUpgrade(httpCtx.Request) {
		return
	}

	httpCtx.RenderResponse()
}

//NotFound ..
func (ctl *Controller) NotFound(httpCtx *HTTPContext) {

	httpCtx.HTTPStatus = http.StatusNotFound

	httpCtx.IsError = true

	httpCtx.ErrNo = 404
	httpCtx.ErrMsg = "NotFound"
}

//ServerError ..
//不要手动调用，用于捕获未知错误，手动请用Throw
//该方法不能使用StopRun，也不能panic，因为会被自动调用
func (ctl *Controller) ServerError(httpCtx *HTTPContext) {

	httpCtx.HTTPStatus = http.StatusInternalServerError

	httpCtx.IsError = true

	httpCtx.ErrNo = 500
	httpCtx.ErrMsg = "ServerError"
}