-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.go
103 lines (91 loc) · 2.02 KB
/
main.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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
//go:build js && wasm
package main
import (
"fmt"
"io"
"log"
"os"
"syscall/js"
"github.com/hack-pad/safejs"
"github.com/magodo/go-wasmww"
)
func main() {
self, err := wasmww.NewSelfSharedConn()
if err != nil {
log.Fatal(err)
}
name, err := self.Name()
if err != nil {
log.Fatal(err)
}
connCh, err := self.SetupConn()
if err != nil {
log.Fatal(err)
}
logger := log.New(os.Stdout, fmt.Sprintf("Worker (%s): ", name), 0)
logger.Printf("Args: %v\n", os.Args)
logger.Printf("Env: %v\n", os.Environ())
location, err := self.Location()
if err != nil {
log.Fatal(err)
}
logger.Printf("Location: %s\n", location)
null := io.Discard
i := 0
for port := range connCh {
port := port
logger := log.New(os.Stdout, fmt.Sprintf("Worker (%s-%d): ", name, i), 0)
i++
go func() {
logger.Printf("started\n")
eventCh, err := port.SetupConn()
if err != nil {
log.Fatal(err)
}
for event := range eventCh {
data, err := event.Data()
if err != nil {
logger.Printf("Error: %v\n", err)
continue
}
str, err := data.String()
if err != nil {
logger.Printf("Error: %v\n", err)
continue
}
switch str {
case "ClosePort":
logger.Printf("Close port\n")
if err := port.Close(); err != nil {
logger.Fatal(err)
}
if self.Idle() {
logger.Printf("Close the worker for idle\n")
if err := self.Close(); err != nil {
log.Fatal(err)
}
}
continue
case "CloseWorker":
logger.Printf("Close the worker\n")
if err := self.Close(); err != nil {
log.Fatal(err)
}
case "WriteToNull":
wasmww.SetWriteSync(
[]wasmww.MsgWriter{wasmww.NewMsgWriterToIoWriter(null)},
[]wasmww.MsgWriter{wasmww.NewMsgWriterToIoWriter(null)},
)
continue
}
logger.Printf("Received message %q\n", str)
// Echo back the message
if err := port.PostMessage(safejs.Safe(js.ValueOf(str)), nil); err != nil {
logger.Printf("Error: %v\n", err)
continue
}
}
}()
}
select {}
}