fix: nil reference while trying to shut down DB in slave mode (#1416)

cloudreve · Dec 16, 2022 · 435a03d · 435a03d
1 parent 4e8ab75
commit 435a03d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -42,7 +42,11 @@ func init() {
 
 func main() {
 	// 关闭数据库连接
-	defer model.DB.Close()
+	defer func() {
+		if model.DB != nil {
+			model.DB.Close()
+		}
+	}()
 
 	if isEject {
 		// 开始导出内置静态资源文件