67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package httplib
|
||
|
||
// 这个暂时用不到
|
||
|
||
import (
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
// DeadCache 死掉的实例缓存
|
||
type DeadCache struct {
|
||
sync.RWMutex
|
||
Data map[string]int64
|
||
Duration time.Duration
|
||
}
|
||
|
||
// NewDeadCache 实例化DeadCache，每个HTTPServer只对应一个DeadCache
|
||
func NewDeadCache(duration time.Duration) DeadCache {
|
||
deadCache := DeadCache{
|
||
Duration: duration,
|
||
Data: make(map[string]int64),
|
||
}
|
||
|
||
go deadCache.Clean()
|
||
return deadCache
|
||
}
|
||
|
||
// Set 放置一个挂掉的实例
|
||
func (dc *DeadCache) Set(key string) {
|
||
dc.Lock()
|
||
defer dc.Unlock()
|
||
dc.Data[key] = time.Now().Unix()
|
||
}
|
||
|
||
// Exists 检查某个实例是否存在
|
||
func (dc *DeadCache) Exists(key string) bool {
|
||
dc.RLock()
|
||
defer dc.RUnlock()
|
||
_, exists := dc.Data[key]
|
||
return exists
|
||
}
|
||
|
||
// Size 计算所有挂掉的实例的个数
|
||
func (dc *DeadCache) Size() int {
|
||
dc.RLock()
|
||
defer dc.RUnlock()
|
||
return len(dc.Data)
|
||
}
|
||
|
||
// Clean 缓存清理方法，需要外层起一个goroutine来调用
|
||
func (dc *DeadCache) Clean() []string {
|
||
t1 := time.NewTicker(1 * time.Second)
|
||
|
||
for {
|
||
<-t1.C
|
||
|
||
now := time.Now().Unix()
|
||
dc.Lock()
|
||
for instance, timestamp := range dc.Data {
|
||
if now-timestamp > int64(dc.Duration) {
|
||
delete(dc.Data, instance)
|
||
}
|
||
}
|
||
dc.Unlock()
|
||
}
|
||
}
|