package stages

import (
	"gitea.anxinyun.cn/container/common_models"
	"log"
)

type StageManager struct {
	in     <-chan *common_models.ProcessData
	Stages []Stage
	out    <-chan *common_models.ProcessData
}

func NewStageManager() *StageManager {
	return &StageManager{}
}

func (the *StageManager) Run() {
	if len(the.Stages) == 0 {
		log.Panicf("Stages.len=%d 无有效处理流程", len(the.Stages))
	}
	for i := 0; i < len(the.Stages); i++ {
		if i == 0 {
			the.Stages[i].In = the.in
		} else {
			the.Stages[i].In = the.Stages[i-1].Out
		}
		the.Stages[i].StageRun()

	}
	the.out = the.Stages[len(the.Stages)-1].Out

	//todo 替换为sink
	go func() {
		for {
			a := <-the.out
			log.Printf("流程处理完毕===>%s", a.DeviceData.Name)
		}
	}()
}

func (the *StageManager) AddSource(source <-chan *common_models.ProcessData) {
	the.in = source
}

func (the *StageManager) AddOut(source chan *common_models.ProcessData) {
	the.out = source
}

func (the *StageManager) AddStages(stages ...Stage) {
	the.Stages = append(the.Stages, stages...)
}