程序意外退出时关闭sqlite指针

internal/events/events.go

@@ -3,8 +3,9 @@ package events
 type Event = string
 
 const (
-	EventStart  Event = "start"  // start loading
-	EventLoaded Event = "loaded" // first load
-	EventQuit   Event = "quit"   // quit node gracefully
-	EventReload Event = "reload" // reload config
+	EventStart      Event = "start"      // start loading
+	EventLoaded     Event = "loaded"     // first load
+	EventQuit       Event = "quit"       // quit node gracefully
+	EventReload     Event = "reload"     // reload config
+	EventTerminated Event = "terminated" // process terminated
 )

internal/events/utils.go

@@ -1,6 +1,7 @@
 package events
 
 import (
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
 	"sync"
 )
 
@@ -59,6 +60,14 @@ func Remove(key interface{}) {
 
 // Notify 通知事件
 func Notify(event Event) {
+	// 特殊事件
+	switch event {
+	case EventQuit:
+		teaconst.IsQuiting = true
+	case EventTerminated:
+		teaconst.IsQuiting = true
+	}
+
 	locker.Lock()
 	m := eventsMap[event]
 	locker.Unlock()