/
thread.go
83 lines (71 loc) · 1.83 KB
/
thread.go
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// Copyright 2018 The Ebiten Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package thread
import (
"context"
"runtime"
)
// OSThread represents an OS thread.
type OSThread struct {
funcs chan func()
done chan struct{}
}
// NewOSThread creates a new thread.
func NewOSThread() *OSThread {
return &OSThread{
funcs: make(chan func()),
done: make(chan struct{}),
}
}
// Loop starts the thread loop until Stop is called on the current OS thread.
//
// Loop must be called on the thread.
func (t *OSThread) Loop(ctx context.Context) error {
runtime.LockOSThread()
defer runtime.UnlockOSThread()
for {
select {
case fn := <-t.funcs:
func() {
defer func() {
t.done <- struct{}{}
}()
fn()
}()
case <-ctx.Done():
return ctx.Err()
}
}
}
// Call calls f on the thread.
//
// Do not call this from the same thread. This would block forever.
//
// Call blocks if Loop is not called.
func (t *OSThread) Call(f func()) {
t.funcs <- f
<-t.done
}
// NoopThread is used to disable threading.
type NoopThread struct{}
// NewNoopThread creates a new thread that does no threading.
func NewNoopThread() *NoopThread {
return &NoopThread{}
}
// Loop does nothing
func (t *NoopThread) Loop(ctx context.Context) error {
return nil
}
// Call executes the func immediately
func (t *NoopThread) Call(f func()) { f() }