Merge branch 'master' of github.com:astaxie/beego

docs/zh/Quickstart.md

@@ -115,7 +115,7 @@
 
 		2013/04/13 19:36:17 [W] [stat views: no such file or directory]
 
-- 模板会自动重新加载不缓存。
+- 模板每次使用都会重新加载，不进行缓存。
 - 如果服务端出错，那么就会在浏览器端显示如下类似的截图：
 
 ![](images/dev.png)
@@ -643,11 +643,11 @@ beego更加人性化的还有一个设计就是支持用户自定义字符串错
 
 ## response处理
 
-response可能会有集中情况：
+response可能会有几种情况：
 
 1. 模板输出
 
-	模板输出上面模板介绍里面已经介绍，beego会在执行完相应的Controller里面的对应的Method之后输出到模板。
+	上面模板介绍里面已经介绍，beego会在执行完相应的Controller里面的对应的Method之后输出到模板。
 
 2. 跳转
 
@@ -686,7 +686,7 @@ beego中使用session相当方便，只要在main入口函数中设置如下：
 		this.TplNames = "index.tpl"
 	}
 
-上面的例子中我们知道session有几个方便的方法：
+session有几个方便的方法：
 
 - SetSession(name string, value interface{})
 - GetSession(name string) interface{}

template.go

@@ -51,34 +51,35 @@ func (self *templatefile) visit(paths string, f os.FileInfo, err error) error {
 	if f == nil {
 		return err
 	}
-	if f.IsDir() {
+	if f.IsDir() || (f.Mode()&os.ModeSymlink) > 0 {
 		return nil
-	} else if (f.Mode() & os.ModeSymlink) > 0 {
+	}
+	if !HasTemplateEXt(paths) {
 		return nil
+	}
+
+	replace := strings.NewReplacer("\\", "/")
+	a := []byte(paths)
+	a = a[len([]byte(self.root)):]
+	subdir := path.Dir(strings.TrimLeft(replace.Replace(string(a)), "/"))
+	if _, ok := self.files[subdir]; ok {
+		self.files[subdir] = append(self.files[subdir], paths)
 	} else {
-		hasExt := false
-		for _, v := range BeeTemplateExt {
-			if strings.HasSuffix(paths, v) {
-				hasExt = true
-				break
-			}
-		}
-		if hasExt {
-			replace := strings.NewReplacer("\\", "/")
-			a := []byte(paths)
-			a = a[len([]byte(self.root)):]
-			subdir := path.Dir(strings.TrimLeft(replace.Replace(string(a)), "/"))
-			if _, ok := self.files[subdir]; ok {
-				self.files[subdir] = append(self.files[subdir], paths)
-			} else {
-				m := make([]string, 1)
-				m[0] = paths
-				self.files[subdir] = m
-			}
+		m := make([]string, 1)
+		m[0] = paths
+		self.files[subdir] = m
+	}
 
+	return nil
+}
+
+func HasTemplateEXt(paths string) bool {
+	for _, v := range BeeTemplateExt {
+		if strings.HasSuffix(paths, "."+v) {
+			return true
 		}
 	}
-	return nil
+	return false
 }
 
 func AddTemplateExt(ext string) {

template_test.go

+package beego
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestBuildTemplate(t *testing.T) {
+	dir := "_beeTmp"
+	files := []string{
+		"1.tpl",
+		"2.html",
+		"3.htmltpl",
+		"4.mystyle",
+	}
+	if err := os.MkdirAll(dir, 0777); err != nil {
+		t.Fatal(err)
+	}
+	for _, name := range files {
+		if _, err := os.Create(filepath.Join(dir, name)); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if err := BuildTemplate(dir); err != nil {
+		t.Fatal(err)
+	}
+	if len(BeeTemplates) != 1 {
+		t.Fatalf("should be 1 but got %v", len(BeeTemplates))
+	}
+	for _, v := range BeeTemplates {
+		if len(v.Templates()) != 3 {
+			t.Errorf("should be 3 but got %v", len(v.Templates()))
+		}
+	}
+
+	AddTemplateExt("mystyle")
+	if err := BuildTemplate(dir); err != nil {
+		t.Fatal(err)
+	}
+	if len(BeeTemplates) != 1 {
+		t.Fatalf("should be 1 but got %v", len(BeeTemplates))
+	}
+	for _, v := range BeeTemplates {
+		if len(v.Templates()) != 4 {
+			t.Errorf("should be 4 but got %v", len(v.Templates()))
+		}
+	}
+}