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

internal/caches/errors.go

@@ -6,10 +6,11 @@ import "errors"
 
 // 常用的几个错误
 var (
-	ErrNotFound       = errors.New("cache not found")
-	ErrFileIsWriting  = errors.New("the file is writing")
-	ErrInvalidRange   = errors.New("invalid range")
-	ErrEntityTooLarge = errors.New("entity too large")
+	ErrNotFound         = errors.New("cache not found")
+	ErrFileIsWriting    = errors.New("the file is writing")
+	ErrInvalidRange     = errors.New("invalid range")
+	ErrEntityTooLarge   = errors.New("entity too large")
+	ErrWritingUnavaible = errors.New("writing unavailable")
 )
 
 // CapacityError 容量错误
@@ -31,7 +32,7 @@ func CanIgnoreErr(err error) bool {
 	if err == nil {
 		return true
 	}
-	if err == ErrFileIsWriting || err == ErrEntityTooLarge {
+	if err == ErrFileIsWriting || err == ErrEntityTooLarge || err == ErrWritingUnavaible {
 		return true
 	}
 	_, ok := err.(*CapacityError)

internal/caches/storage_file.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
 	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
 	"github.com/TeaOSLab/EdgeNode/internal/events"
 	"github.com/TeaOSLab/EdgeNode/internal/goman"
 	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
@@ -314,6 +315,11 @@ func (this *FileStorage) openReader(key string, allowMemory bool, useStale bool,
 
 // OpenWriter 打开缓存文件等待写入
 func (this *FileStorage) OpenWriter(key string, expiredAt int64, status int, size int64, maxSize int64, isPartial bool) (Writer, error) {
+	// 是否正在退出
+	if teaconst.IsQuiting {
+		return nil, ErrWritingUnavaible
+	}
+
 	// 是否已忽略
 	if this.ignoreKeys.Has(key) {
 		return nil, ErrEntityTooLarge
@@ -525,6 +531,11 @@ func (this *FileStorage) OpenWriter(key string, expiredAt int64, status int, siz
 
 // AddToList 添加到List
 func (this *FileStorage) AddToList(item *Item) {
+	// 是否正在退出
+	if teaconst.IsQuiting {
+		return
+	}
+
 	if this.memoryStorage != nil {
 		if item.Type == ItemTypeMemory {
 			this.memoryStorage.AddToList(item)
@@ -542,6 +553,11 @@ func (this *FileStorage) AddToList(item *Item) {
 
 // Delete 删除某个键值对应的缓存
 func (this *FileStorage) Delete(key string) error {
+	// 是否正在退出
+	if teaconst.IsQuiting {
+		return nil
+	}
+
 	this.locker.Lock()
 	defer this.locker.Unlock()
 
@@ -646,6 +662,11 @@ func (this *FileStorage) CleanAll() error {
 
 // Purge 清理过期的缓存
 func (this *FileStorage) Purge(keys []string, urlType string) error {
+	// 是否正在退出
+	if teaconst.IsQuiting {
+		return nil
+	}
+
 	this.locker.Lock()
 	defer this.locker.Unlock()