82 lines
1.3 KiB
Go
82 lines
1.3 KiB
Go
package common
|
|
|
|
import (
|
|
"sync/atomic"
|
|
)
|
|
|
|
var TopicListThaw ThawInt
|
|
|
|
type ThawInt interface {
|
|
Thawed() bool
|
|
Thaw()
|
|
|
|
Tick() error
|
|
}
|
|
|
|
type SingleServerThaw struct {
|
|
DefaultThaw
|
|
}
|
|
|
|
func NewSingleServerThaw() *SingleServerThaw {
|
|
t := &SingleServerThaw{}
|
|
if Config.ServerCount == 1 {
|
|
Tasks.Sec.Add(t.Tick)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func (t *SingleServerThaw) Thawed() bool {
|
|
if Config.ServerCount == 1 {
|
|
return t.DefaultThaw.Thawed()
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (t *SingleServerThaw) Thaw() {
|
|
if Config.ServerCount == 1 {
|
|
t.DefaultThaw.Thaw()
|
|
}
|
|
}
|
|
|
|
type TestThaw struct {
|
|
}
|
|
|
|
func NewTestThaw() *TestThaw {
|
|
return &TestThaw{}
|
|
}
|
|
func (t *TestThaw) Thawed() bool {
|
|
return true
|
|
}
|
|
func (t *TestThaw) Thaw() {
|
|
}
|
|
func (t *TestThaw) Tick() error {
|
|
return nil
|
|
}
|
|
|
|
type DefaultThaw struct {
|
|
thawed int64
|
|
}
|
|
|
|
func NewDefaultThaw() *DefaultThaw {
|
|
t := &DefaultThaw{}
|
|
Tasks.Sec.Add(t.Tick)
|
|
return t
|
|
}
|
|
|
|
// Decrement the thawed counter once a second until it goes cold
|
|
func (t *DefaultThaw) Tick() error {
|
|
prior := t.thawed
|
|
if prior > 0 {
|
|
atomic.StoreInt64(&t.thawed, prior-1)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (t *DefaultThaw) Thawed() bool {
|
|
return t.thawed > 0
|
|
}
|
|
|
|
func (t *DefaultThaw) Thaw() {
|
|
atomic.StoreInt64(&t.thawed, 3)
|
|
}
|