|
|
@@ -13,7 +13,7 @@ import (
|
|
|
)
|
|
|
|
|
|
var bindHost, jsonEncoder, dbDriver, dbConnectionString string
|
|
|
-var prefork bool
|
|
|
+var prefork, useQuickTemplate bool
|
|
|
|
|
|
func init() {
|
|
|
// init flags
|
|
|
@@ -22,6 +22,7 @@ func init() {
|
|
|
flag.StringVar(&jsonEncoder, "json_encoder", "none", "json encoder: none, easyjson or sjson")
|
|
|
flag.StringVar(&dbDriver, "db", "none", "db connection driver [values: none or pgx or mongo]")
|
|
|
flag.StringVar(&dbConnectionString, "db_connection_string", "", "db connection string")
|
|
|
+ flag.BoolVar(&useQuickTemplate, "quicktemplate", false, "use quicktemplate")
|
|
|
|
|
|
flag.Parse()
|
|
|
}
|
|
|
@@ -60,6 +61,13 @@ func main() {
|
|
|
var dbHandler atreugo.View
|
|
|
var queriesHandler atreugo.View
|
|
|
var updateHandler atreugo.View
|
|
|
+ var fortuneHandler atreugo.View
|
|
|
+
|
|
|
+ if useQuickTemplate {
|
|
|
+ fortuneHandler = handlers.FortuneQuickHandler(db)
|
|
|
+ } else {
|
|
|
+ fortuneHandler = handlers.FortuneHandler(db)
|
|
|
+ }
|
|
|
|
|
|
switch jsonEncoder {
|
|
|
case "easyjson":
|
|
|
@@ -91,8 +99,7 @@ func main() {
|
|
|
server.GET("/json", jsonHandler)
|
|
|
server.GET("/db", dbHandler)
|
|
|
server.GET("/queries", queriesHandler)
|
|
|
- server.GET("/fortune", handlers.FortuneHandler(db))
|
|
|
- server.GET("/fortune-quick", handlers.FortuneQuickHandler(db))
|
|
|
+ server.GET("/fortune", fortuneHandler)
|
|
|
server.GET("/update", updateHandler)
|
|
|
|
|
|
if err := server.ListenAndServe(); err != nil {
|
Sergio Andrés Virviescas Santana