Dragonfly2

1
/*
2
 *     Copyright 2020 The Dragonfly Authors
3
 *
4
 * Licensed under the Apache License, Version 2.0 (the "License");
5
 * you may not use this file except in compliance with the License.
6
 * You may obtain a copy of the License at
7
 *
8
 *      http://www.apache.org/licenses/LICENSE-2.0
9
 *
10
 * Unless required by applicable law or agreed to in writing, software
11
 * distributed under the License is distributed on an "AS IS" BASIS,
12
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
 * See the License for the specific language governing permissions and
14
 * limitations under the License.
15
 */
16

17
package gc
18

19
import (
20
	"time"
21

22
	logger "d7y.io/dragonfly/v2/internal/dflog"
23
)
24

25
type GC interface {
26
	TryGC() (bool, error)
27
}
28

29
type Manager interface {
30
	Start()
31
	Stop()
32
}
33

34
type gcManager struct {
35
	interval time.Duration
36
	done     chan bool
37
}
38

39
var _ Manager = (*gcManager)(nil)
40

41
var allGCTasks = map[string]GC{}
42

43
func Register(name string, gc GC) {
44
	allGCTasks[name] = gc
45
}
46

47
func NewManager(interval time.Duration) Manager {
48
	return &gcManager{
49
		interval: interval,
50
		done:     make(chan bool),
51
	}
52
}
53

54
func (g gcManager) Start() {
55
	go func() {
56
		tick := time.NewTicker(g.interval)
57
		for {
58
			select {
59
			case <-tick.C:
60
				for name, gc := range allGCTasks {
61
					var log = logger.With("component", name)
62
					log.Debugf("start gc")
63
					_, err := gc.TryGC()
64
					if err != nil {
65
						log.Errorf("gc %s error: %s", name, err)
66
					}
67
					log.Debugf("gc done")
68
				}
69
			case <-g.done:
70
				logger.Infof("gc exited")
71
				return
72
			}
73
		}
74
	}()
75
}
76

77
func (g gcManager) Stop() {
78
	close(g.done)
79
}
80