Переглянути джерело

Merge pull request #51 from Jinnrry/bugfix-init

bug fix
木木的木头 2 роки тому
батько
коміт
db49e14f3d
2 змінених файлів з 12 додано та 4 видалено
  1. 3 0
      server/config/config.go
  2. 9 4
      server/db/init.go

+ 3 - 0
server/config/config.go

@@ -91,6 +91,9 @@ func Init() {
 			if e != nil {
 				panic(e)
 			}
+			if len(i) == 0 || strings.TrimSpace(string(i)) == "" {
+				return nil
+			}
 			if strings.Contains(path, "data") {
 				Instance.TablesInitData[tableName] = string(i)
 			} else {

+ 9 - 4
server/db/init.go

@@ -68,6 +68,9 @@ func checkTable() {
 	}
 
 	for tableName, createSQL := range config.Instance.Tables {
+		if createSQL == "" {
+			continue
+		}
 		if _, ok := existTable[tableName]; !ok {
 			_, err = Instance.Exec(createSQL)
 			log.Infof("Create Table: %s", createSQL)
@@ -76,10 +79,12 @@ func checkTable() {
 			}
 
 			if initData, ok := config.Instance.TablesInitData[tableName]; ok {
-				_, err = Instance.Exec(initData)
-				log.Infof("Init Table: %s", initData)
-				if err != nil {
-					panic(err)
+				if initData != "" {
+					_, err = Instance.Exec(initData)
+					log.Infof("Init Table: %s", initData)
+					if err != nil {
+						panic(err)
+					}
 				}
 			}