浏览代码

Add errHandler mid ware

Unknown 11 年之前
父节点
当前提交
56cf05be77
共有 3 个文件被更改,包括 37 次插入5 次删除
  1. 32 0
      modules/log/error.go
  2. 4 5
      routers/repo/repo.go
  3. 1 0
      web.go

+ 32 - 0
modules/log/error.go

@@ -0,0 +1,32 @@
+// Copyright 2014 The Gogs Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package log
+
+import (
+	"github.com/codegangsta/martini"
+	"github.com/martini-contrib/render"
+)
+
+// ErrHandler is an interface for custom error handler.
+type ErrHandler interface {
+	Handle(string, render.Render, error)
+}
+
+// ErrHandle is a Middleware that maps a ErrHandler service into the Martini handler chain.
+func ErrHandle() martini.Handler {
+	return func(context martini.Context) {
+		context.MapTo(&errHandler{}, (*ErrHandler)(nil))
+	}
+}
+
+type errHandler struct {
+}
+
+func (eh *errHandler) Handle(title string, r render.Render, err error) {
+	Error("%s: %v", title, err)
+	r.HTML(200, "base/error", map[string]interface{}{
+		"ErrorMsg": err,
+	})
+}

+ 4 - 5
routers/repo/repo.go

@@ -5,9 +5,10 @@
 package repo
 
 import (
+	"net/http"
+
 	"github.com/martini-contrib/render"
 	"github.com/martini-contrib/sessions"
-	"net/http"
 
 	"github.com/gogits/gogs/models"
 	"github.com/gogits/gogs/modules/auth"
@@ -15,7 +16,7 @@ import (
 	"github.com/gogits/gogs/modules/log"
 )
 
-func Create(form auth.CreateRepoForm, req *http.Request, r render.Render, data base.TmplData, session sessions.Session) {
+func Create(form auth.CreateRepoForm, req *http.Request, r render.Render, data base.TmplData, session sessions.Session, eh log.ErrHandler) {
 	data["Title"] = "Create repository"
 
 	if req.Method == "GET" {
@@ -63,9 +64,7 @@ func Create(form auth.CreateRepoForm, req *http.Request, r render.Render, data b
 		return
 	}
 
-	data["ErrorMsg"] = err
-	log.Error("repo.Create: %v", err)
-	r.HTML(200, "base/error", data)
+	eh.Handle("repo.Create", r, err)
 }
 
 func Delete(form auth.DeleteRepoForm, req *http.Request, r render.Render, data base.TmplData, session sessions.Session) {

+ 1 - 0
web.go

@@ -50,6 +50,7 @@ func runWeb(*cli.Context) {
 	// Middlewares.
 	m.Use(render.Renderer(render.Options{Funcs: []template.FuncMap{AppHelpers}}))
 	m.Use(base.InitContext())
+	m.Use(log.ErrHandle())
 
 	// TODO: should use other store because cookie store is not secure.
 	store := sessions.NewCookieStore([]byte("secret123"))