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
|
package service
import (
"github.com/spiral/roadrunner"
"net"
"os/exec"
"strings"
"time"
)
type PoolConfig struct {
Command string
Relay string
Number uint64
MaxJobs uint64
Timeouts struct {
Allocate int
Destroy int
}
}
func (f *PoolConfig) NewServer() (*roadrunner.Server, func(), error) {
relays, terminator, err := f.relayFactory()
if err != nil {
terminator()
return nil, nil, err
}
rr := roadrunner.NewServer(f.cmd(), relays)
if err := rr.Configure(f.rrConfig()); err != nil {
return nil, nil, err
}
return rr, nil, nil
}
func (f *PoolConfig) rrConfig() roadrunner.Config {
return roadrunner.Config{
NumWorkers: f.Number,
MaxExecutions: f.MaxJobs,
AllocateTimeout: time.Second * time.Duration(f.Timeouts.Allocate),
DestroyTimeout: time.Second * time.Duration(f.Timeouts.Destroy),
}
}
func (f *PoolConfig) cmd() func() *exec.Cmd {
cmd := strings.Split(f.Command, " ")
return func() *exec.Cmd { return exec.Command(cmd[0], cmd[1:]...) }
}
func (f *PoolConfig) relayFactory() (roadrunner.Factory, func(), error) {
if f.Relay == "pipes" || f.Relay == "pipe" {
return roadrunner.NewPipeFactory(), nil, nil
}
dsn := strings.Split(f.Relay, "://")
if len(dsn) != 2 {
return nil, nil, dsnError
}
ln, err := net.Listen(dsn[0], dsn[1])
if err != nil {
return nil, nil, err
}
return roadrunner.NewSocketFactory(ln, time.Minute), func() { ln.Close() }, nil
}
|
