支持模板渲染 / 修复 bug

pkg/renders/render.go

@@ -0,0 +1,29 @@
+package renders
+
+import (
+	"io"
+	"net/http"
+	"sync"
+)
+
+var (
+	renders = make(map[string]Render)
+	lock    = &sync.Mutex{}
+)
+
+type Render interface {
+	Render(w http.ResponseWriter, r *http.Request, input io.Reader) error
+}
+
+func RegisterRender(fType string, r Render) {
+	lock.Lock()
+	defer lock.Unlock()
+	if renders[fType] != nil {
+		panic("duplicate render type: " + fType)
+	}
+	renders[fType] = r
+}
+
+func GetRender(key string) Render {
+	return renders[key]
+}