增加手动建索引功能

xuxinhua · xuxinhua · commit cc00529a7553 · 2015-03-08T21:47:08.000+08:00
diff --git a/websites/code/studygolang/src/server/indexer/main.go b/websites/code/studygolang/src/server/indexer/main.go
@@ -10,6 +10,7 @@ import (
 	"math/rand"
 	"runtime"
 	"time"
+	"flag"
 	//"path/filepath"
 
 	"github.com/robfig/cron"
@@ -25,26 +26,37 @@ func init() {
 }
 
 func main() {
+	var manualIndex bool
+	flag.BoolVar(&manualIndex, "manual", false, "do manual index once or not")
+	flag.Parse()
+
+	if manualIndex {
+		indexing(true)
+	}
 
 	c := cron.New()
 	// 构建 solr 需要的索引数据
 	// 一天一次全量
 	c.AddFunc("@daily", func() {
-		logger.Infoln("indexing start...")
-
-		start := time.Now()
-		defer func() {
-			logger.Infoln("indexing spend time:", time.Now().Sub(start))
-		}()
-
-		service.Indexing(true)
+		indexing(true)
 	})
 
 	c.Start()
 
 	select {}
 }
 
+func indexing(isAll bool) {
+	logger.Infoln("indexing start...")
+
+	start := time.Now()
+	defer func() {
+		logger.Infoln("indexing spend time:", time.Now().Sub(start))
+	}()
+
+	service.Indexing(isAll)
+}
+
 // 保存PID
 func SavePid() {
 	/*
