33 lines
893 B
Go
33 lines
893 B
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"kenaito-dns/dao"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var KeyResolveRecordMap sync.Map
|
||
|
|
var IdResolveRecordMap sync.Map
|
||
|
|
|
||
|
|
func ReloadCache() {
|
||
|
|
resolveRecords := dao.FindResolveRecordByVersion(dao.GetResolveVersion())
|
||
|
|
for _, record := range resolveRecords {
|
||
|
|
// id -> resolveRecord
|
||
|
|
IdResolveRecordMap.Store(record.Id, record)
|
||
|
|
// key -> resolveRecord
|
||
|
|
cacheKey := fmt.Sprintf("%s-%s", record.Name, record.RecordType)
|
||
|
|
records, ok := KeyResolveRecordMap.Load(cacheKey)
|
||
|
|
if !ok {
|
||
|
|
fmt.Println("读取缓存失败, key=" + cacheKey)
|
||
|
|
var tempRecords []dao.ResolveRecord
|
||
|
|
tempRecords = append(tempRecords, record)
|
||
|
|
KeyResolveRecordMap.Store(cacheKey, tempRecords)
|
||
|
|
} else {
|
||
|
|
fmt.Println("读取缓存成功, key=" + cacheKey)
|
||
|
|
var newRecords = records.([]dao.ResolveRecord)
|
||
|
|
records = append(newRecords, record)
|
||
|
|
KeyResolveRecordMap.Store(cacheKey, records)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|