🚧 构建目录树

bootstrap/model.go

@@ -12,24 +12,28 @@ import (
 func InitModel() bool {
 	log.Infof("初始化数据库...")
 	switch conf.Conf.Database.Type {
-	case "sqlite3":{
-		needMigrate := !utils.Exists(conf.Conf.Database.DBFile)
-		db, err := gorm.Open(sqlite.Open(conf.Conf.Database.DBFile), &gorm.Config{})
-		if err!=nil {
-			log.Errorf("连接数据库出现错误:%s", err.Error())
-			return false
-		}
-		conf.DB = db
-		if needMigrate {
-			log.Infof("迁移数据库...")
-			err = conf.DB.AutoMigrate(&models.File{})
-			if err !=nil {
-				log.Errorf("数据库迁移失败:%s",err.Error())
+	case "sqlite3":
+		{
+			needMigrate := !utils.Exists(conf.Conf.Database.DBFile)
+			db, err := gorm.Open(sqlite.Open(conf.Conf.Database.DBFile), &gorm.Config{})
+			if err != nil {
+				log.Errorf("连接数据库出现错误:%s", err.Error())
 				return false
 			}
+			conf.DB = db
+			if needMigrate {
+				log.Infof("迁移数据库...")
+				err = conf.DB.AutoMigrate(&models.File{})
+				if err != nil {
+					log.Errorf("数据库迁移失败:%s", err.Error())
+					return false
+				}
+				if err := models.BuildTree(); err != nil {
+					log.Errorf("构建目录树失败:%s", err.Error())
+				}
+			}
+			return true
 		}
-		return true
-	}
 	default:
 		log.Errorf("不支持的数据库类型:%s", conf.Conf.Database.Type)
 		return false