Hochwasser/rpc/dottir.go

131 lines
2.7 KiB
Go
Raw Normal View History

package rpc
import (
"fmt"
"image"
"log"
2020-02-11 14:46:58 +01:00
"net"
"net/rpc"
"sync"
2020-02-12 11:20:15 +01:00
"time"
2020-02-12 11:20:15 +01:00
"github.com/SpeckiJ/Hochwasser/pixelflut"
"github.com/SpeckiJ/Hochwasser/render"
2020-02-12 11:20:15 +01:00
)
2020-02-11 14:46:58 +01:00
func ConnectHevring(ránAddress string, stop chan bool, wg *sync.WaitGroup) *Hevring {
h := new(Hevring)
rpc.Register(h)
2020-02-11 14:46:58 +01:00
fmt.Printf("[hevring] greeting Rán at %s\n", ránAddress)
conn, err := net.Dial("tcp", ránAddress)
if err != nil {
log.Fatal(err)
}
2020-02-11 14:46:58 +01:00
go rpc.ServeConn(conn)
fmt.Printf("[hevring] awaiting task from Rán\n")
2022-01-02 08:45:11 +01:00
// print performance
go func() {
for {
time.Sleep(5 * time.Second)
if pixelflut.PerformanceReporter.Enabled {
fmt.Println(pixelflut.PerformanceReporter)
}
}
}()
// add listener to stop the task, if this hevring should stop
// (either because Rán told us so, or we received an interrupt)
h.quit = stop
h.wg = wg
h.wg.Add(1)
go func() {
<-h.quit
h.quit = nil
if h.taskQuit != nil {
close(h.taskQuit)
h.taskQuit = nil
}
h.wg.Done()
}()
return h
2020-02-11 14:46:58 +01:00
}
2020-02-12 13:55:04 +01:00
type Hevring struct {
PreviewPath string
task pixelflut.FlutTask
2022-01-02 08:45:11 +01:00
taskQuit chan bool // if closed, task is stopped.
quit chan bool // if closed, kills this hevring
wg *sync.WaitGroup
2020-02-12 13:55:04 +01:00
}
2020-02-12 11:20:15 +01:00
2020-02-11 14:46:58 +01:00
type FlutAck struct{ Ok bool }
type FlutStatus struct {
*pixelflut.Performance
Ok bool
Fluting bool
}
2020-12-31 18:44:56 +01:00
func (h *Hevring) Flut(task pixelflut.FlutTask, reply *FlutAck) error {
2020-02-14 22:29:58 +01:00
// stop old task if new task is received
if h.taskQuit != nil {
close(h.taskQuit)
2020-02-12 13:55:04 +01:00
}
fmt.Printf("[hevring] Rán gave us work!\n%v\n", task)
2020-02-12 13:55:04 +01:00
h.task = task
h.taskQuit = make(chan bool)
2020-12-31 18:44:56 +01:00
go pixelflut.Flut(task, h.taskQuit, nil)
go h.savePreview(task.Img)
2020-02-11 14:46:58 +01:00
reply.Ok = true
return nil
}
func (h *Hevring) Status(metrics bool, reply *FlutStatus) error {
pixelflut.PerformanceReporter.Enabled = metrics
reply.Performance = pixelflut.PerformanceReporter
2020-02-11 14:46:58 +01:00
reply.Ok = true
reply.Fluting = h.taskQuit != nil
2020-02-11 14:46:58 +01:00
return nil
}
func (h *Hevring) Stop(x int, reply *FlutAck) error {
2020-02-14 22:29:58 +01:00
if h.taskQuit != nil {
2020-02-12 13:55:04 +01:00
fmt.Println("[hevring] stopping task")
2020-12-31 18:44:56 +01:00
h.task = pixelflut.FlutTask{}
close(h.taskQuit)
h.taskQuit = nil
2020-02-12 13:55:04 +01:00
reply.Ok = true
}
2020-02-11 14:46:58 +01:00
return nil
}
2020-02-12 11:20:15 +01:00
func (h *Hevring) Die(x int, reply *FlutAck) error {
2020-02-14 22:29:58 +01:00
// @robustness: waiting for reply to be sent via timeout
// @incomplete: should try to reconnect for a bit first
go func() {
fmt.Println("[hevring] Rán disconnected, stopping")
time.Sleep(100 * time.Millisecond)
if h.quit != nil {
close(h.quit)
}
2020-02-12 11:20:15 +01:00
}()
reply.Ok = true
return nil
}
func (h Hevring) savePreview(img image.Image) {
if h.PreviewPath != "" && img != nil {
err := render.WriteImage(h.PreviewPath, img)
if err != nil {
fmt.Printf("[hevring] unable to write preview: %s\n", err)
}
}
}