Refine the rest structure.

code/rest/base_controller.go

@@ -0,0 +1,126 @@
+package rest
+
+import (
+	"fmt"
+	"github.com/json-iterator/go"
+	"go/types"
+	"net/http"
+	"tank/code/tool/result"
+)
+
+type IController interface {
+	IBean
+	//注册自己固定的路由。
+	RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request)
+	//处理一些特殊的路由。
+	HandleRoutes(writer http.ResponseWriter, request *http.Request) (func(writer http.ResponseWriter, request *http.Request), bool)
+}
+type BaseController struct {
+	Bean
+	userDao    *UserDao
+	sessionDao *SessionDao
+}
+
+func (this *BaseController) Init() {
+
+	this.Bean.Init()
+
+	//手动装填本实例的Bean.
+	b := CONTEXT.GetBean(this.userDao)
+	if b, ok := b.(*UserDao); ok {
+		this.userDao = b
+	}
+
+	b = CONTEXT.GetBean(this.sessionDao)
+	if b, ok := b.(*SessionDao); ok {
+		this.sessionDao = b
+	}
+
+}
+
+//注册自己的路由。
+func (this *BaseController) RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request) {
+	//每个Controller需要主动注册自己的路由。
+	return make(map[string]func(writer http.ResponseWriter, request *http.Request))
+}
+
+//处理一些特殊的接口，比如参数包含在路径中,一般情况下，controller不将参数放在url路径中
+func (this *BaseController) HandleRoutes(writer http.ResponseWriter, request *http.Request) (func(writer http.ResponseWriter, request *http.Request), bool) {
+	return nil, false
+}
+
+//需要进行登录验证的wrap包装
+func (this *BaseController) Wrap(f func(writer http.ResponseWriter, request *http.Request) *result.WebResult, qualifiedRole string) func(w http.ResponseWriter, r *http.Request) {
+
+	return func(writer http.ResponseWriter, request *http.Request) {
+
+		//writer和request赋值给自己。
+
+		var webResult *result.WebResult = nil
+
+		//只有游客接口不需要登录
+		if qualifiedRole != USER_ROLE_GUEST {
+			user := this.checkUser(writer, request)
+
+			if user.Status == USER_STATUS_DISABLED {
+				//判断用户是否被禁用。
+				webResult = result.ConstWebResult(result.CODE_WRAPPER_USER_DISABLED)
+			} else {
+				if qualifiedRole == USER_ROLE_ADMINISTRATOR && user.Role != USER_ROLE_ADMINISTRATOR {
+					webResult = result.ConstWebResult(result.CODE_WRAPPER_UNAUTHORIZED)
+				} else {
+					webResult = f(writer, request)
+				}
+			}
+
+		} else {
+			webResult = f(writer, request)
+		}
+
+		//输出的是json格式
+		if webResult != nil {
+			//返回的内容申明是json，utf-8
+			writer.Header().Set("Content-Type", "application/json;charset=UTF-8")
+
+			//用json的方式输出返回值。
+			b, err := jsoniter.ConfigCompatibleWithStandardLibrary.Marshal(webResult)
+
+			this.PanicError(err)
+
+			writer.WriteHeader(result.FetchHttpStatus(webResult.Code))
+
+			_, err = fmt.Fprintf(writer, string(b))
+			this.PanicError(err)
+		} else {
+			//输出的内容是二进制的。
+
+		}
+
+	}
+}
+
+//返回成功的结果。支持放置三种类型 1.字符串 2. WebResult对象 3.空指针 4.任意类型
+func (this *BaseController) Success(data interface{}) *result.WebResult {
+	var webResult *result.WebResult = nil
+	if value, ok := data.(string); ok {
+		//返回一句普通的消息
+		webResult = &result.WebResult{Code: result.CODE_WRAPPER_OK.Code, Msg: value}
+	} else if value, ok := data.(*result.WebResult); ok {
+		//返回一个webResult对象
+		webResult = value
+	} else if _, ok := data.(types.Nil); ok {
+		//返回一个空指针
+		webResult = result.ConstWebResult(result.CODE_WRAPPER_OK)
+	} else {
+		//返回的类型不明确。
+		webResult = &result.WebResult{Code: result.CODE_WRAPPER_OK.Code, Data: data}
+	}
+	return webResult
+}
+
+//允许跨域请求
+func (this *BaseController) allowCORS(writer http.ResponseWriter) {
+	writer.Header().Add("Access-Control-Allow-Origin", "*")
+	writer.Header().Add("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE")
+	writer.Header().Add("Access-Control-Max-Age", "3600")
+}