ebiten/internal/thread/thread.go

// 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"
)

// Thread represents an OS thread.
type Thread struct {
	funcs   chan func() error
	results chan error
	closed  chan struct{}
}

// New creates a new thread.
//
// It is assumed that the OS thread is fixed by runtime.LockOSThread when New is called.
func New() *Thread {
	return &Thread{
		funcs:   make(chan func() error),
		results: make(chan error),
		closed:  make(chan struct{}),
	}
}

// Loop starts the thread loop.
//
// Loop must be called on the thread.
func (t *Thread) Loop(context context.Context) {
	defer close(t.closed)
	defer close(t.results)
	defer close(t.funcs)
loop:
	for {
		select {
		case f := <-t.funcs:
			t.results <- f()
		case <-context.Done():
			break loop
		}
	}
}

// Call calls f on the thread.
//
// Do not call this from the same thread. This would block forever.
//
// Call panics when Loop already ends.
func (t *Thread) Call(f func() error) error {
	select {
	case t.funcs <- f:
		return <-t.results
	case <-t.closed:
		panic("thread: this thread is already terminated")
	}
}