1
// Copyright 2018 The CubeFS Authors.
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
7
// http://www.apache.org/licenses/LICENSE-2.0
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
12
// implied. See the License for the specific language governing
13
// permissions and limitations under the License.
22
// ExtentMapItem stores the extent entity pointer and the element
23
// pointer of the extent entity in a cache list.
24
type ExtentMapItem struct {
29
// ExtentCache is an implementation of the ExtentCache with LRU support.
30
type ExtentCache struct {
31
extentMap map[uint64]*ExtentMapItem
33
tinyExtents map[uint64]*Extent
39
// NewExtentCache creates and returns a new ExtentCache instance.
40
func NewExtentCache(capacity int) *ExtentCache {
42
extentMap: make(map[uint64]*ExtentMapItem),
43
extentList: list.New(),
45
tinyExtents: make(map[uint64]*Extent),
49
// Put puts an extent object into the cache.
50
func (cache *ExtentCache) Put(e *Extent) {
51
if IsTinyExtent(e.extentID) {
53
cache.tinyExtents[e.extentID] = e
54
cache.tinyLock.Unlock()
58
defer cache.lock.Unlock()
59
item := &ExtentMapItem{
61
element: cache.extentList.PushBack(e),
63
cache.extentMap[e.extentID] = item
67
// Get gets the extent from the cache.
68
func (cache *ExtentCache) Get(extentID uint64) (e *Extent, ok bool) {
69
if IsTinyExtent(extentID) {
70
cache.tinyLock.RLock()
71
e, ok = cache.tinyExtents[extentID]
72
cache.tinyLock.RUnlock()
76
defer cache.lock.Unlock()
77
var item *ExtentMapItem
78
if item, ok = cache.extentMap[extentID]; ok {
79
if !IsTinyExtent(extentID) {
80
cache.extentList.MoveToBack(item.element)
87
// Del deletes the extent stored in the cache.
88
func (cache *ExtentCache) Del(extentID uint64) {
89
if IsTinyExtent(extentID) {
93
defer cache.lock.Unlock()
98
if item, ok = cache.extentMap[extentID]; ok {
99
delete(cache.extentMap, extentID)
100
cache.extentList.Remove(item.element)
106
// Clear closes all the extents stored in the cache.
107
func (cache *ExtentCache) Clear() {
108
cache.tinyLock.RLock()
109
for _, extent := range cache.tinyExtents {
112
cache.tinyLock.RUnlock()
115
defer cache.lock.Unlock()
116
for e := cache.extentList.Front(); e != nil; {
119
ec := curr.Value.(*Extent)
120
delete(cache.extentMap, ec.extentID)
123
cache.extentList.Remove(curr)
125
cache.extentList = list.New()
126
cache.extentMap = make(map[uint64]*ExtentMapItem)
129
// Size returns number of extents stored in the cache.
130
func (cache *ExtentCache) Size() int {
132
defer cache.lock.RUnlock()
133
return cache.extentList.Len()
136
func (cache *ExtentCache) evict() {
137
if cache.capacity <= 0 {
140
needRemove := cache.extentList.Len() - cache.capacity
141
for i := 0; i < needRemove; i++ {
142
if e := cache.extentList.Front(); e != nil {
143
front := e.Value.(*Extent)
144
if IsTinyExtent(front.extentID) {
147
delete(cache.extentMap, front.extentID)
148
cache.extentList.Remove(e)
154
// Flush synchronizes the extent stored in the cache to the disk.
155
func (cache *ExtentCache) Flush() {
156
cache.tinyLock.RLock()
157
for _, extent := range cache.tinyExtents {
160
cache.tinyLock.RUnlock()
163
defer cache.lock.RUnlock()
164
for _, item := range cache.extentMap {