EdgeNode/internal/caches/list_file_hash_map.go

// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .

package caches

import (
	"github.com/TeaOSLab/EdgeNode/internal/zero"
	"sync"
)

// FileListHashMap
type FileListHashMap struct {
	m       map[string]zero.Zero
	locker  sync.RWMutex
	isReady bool
}

func NewFileListHashMap() *FileListHashMap {
	return &FileListHashMap{
		m:       map[string]zero.Zero{},
		isReady: false,
	}
}

func (this *FileListHashMap) Load(db *FileListDB) error {
	var lastId int64
	for {
		hashList, maxId, err := db.ListHashes(lastId)
		if err != nil {
			return err
		}
		if len(hashList) == 0 {
			break
		}
		for _, hash := range hashList {
			this.Add(hash)
		}
		lastId = maxId
	}

	this.isReady = true
	return nil
}

func (this *FileListHashMap) Add(hash string) {
	this.locker.Lock()
	this.m[hash] = zero.New()
	this.locker.Unlock()
}

func (this *FileListHashMap) Delete(hash string) {
	this.locker.Lock()
	delete(this.m, hash)
	this.locker.Unlock()
}

func (this *FileListHashMap) Exist(hash string) bool {
	if !this.isReady {
		// 只有完全Ready时才能判断是否为false
		return true
	}
	this.locker.RLock()
	_, ok := this.m[hash]
	this.locker.RUnlock()
	return ok
}

func (this *FileListHashMap) Clean() {
	this.locker.Lock()
	this.m = map[string]zero.Zero{}
	this.locker.Unlock()
}

func (this *FileListHashMap) IsReady() bool {
	return this.isReady
}