优化OpenFileCache

This commit is contained in:
刘祥超
2022-12-05 11:16:04 +08:00
parent 1a05f56149
commit 87bc1a7e03
2 changed files with 24 additions and 2 deletions

View File

@@ -129,6 +129,9 @@ func (this *OpenFileCache) Close(filename string) {
pool, ok := this.poolMap[filename] pool, ok := this.poolMap[filename]
if ok { if ok {
// 设置关闭状态
pool.SetClosing()
delete(this.poolMap, filename) delete(this.poolMap, filename)
this.poolList.Remove(pool.linkItem) this.poolList.Remove(pool.linkItem)
_ = this.watcher.Remove(filename) _ = this.watcher.Remove(filename)

View File

@@ -12,6 +12,7 @@ type OpenFilePool struct {
linkItem *linkedlist.Item linkItem *linkedlist.Item
filename string filename string
version int64 version int64
isClosed bool
} }
func NewOpenFilePool(filename string) *OpenFilePool { func NewOpenFilePool(filename string) *OpenFilePool {
@@ -29,6 +30,11 @@ func (this *OpenFilePool) Filename() string {
} }
func (this *OpenFilePool) Get() (*OpenFile, bool) { func (this *OpenFilePool) Get() (*OpenFile, bool) {
// 如果已经关闭,直接返回
if this.isClosed {
return nil, false
}
select { select {
case file := <-this.c: case file := <-this.c:
if file != nil { if file != nil {
@@ -48,10 +54,19 @@ func (this *OpenFilePool) Get() (*OpenFile, bool) {
} }
func (this *OpenFilePool) Put(file *OpenFile) bool { func (this *OpenFilePool) Put(file *OpenFile) bool {
// 如果已关闭,则不接受新的文件
if this.isClosed {
_ = file.Close()
return false
}
// 检查文件版本号
if this.version > 0 && file.version > 0 && file.version != this.version { if this.version > 0 && file.version > 0 && file.version != this.version {
_ = file.Close() _ = file.Close()
return false return false
} }
// 加入Pool
select { select {
case this.c <- file: case this.c <- file:
return true return true
@@ -66,14 +81,18 @@ func (this *OpenFilePool) Len() int {
return len(this.c) return len(this.c)
} }
func (this *OpenFilePool) SetClosing() {
this.isClosed = true
}
func (this *OpenFilePool) Close() { func (this *OpenFilePool) Close() {
Loop: this.isClosed = true
for { for {
select { select {
case file := <-this.c: case file := <-this.c:
_ = file.Close() _ = file.Close()
default: default:
break Loop return
} }
} }
} }