- implement worker debug mode

1 files changed, 54 insertions, 36 deletions

diff --git a/static_pool.go b/static_pool.go
index 2905742b..6f247d9e 100755
--- a/static_pool.go
+++ b/static_pool.go
@@ -43,8 +43,9 @@ type StaticPool struct {
 func NewPool(ctx context.Context, cmd func() *exec.Cmd, factory Factory, cfg Config) (Pool, error) {
 	cfg.InitDefaults()
 
-	if cfg.HeavyLoad {
+	if cfg.Debug {
 		cfg.NumWorkers = 0
+		cfg.MaxJobs = 1
 	}
 
 	p := &StaticPool{
@@ -86,26 +87,30 @@ func NewPool(ctx context.Context, cmd func() *exec.Cmd, factory Factory, cfg Con
 }
 
 // AddListener connects event listener to the pool.
-func (p *StaticPool) AddListener(listener util.EventListener) {
-	p.events.AddListener(listener)
+func (sp *StaticPool) AddListener(listener util.EventListener) {
+	sp.events.AddListener(listener)
 }
 
 // Config returns associated pool configuration. Immutable.
-func (p *StaticPool) GetConfig() Config {
-	return p.cfg
+func (sp *StaticPool) GetConfig() Config {
+	return sp.cfg
 }
 
 // Workers returns worker list associated with the pool.
-func (p *StaticPool) Workers() (workers []WorkerBase) {
-	return p.ww.WorkersList()
+func (sp *StaticPool) Workers() (workers []WorkerBase) {
+	return sp.ww.WorkersList()
 }
 
-func (p *StaticPool) RemoveWorker(ctx context.Context, wb WorkerBase) error {
-	return p.ww.RemoveWorker(ctx, wb)
+func (sp *StaticPool) RemoveWorker(ctx context.Context, wb WorkerBase) error {
+	return sp.ww.RemoveWorker(ctx, wb)
 }
 
-func (p *StaticPool) Exec(rqs Payload) (Payload, error) {
-	w, err := p.ww.GetFreeWorker(context.Background())
+func (sp *StaticPool) Exec(p Payload) (Payload, error) {
+	if sp.cfg.Debug {
+		return sp.execDebug(p)
+	}
+
+	w, err := sp.ww.GetFreeWorker(context.Background())
 	if err != nil && errors.Is(err, ErrWatcherStopped) {
 		return EmptyPayload, ErrWatcherStopped
 	} else if err != nil {
@@ -114,30 +119,30 @@ func (p *StaticPool) Exec(rqs Payload) (Payload, error) {
 
 	sw := w.(SyncWorker)
 
-	rsp, err := sw.Exec(rqs)
+	rsp, err := sw.Exec(p)
 	if err != nil {
 		// soft job errors are allowed
-		if _, jobError := err.(JobError); jobError {
-			if p.cfg.MaxJobs != 0 && w.State().NumExecs() >= p.cfg.MaxJobs {
-				err := p.ww.AllocateNew(bCtx)
+		if _, jobError := err.(ExecError); jobError {
+			if sp.cfg.MaxJobs != 0 && w.State().NumExecs() >= sp.cfg.MaxJobs {
+				err := sp.ww.AllocateNew(bCtx)
 				if err != nil {
-					p.events.Push(PoolEvent{Event: EventPoolError, Payload: err})
+					sp.events.Push(PoolEvent{Event: EventPoolError, Payload: err})
 				}
 
 				w.State().Set(StateInvalid)
 				err = w.Stop(bCtx)
 				if err != nil {
-					p.events.Push(WorkerEvent{Event: EventWorkerError, Worker: w, Payload: err})
+					sp.events.Push(WorkerEvent{Event: EventWorkerError, Worker: w, Payload: err})
 				}
 			} else {
-				p.ww.PushWorker(w)
+				sp.ww.PushWorker(w)
 			}
 
 			return EmptyPayload, err
 		}
 
 		sw.State().Set(StateInvalid)
-		p.events.Push(PoolEvent{Event: EventWorkerDestruct, Payload: w})
+		sp.events.Push(PoolEvent{Event: EventWorkerDestruct, Payload: w})
 		errS := w.Stop(bCtx)
 
 		if errS != nil {
@@ -152,25 +157,38 @@ func (p *StaticPool) Exec(rqs Payload) (Payload, error) {
 		w.State().Set(StateInvalid)
 		err = w.Stop(bCtx)
 		if err != nil {
-			p.events.Push(WorkerEvent{Event: EventWorkerError, Worker: w, Payload: err})
+			sp.events.Push(WorkerEvent{Event: EventWorkerError, Worker: w, Payload: err})
 		}
 
-		return p.Exec(rqs)
+		return sp.Exec(p)
 	}
 
-	if p.cfg.MaxJobs != 0 && w.State().NumExecs() >= p.cfg.MaxJobs {
-		err = p.ww.AllocateNew(bCtx)
+	if sp.cfg.MaxJobs != 0 && w.State().NumExecs() >= sp.cfg.MaxJobs {
+		err = sp.ww.AllocateNew(bCtx)
 		if err != nil {
 			return EmptyPayload, err
 		}
 	} else {
-		p.muw.Lock()
-		p.ww.PushWorker(w)
-		p.muw.Unlock()
+		sp.ww.PushWorker(w)
 	}
 	return rsp, nil
 }
 
+func (sp *StaticPool) execDebug(p Payload) (Payload, error) {
+	sw, err := sp.ww.allocator()
+	if err != nil {
+		return EmptyPayload, err
+	}
+
+	r, err := sw.(SyncWorker).Exec(p)
+
+	if stopErr := sw.Stop(context.Background()); stopErr != nil {
+		sp.events.Push(WorkerEvent{Event: EventWorkerError, Worker: sw, Payload: err})
+	}
+
+	return r, err
+}
+
 // Exec one task with given payload and context, returns result or error.
 // func (p *StaticPool) ExecWithContext(ctx context.Context, rqs Payload) (Payload, error) {
 //	// todo: why TODO passed here?
@@ -204,7 +222,7 @@ func (p *StaticPool) Exec(rqs Payload) (Payload, error) {
 //		}
 //
 //		// soft job errors are allowed
-//		if _, jobError := err.(JobError); jobError {
+//		if _, jobError := err.(ExecError); jobError {
 //			p.ww.PushWorker(w)
 //			return EmptyPayload, err
 //		}
@@ -243,18 +261,18 @@ func (p *StaticPool) Exec(rqs Payload) (Payload, error) {
 // }
 
 // Destroy all underlying stack (but let them to complete the task).
-func (p *StaticPool) Destroy(ctx context.Context) {
-	p.ww.Destroy(ctx)
+func (sp *StaticPool) Destroy(ctx context.Context) {
+	sp.ww.Destroy(ctx)
 }
 
 // allocate required number of stack
-func (p *StaticPool) allocateWorkers(ctx context.Context, numWorkers int64) ([]WorkerBase, error) {
+func (sp *StaticPool) allocateWorkers(ctx context.Context, numWorkers int64) ([]WorkerBase, error) {
 	var workers []WorkerBase
 
 	// constant number of stack simplify logic
 	for i := int64(0); i < numWorkers; i++ {
-		ctx, cancel := context.WithTimeout(ctx, p.cfg.AllocateTimeout)
-		w, err := p.factory.SpawnWorkerWithContext(ctx, p.cmd())
+		ctx, cancel := context.WithTimeout(ctx, sp.cfg.AllocateTimeout)
+		w, err := sp.factory.SpawnWorkerWithContext(ctx, sp.cmd())
 		if err != nil {
 			cancel()
 			return nil, err
@@ -265,11 +283,11 @@ func (p *StaticPool) allocateWorkers(ctx context.Context, numWorkers int64) ([]W
 	return workers, nil
 }
 
-func (p *StaticPool) checkMaxJobs(ctx context.Context, w WorkerBase) error {
-	if p.cfg.MaxJobs != 0 && w.State().NumExecs() >= p.cfg.MaxJobs {
-		err := p.ww.AllocateNew(ctx)
+func (sp *StaticPool) checkMaxJobs(ctx context.Context, w WorkerBase) error {
+	if sp.cfg.MaxJobs != 0 && w.State().NumExecs() >= sp.cfg.MaxJobs {
+		err := sp.ww.AllocateNew(ctx)
 		if err != nil {
-			p.events.Push(PoolEvent{Event: EventPoolError, Payload: err})
+			sp.events.Push(PoolEvent{Event: EventPoolError, Payload: err})
 			return err
 		}
 	}