refactor(main.go): 移动记录集合获取逻辑到main函数

functions.go

@@ -62,10 +62,6 @@ func loadConfig(c *cli.Context) error {
 	if err != nil {
 		return errors.New("open database error: " + err.Error())
 	}
-	config.records, err = GetRecordCollection()
-	if err != nil {
-		return errors.New("get record collection error: " + err.Error())
-	}
 	return nil
 }
 

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"errors"
 	"fmt"
 	"os"
 
@@ -226,6 +227,10 @@ func main() {
 				logger.Debugf("[MAIN]: update mode: %s", mode)
 				config.isIncrementalUpdate = !c.Bool("no-incremental-update") //是否使用增量更新
 				logger.Debugf("[MAIN]: incremental update: %t", config.isIncrementalUpdate)
+				config.records, err = GetRecordCollection()
+				if err != nil {
+					return errors.New("get record collection error: " + err.Error())
+				}
 				localStrms := make(map[string]*Strm, 0)
 				remoteStrms := make(map[string]*Strm, 0)
 				addStrms := make([]*Strm, 0)

strm.go

@@ -116,6 +116,9 @@ func GetRecordCollection() (map[string]int, error) {
 		// Assume bucket exists and has keys
 		b := tx.Bucket([]byte("strm"))
 		byts := b.Get([]byte("records"))
+		if byts == nil {
+			return fmt.Errorf("records not found, must use update-database first")
+		}
 		return json.Unmarshal(byts, &records)
 	})
 	if err != nil {