how to use simple_one_for_one under supervisor tree

[zile2000]

Hi:
I'm writing something which use supervisor:start_child to dynamic create worker process.
But my code show's me these processes share same struct which I just make for gen.Server behavior.
Maybe I should write some creater code with method to create the behavior struct , but I donnot know how to do this with supervisor.
Is there some demo code show me how to do this ?
Or I just can do this with spawn process which never under supervisor tree ?

[halturin]

Hello, thanks for the question. There was the same question a while ago #92 (comment)

I would recommend to use this kind of supervisor in this way

   control_process (based on gen.Server)
       |
       ---- spawn sup_process (based on gen.Supervisor with "one for one" strategy)
                     |
                     ------ worker_process (based on gen.Server)

so the code will be like:

// definition
var (
	numWorkers = 5
)

type control struct {
   gen.Server
}
type controlState struct {
   sup *supWorkers
   supProcess gen.Process
   workers    []etf.Pid
   counter    uint64
}

type supWorkers struct {
   gen.Supervisor
}

type worker struct {
  gen.Server
}

/// callbacks for the 'control' process
func (c *control) Init(process *gen.ServerProcess, args ...etf.Term) error { 

    sup := &supWorkers{}
    supOptions := gen.ProcessOptions{
        Context: process.Context(), // inherited context to make sup process be hard linked with the control one
    }
    supProcess, err := process.Spawn("", supOptions, sup)
    state := &controlState{
      sup: sup,
      supProcess: supProcess,
    }
     for i := 0; i < numWorkers; i++ { // start 5 worker processes
		child, err := sup.StartChild(supProcess, "worker")
		if err != nil {
			return err
		}
		state.workers = append(state.workers, child.Self())

               // monitor process in order to restart it on termination
		process.MonitorProcess(child.Self())
	}

   process.State = state
    return nil
}

func (c *control) HandleCast(process *gen.ServerProcess, message etf.Term) gen.ServerStatus {
	state := process.State.(*controlState)

        // use "round robin" to choose the worker process
	n := int(state.counter % uint64(numWorkers))
	state.counter++
        // forward message 
	process.Cast(state.workers[n], message)

	return gen.ServerStatusOK
}

func (c *control) HandleInfo(process *gen.ServerProcess, message etf.Term) gen.ServerStatus {
	state := process.State.(*controlState)
	switch m := message.(type) {
	case gen.MessageDown:
		for i, pid := range state.workers {
			if pid != m.Pid {
				continue
			}
			// restart terminated worker
			child, err := state.sup.StartChild(state.supProcess, "worker")
			if err != nil {
				return err
			}
			process.MonitorProcess(child.Self())
			state.workers[i] = child.Self()
			break
		}
	default:
		fmt.Printf("unknown request: %#v\n", m)
	}
	return gen.ServerStatusOK
}


// callback for the supervisor
func (s *supWorkers) Init(args ...etf.Term) (gen.SupervisorSpec, error) {
	return gen.SupervisorSpec{
		Name: "sup_workers",
		Children: []gen.SupervisorChildSpec{
			gen.SupervisorChildSpec{
				Name:  "worker",
				Child: &worker{},
			},
		},
		Strategy: gen.SupervisorStrategy{
			Type:      gen.SupervisorStrategySimpleOneForOne,
			Intensity: 5,
			Period:    5,
			Restart:   gen.SupervisorStrategyRestartTemporary,
		},
	}, nil
}

// callbacks for the 'worker' process
func (w *worker) Init(process *gen.ServerProcess, args ...etf.Term) error { 
    // do whatever you need for the initialization
    return nil
}

func (w *worker) HandleCast(process *gen.ServerProcess, message etf.Term) gen.ServerStatus {
        // handle cast message forwarded from the 'control' process
	return gen.ServerStatusOK
}

you may want to use SupervisorStrategyRestartPermanent as a restart strategy so you don't need to monitor child process and restart it on termination. supervisor will manage it by itself.

[halturin]

if you need to forward Call request you should:

• forward from value along with the message
• use gen.ServerStatusIgnore as a return value in the control's HandleCall callback
• in worker's HandleInfo (or HandleCast) use process.ReplyTo(from, replymessage)

type ForwardCall {
    from gen.ServerFrom
    message etf.Term
}

func (c *control) HandleCall(process *gen.ServerProcess, from gen.ServerFrom, message etf.Term) (etf.Term, gen.ServerStatus) {
	state := process.State.(*controlState)

        // use "round robin" to choose the worker process
	n := int(state.counter % uint64(numWorkers))
	state.counter++
        // forward message 
	process.Send(state.workers[n], ForwardCall{from,message})

	return nil, gen.ServerStatusIgnore
}
func (w *worker) HandleInfo(process *gen.ServerProcess, message etf.Term) gen.ServerStatus {
        // handle call message forwarded from the 'control' process
        switch m := message.(type) {
         case ForwardCall:
                // handle message
                reply := "got it"
                process.ReplyTo(m.from, reply)
        }
	return gen.ServerStatusOK
}

[zile2000]

Thank you .
From #92 I got it.
func (e *example) Init(process *gen.ServerProcess, args ...etf.Term) error { // State has interface{} type process.State = &private{i: 123} // this value is only available for this process e.shared = 567 // all spawned processes with the same object &example{...} have access to this value return nil }
I should save data to process.State .
This tastes really erlang 😂 .

[halturin]

This tastes really erlang

inspired by :)