server/warmup.go at main · urnetwork/server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package server

import (
	"sync"
)

var warmupLock sync.Mutex
var warmupUnits []func()
var resetUnits []func()
var warmedUp = false

func OnWarmup(unit func()) {
	runNow := false
	func() {
		warmupLock.Lock()
		defer warmupLock.Unlock()

		if warmedUp {
			runNow = true
		} else {
			warmupUnits = append(warmupUnits, unit)
		}
	}()
	if runNow {
		unit()
	}
}

func Warmup() {
	var runUnits []func()
	func() {
		warmupLock.Lock()
		defer warmupLock.Unlock()

		runUnits = warmupUnits
		warmupUnits = nil
		warmedUp = true
	}()
	for _, unit := range runUnits {
		unit()
	}
}

func OnReset(unit func()) {
	warmupLock.Lock()
	defer warmupLock.Unlock()

	resetUnits = append(resetUnits, unit)
}

func Reset() {
	var runUnits []func()
	func() {
		warmupLock.Lock()
		defer warmupLock.Unlock()

		runUnits = resetUnits
		warmupUnits = nil
		warmedUp = false
	}()
	for _, unit := range runUnits {
		unit()
	}
}