summaryrefslogtreecommitdiff
path: root/transport/pipe
diff options
context:
space:
mode:
authorValery Piashchynski <[email protected]>2022-01-12 00:19:48 +0300
committerValery Piashchynski <[email protected]>2022-01-12 00:19:48 +0300
commit215c7c91937bf65704db18a59a327a3c64e43530 (patch)
tree85d2b32ddf7230064e620bb59f76ef4bfd6b24cf /transport/pipe
parent7b5d220f0f1be155d83d887cd4996bdf4394c570 (diff)
pass logger from the factory
Signed-off-by: Valery Piashchynski <[email protected]>
Diffstat (limited to 'transport/pipe')
-rwxr-xr-xtransport/pipe/pipe_factory.go173
-rw-r--r--transport/pipe/pipe_factory_spawn_test.go427
-rwxr-xr-xtransport/pipe/pipe_factory_test.go511
3 files changed, 0 insertions, 1111 deletions
diff --git a/transport/pipe/pipe_factory.go b/transport/pipe/pipe_factory.go
deleted file mode 100755
index c70b3f65..00000000
--- a/transport/pipe/pipe_factory.go
+++ /dev/null
@@ -1,173 +0,0 @@
-package pipe
-
-import (
- "context"
- "os/exec"
-
- "github.com/spiral/goridge/v3/pkg/pipe"
- "github.com/spiral/roadrunner/v2/internal"
- "github.com/spiral/roadrunner/v2/worker"
-)
-
-// Factory connects to stack using standard
-// streams (STDIN, STDOUT pipes).
-type Factory struct{}
-
-// NewPipeFactory returns new factory instance and starts
-// listening
-func NewPipeFactory() *Factory {
- return &Factory{}
-}
-
-type sr struct {
- w *worker.Process
- err error
-}
-
-// SpawnWorkerWithTimeout creates new Process and connects it to goridge relay,
-// method Wait() must be handled on level above.
-func (f *Factory) SpawnWorkerWithTimeout(ctx context.Context, cmd *exec.Cmd) (*worker.Process, error) {
- spCh := make(chan sr)
- go func() {
- w, err := worker.InitBaseWorker(cmd)
- if err != nil {
- select {
- case spCh <- sr{
- w: nil,
- err: err,
- }:
- return
- default:
- return
- }
- }
-
- in, err := cmd.StdoutPipe()
- if err != nil {
- select {
- case spCh <- sr{
- w: nil,
- err: err,
- }:
- return
- default:
- return
- }
- }
-
- out, err := cmd.StdinPipe()
- if err != nil {
- select {
- case spCh <- sr{
- w: nil,
- err: err,
- }:
- return
- default:
- return
- }
- }
-
- // Init new PIPE relay
- relay := pipe.NewPipeRelay(in, out)
- w.AttachRelay(relay)
-
- // Start the worker
- err = w.Start()
- if err != nil {
- select {
- case spCh <- sr{
- w: nil,
- err: err,
- }:
- return
- default:
- return
- }
- }
-
- // used as a ping
- _, err = internal.Pid(relay)
- if err != nil {
- _ = w.Kill()
- select {
- case spCh <- sr{
- w: nil,
- err: err,
- }:
- return
- default:
- _ = w.Kill()
- return
- }
- }
-
- select {
- case
- // return worker
- spCh <- sr{
- w: w,
- err: nil,
- }:
- // everything ok, set ready state
- w.State().Set(worker.StateReady)
- return
- default:
- _ = w.Kill()
- return
- }
- }()
-
- select {
- case <-ctx.Done():
- return nil, ctx.Err()
- case res := <-spCh:
- if res.err != nil {
- return nil, res.err
- }
- return res.w, nil
- }
-}
-
-func (f *Factory) SpawnWorker(cmd *exec.Cmd) (*worker.Process, error) {
- w, err := worker.InitBaseWorker(cmd)
- if err != nil {
- return nil, err
- }
-
- in, err := cmd.StdoutPipe()
- if err != nil {
- return nil, err
- }
-
- out, err := cmd.StdinPipe()
- if err != nil {
- return nil, err
- }
-
- // Init new PIPE relay
- relay := pipe.NewPipeRelay(in, out)
- w.AttachRelay(relay)
-
- // Start the worker
- err = w.Start()
- if err != nil {
- return nil, err
- }
-
- // errors bundle
- _, err = internal.Pid(relay)
- if err != nil {
- _ = w.Kill()
- return nil, err
- }
-
- // everything ok, set ready state
- w.State().Set(worker.StateReady)
- return w, nil
-}
-
-// Close the factory.
-func (f *Factory) Close() error {
- return nil
-}
diff --git a/transport/pipe/pipe_factory_spawn_test.go b/transport/pipe/pipe_factory_spawn_test.go
deleted file mode 100644
index 9aa12564..00000000
--- a/transport/pipe/pipe_factory_spawn_test.go
+++ /dev/null
@@ -1,427 +0,0 @@
-package pipe
-
-import (
- "os/exec"
- "sync"
- "testing"
- "time"
-
- "github.com/spiral/errors"
- "github.com/spiral/roadrunner/v2/payload"
- "github.com/spiral/roadrunner/v2/worker"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
-)
-
-func Test_GetState2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- go func() {
- assert.NoError(t, w.Wait())
- assert.Equal(t, worker.StateStopped, w.State().Value())
- }()
-
- assert.NoError(t, err)
- assert.NotNil(t, w)
-
- assert.Equal(t, worker.StateReady, w.State().Value())
- assert.NoError(t, w.Stop())
-}
-
-func Test_Kill2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- wg := &sync.WaitGroup{}
- wg.Add(1)
- go func() {
- defer wg.Done()
- assert.Error(t, w.Wait())
- assert.Equal(t, worker.StateErrored, w.State().Value())
- }()
-
- assert.NoError(t, err)
- assert.NotNil(t, w)
-
- assert.Equal(t, worker.StateReady, w.State().Value())
- err = w.Kill()
- if err != nil {
- t.Errorf("error killing the Process: error %v", err)
- }
- wg.Wait()
-}
-
-func Test_Pipe_Start2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.NoError(t, err)
- assert.NotNil(t, w)
-
- go func() {
- assert.NoError(t, w.Wait())
- }()
-
- assert.NoError(t, w.Stop())
-}
-
-func Test_Pipe_StartError2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- err := cmd.Start()
- if err != nil {
- t.Errorf("error running the command: error %v", err)
- }
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_PipeError3(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- _, err := cmd.StdinPipe()
- if err != nil {
- t.Errorf("error creating the STDIN pipe: error %v", err)
- }
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_PipeError4(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- _, err := cmd.StdinPipe()
- if err != nil {
- t.Errorf("error creating the STDIN pipe: error %v", err)
- }
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_Failboot2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/failboot.php")
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.Nil(t, w)
- assert.Error(t, err)
-}
-
-func Test_Pipe_Invalid2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/invalid.php")
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_Echo2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.NoError(t, err)
-
- sw := worker.From(w)
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
-
- assert.NoError(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- go func() {
- if w.Wait() != nil {
- t.Fail()
- }
- }()
-
- assert.Equal(t, "hello", res.String())
- err = w.Stop()
- assert.NoError(t, err)
-}
-
-func Test_Pipe_Broken2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "broken", "pipes")
- w, err := NewPipeFactory().SpawnWorker(cmd)
- assert.NoError(t, err)
- require.NotNil(t, w)
-
- sw := worker.From(w)
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.Error(t, err)
- assert.Nil(t, res)
-
- time.Sleep(time.Second)
- err = w.Stop()
- assert.Error(t, err)
-}
-
-func Benchmark_Pipe_SpawnWorker_Stop2(b *testing.B) {
- f := NewPipeFactory()
- for n := 0; n < b.N; n++ {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- w, _ := f.SpawnWorker(cmd)
- go func() {
- if w.Wait() != nil {
- b.Fail()
- }
- }()
-
- err := w.Stop()
- if err != nil {
- b.Errorf("error stopping the worker: error %v", err)
- }
- }
-}
-
-func Benchmark_Pipe_Worker_ExecEcho2(b *testing.B) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorker(cmd)
- sw := worker.From(w)
-
- b.ReportAllocs()
- b.ResetTimer()
- go func() {
- err := w.Wait()
- if err != nil {
- b.Errorf("error waiting the worker: error %v", err)
- }
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- b.Errorf("error stopping the worker: error %v", err)
- }
- }()
-
- for n := 0; n < b.N; n++ {
- if _, err := sw.Exec(&payload.Payload{Body: []byte("hello")}); err != nil {
- b.Fail()
- }
- }
-}
-
-func Benchmark_Pipe_Worker_ExecEcho4(b *testing.B) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- w, err := NewPipeFactory().SpawnWorker(cmd)
- if err != nil {
- b.Fatal(err)
- }
-
- defer func() {
- err = w.Stop()
- if err != nil {
- b.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- for n := 0; n < b.N; n++ {
- if _, err := sw.Exec(&payload.Payload{Body: []byte("hello")}); err != nil {
- b.Fail()
- }
- }
-}
-
-func Benchmark_Pipe_Worker_ExecEchoWithoutContext2(b *testing.B) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- w, err := NewPipeFactory().SpawnWorker(cmd)
- if err != nil {
- b.Fatal(err)
- }
-
- defer func() {
- err = w.Stop()
- if err != nil {
- b.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- for n := 0; n < b.N; n++ {
- if _, err := sw.Exec(&payload.Payload{Body: []byte("hello")}); err != nil {
- b.Fail()
- }
- }
-}
-
-func Test_Echo2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- if err != nil {
- t.Fatal(err)
- }
-
- sw := worker.From(w)
-
- go func() {
- assert.NoError(t, sw.Wait())
- }()
- defer func() {
- err = sw.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
-
- assert.Nil(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- assert.Equal(t, "hello", res.String())
-}
-
-func Test_BadPayload2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorker(cmd)
-
- sw := worker.From(w)
-
- go func() {
- assert.NoError(t, sw.Wait())
- }()
- defer func() {
- err := sw.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- res, err := sw.Exec(&payload.Payload{})
- assert.Error(t, err)
- assert.Nil(t, res)
-
- assert.Contains(t, err.Error(), "payload can not be empty")
-}
-
-func Test_String2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorker(cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- assert.Contains(t, w.String(), "php ../../tests/client.php echo pipes")
- assert.Contains(t, w.String(), "ready")
- assert.Contains(t, w.String(), "num_execs: 0")
-}
-
-func Test_Echo_Slow2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/slow-client.php", "echo", "pipes", "10", "10")
-
- w, _ := NewPipeFactory().SpawnWorker(cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
-
- assert.Nil(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- assert.Equal(t, "hello", res.String())
-}
-
-func Test_Broken2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "broken", "pipes")
-
- w, err := NewPipeFactory().SpawnWorker(cmd)
- if err != nil {
- t.Fatal(err)
- }
-
- sw := worker.From(w)
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.NotNil(t, err)
- assert.Nil(t, res)
-
- time.Sleep(time.Second * 3)
- assert.Error(t, w.Stop())
-}
-
-func Test_Error2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "error", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorker(cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
-
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.NotNil(t, err)
- assert.Nil(t, res)
-
- if errors.Is(errors.SoftJob, err) == false {
- t.Fatal("error should be of type errors.ErrSoftJob")
- }
- assert.Contains(t, err.Error(), "hello")
-}
-
-func Test_NumExecs2(t *testing.T) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorker(cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- _, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- if err != nil {
- t.Errorf("fail to execute payload: error %v", err)
- }
- assert.Equal(t, uint64(1), w.State().NumExecs())
-
- _, err = sw.Exec(&payload.Payload{Body: []byte("hello")})
- if err != nil {
- t.Errorf("fail to execute payload: error %v", err)
- }
- assert.Equal(t, uint64(2), w.State().NumExecs())
-
- _, err = sw.Exec(&payload.Payload{Body: []byte("hello")})
- if err != nil {
- t.Errorf("fail to execute payload: error %v", err)
- }
- assert.Equal(t, uint64(3), w.State().NumExecs())
-}
diff --git a/transport/pipe/pipe_factory_test.go b/transport/pipe/pipe_factory_test.go
deleted file mode 100755
index cbf1431a..00000000
--- a/transport/pipe/pipe_factory_test.go
+++ /dev/null
@@ -1,511 +0,0 @@
-package pipe
-
-import (
- "context"
- "os/exec"
- "sync"
- "testing"
- "time"
-
- "github.com/spiral/errors"
- "github.com/spiral/roadrunner/v2/payload"
- "github.com/spiral/roadrunner/v2/worker"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
-)
-
-func Test_GetState(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- go func() {
- assert.NoError(t, w.Wait())
- assert.Equal(t, worker.StateStopped, w.State().Value())
- }()
-
- assert.NoError(t, err)
- assert.NotNil(t, w)
-
- assert.Equal(t, worker.StateReady, w.State().Value())
- err = w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
-}
-
-func Test_Kill(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- wg := &sync.WaitGroup{}
- wg.Add(1)
- go func() {
- defer wg.Done()
- assert.Error(t, w.Wait())
- assert.Equal(t, worker.StateErrored, w.State().Value())
- }()
-
- assert.NoError(t, err)
- assert.NotNil(t, w)
-
- assert.Equal(t, worker.StateReady, w.State().Value())
- err = w.Kill()
- if err != nil {
- t.Errorf("error killing the Process: error %v", err)
- }
- wg.Wait()
-}
-
-func Test_Pipe_Start(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- assert.NoError(t, err)
- assert.NotNil(t, w)
-
- go func() {
- assert.NoError(t, w.Wait())
- }()
-
- assert.NoError(t, w.Stop())
-}
-
-func Test_Pipe_StartError(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- err := cmd.Start()
- if err != nil {
- t.Errorf("error running the command: error %v", err)
- }
-
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_PipeError(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- _, err := cmd.StdinPipe()
- if err != nil {
- t.Errorf("error creating the STDIN pipe: error %v", err)
- }
-
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_PipeError2(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- // error cause
- _, err := cmd.StdinPipe()
- if err != nil {
- t.Errorf("error creating the STDIN pipe: error %v", err)
- }
-
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_Failboot(t *testing.T) {
- cmd := exec.Command("php", "../../tests/failboot.php")
- ctx := context.Background()
-
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
-
- assert.Nil(t, w)
- assert.Error(t, err)
-}
-
-func Test_Pipe_Invalid(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("php", "../../tests/invalid.php")
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- assert.Error(t, err)
- assert.Nil(t, w)
-}
-
-func Test_Pipe_Echo(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- if err != nil {
- t.Fatal(err)
- }
-
- defer func() {
- err = w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.NoError(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- go func() {
- if w.Wait() != nil {
- t.Fail()
- }
- }()
-
- assert.Equal(t, "hello", res.String())
-}
-
-func Test_Pipe_Echo_Script(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("sh", "../../tests/pipes_test_script.sh")
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- err = w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.NoError(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- go func() {
- if w.Wait() != nil {
- t.Fail()
- }
- }()
-
- assert.Equal(t, "hello", res.String())
-}
-
-func Test_Pipe_Broken(t *testing.T) {
- t.Parallel()
- cmd := exec.Command("php", "../../tests/client.php", "broken", "pipes")
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- require.NoError(t, err)
- require.NotNil(t, w)
-
- go func() {
- errW := w.Wait()
- require.Error(t, errW)
- }()
-
- sw := worker.From(w)
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.Error(t, err)
- assert.Nil(t, res)
-
- time.Sleep(time.Second)
- err = w.Stop()
- assert.NoError(t, err)
-}
-
-func Benchmark_Pipe_SpawnWorker_Stop(b *testing.B) {
- f := NewPipeFactory()
- for n := 0; n < b.N; n++ {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- w, _ := f.SpawnWorkerWithTimeout(context.Background(), cmd)
- go func() {
- if w.Wait() != nil {
- b.Fail()
- }
- }()
-
- err := w.Stop()
- if err != nil {
- b.Errorf("error stopping the worker: error %v", err)
- }
- }
-}
-
-func Benchmark_Pipe_Worker_ExecEcho(b *testing.B) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorkerWithTimeout(context.Background(), cmd)
- sw := worker.From(w)
-
- b.ReportAllocs()
- b.ResetTimer()
- go func() {
- err := w.Wait()
- if err != nil {
- b.Errorf("error waiting the worker: error %v", err)
- }
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- b.Errorf("error stopping the worker: error %v", err)
- }
- }()
-
- for n := 0; n < b.N; n++ {
- if _, err := sw.Exec(&payload.Payload{Body: []byte("hello")}); err != nil {
- b.Fail()
- }
- }
-}
-
-func Benchmark_Pipe_Worker_ExecEcho3(b *testing.B) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- if err != nil {
- b.Fatal(err)
- }
-
- defer func() {
- err = w.Stop()
- if err != nil {
- b.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- for n := 0; n < b.N; n++ {
- if _, err := sw.Exec(&payload.Payload{Body: []byte("hello")}); err != nil {
- b.Fail()
- }
- }
-}
-
-func Benchmark_Pipe_Worker_ExecEchoWithoutContext(b *testing.B) {
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
- ctx := context.Background()
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- if err != nil {
- b.Fatal(err)
- }
-
- defer func() {
- err = w.Stop()
- if err != nil {
- b.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- for n := 0; n < b.N; n++ {
- if _, err := sw.Exec(&payload.Payload{Body: []byte("hello")}); err != nil {
- b.Fail()
- }
- }
-}
-
-func Test_Echo(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- if err != nil {
- t.Fatal(err)
- }
-
- sw := worker.From(w)
- go func() {
- assert.NoError(t, sw.Wait())
- }()
- defer func() {
- err = sw.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
-
- assert.Nil(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- assert.Equal(t, "hello", res.String())
-}
-
-func Test_BadPayload(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
-
- sw := worker.From(w)
-
- go func() {
- assert.NoError(t, sw.Wait())
- }()
- defer func() {
- err := sw.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- res, err := sw.Exec(&payload.Payload{})
-
- assert.Error(t, err)
- assert.Nil(t, res)
-
- assert.Contains(t, err.Error(), "payload can not be empty")
-}
-
-func Test_String(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- assert.Contains(t, w.String(), "php ../../tests/client.php echo pipes")
- assert.Contains(t, w.String(), "ready")
- assert.Contains(t, w.String(), "num_execs: 0")
-}
-
-func Test_Echo_Slow(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/slow-client.php", "echo", "pipes", "10", "10")
-
- w, _ := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
-
- assert.Nil(t, err)
- assert.NotNil(t, res)
- assert.NotNil(t, res.Body)
- assert.Empty(t, res.Context)
-
- assert.Equal(t, "hello", res.String())
-}
-
-func Test_Broken(t *testing.T) {
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "broken", "pipes")
-
- w, err := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- if err != nil {
- t.Fatal(err)
- }
-
- sw := worker.From(w)
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.NotNil(t, err)
- assert.Nil(t, res)
-
- time.Sleep(time.Second * 3)
- assert.Error(t, w.Stop())
-}
-
-func Test_Error(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "error", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
-
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- res, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- assert.NotNil(t, err)
- assert.Nil(t, res)
-
- if errors.Is(errors.SoftJob, err) == false {
- t.Fatal("error should be of type errors.ErrSoftJob")
- }
- assert.Contains(t, err.Error(), "hello")
-}
-
-func Test_NumExecs(t *testing.T) {
- t.Parallel()
- ctx := context.Background()
- cmd := exec.Command("php", "../../tests/client.php", "echo", "pipes")
-
- w, _ := NewPipeFactory().SpawnWorkerWithTimeout(ctx, cmd)
- go func() {
- assert.NoError(t, w.Wait())
- }()
- defer func() {
- err := w.Stop()
- if err != nil {
- t.Errorf("error stopping the Process: error %v", err)
- }
- }()
-
- sw := worker.From(w)
-
- _, err := sw.Exec(&payload.Payload{Body: []byte("hello")})
- if err != nil {
- t.Errorf("fail to execute payload: error %v", err)
- }
- assert.Equal(t, uint64(1), w.State().NumExecs())
-
- _, err = sw.Exec(&payload.Payload{Body: []byte("hello")})
- if err != nil {
- t.Errorf("fail to execute payload: error %v", err)
- }
- assert.Equal(t, uint64(2), w.State().NumExecs())
-
- _, err = sw.Exec(&payload.Payload{Body: []byte("hello")})
- if err != nil {
- t.Errorf("fail to execute payload: error %v", err)
- }
- assert.Equal(t, uint64(3), w.State().NumExecs())
-}