A sqlite close

goploy-devops · Mar 28, 2022 · c923975 · c923975
1 parent eddfb49
commit c923975
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -107,6 +107,12 @@ func main() {
 			fmt.Printf("Task shutdown failed, err: %s\n", err.Error())
 		}
 		println("Task shutdown gracefully")
+
+		println("SQLite is trying to shutdown, wait for a minute")
+		if err := model.Shutdown(); err != nil {
+			fmt.Printf("SQLite shutdown failed, err: %s\n", err.Error())
+		}
+		println("SQLite shutdown gracefully")
 	}()
 	if config.Toml.Web.Port != "" {
 		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {

diff --git a/model/Model.go b/model/Model.go
@@ -51,6 +51,10 @@ func Init() {
 	core.Log(core.INFO, fmt.Sprintf("server id %d", goployServerID))
 }
 
+func Shutdown() error {
+	return DB.Close()
+}
+
 func createTable() error {
 	conn := DB.Get(nil)
 	defer DB.Put(conn)