core

type Gang struct {
	Name       string
	WaitTime   time.Duration
	CreateTime time.Time

	// strict-mode or non-strict-mode
	Mode              string
	MinRequiredNumber int
	TotalChildrenNum  int
	GangGroupId       string
	GangGroup         []string
	Children          map[string]*v1.Pod
	// pods that have already assumed(waiting in Permit stage)
	WaitingForBindChildren map[string]*v1.Pod
	// pods that have already bound
	BoundChildren map[string]*v1.Pod
	// OnceResourceSatisfied indicates whether the gang has ever reached the ResourceSatisfied state，which means the
	// children number has reached the minNum in the early step,
	// once this variable is set true, it is irreversible.
	OnceResourceSatisfied bool

	// only-waiting, only consider waiting pods
	// waiting-and-running, consider waiting and running pods
	// waiting-running-succeed, consider waiting, running and succeed pods
	// once-satisfied, once gang is satisfied, no need to consider any status pods
	GangMatchPolicy string

	// if the podGroup should be passed at PreFilter stage(Strict-Mode)
	ScheduleCycleValid bool
	// these fields used to count the cycle
	// For example, at the beginning, `scheduleCycle` is 1, and each pod's cycle in `childrenScheduleRoundMap` is 0. When each pod comes to PreFilter,
	// we will check if the pod's value in `childrenScheduleRoundMap` is smaller than Gang's `scheduleCycle`, If result is positive,
	// we set the pod's cycle in `childrenScheduleRoundMap` equal with `scheduleCycle` and pass the check. If result is negative, means
	// the pod has been scheduled in this cycle, so we should reject it. With `totalChildrenNum`'s help, when the last pod comes to make all
	// `childrenScheduleRoundMap`'s values equal to `scheduleCycle`, Gang's `scheduleCycle` will be added by 1, which means a new schedule cycle.
	ScheduleCycle            int
	ChildrenScheduleRoundMap map[string]int

	GangFrom    string
	HasGangInit bool
	// contains filtered or unexported fields
}

type GangCache struct {
	// contains filtered or unexported fields
}

type GangSummary struct {
	Name                     string         `json:"name"`
	WaitTime                 time.Duration  `json:"waitTime"`
	CreateTime               time.Time      `json:"createTime"`
	Mode                     string         `json:"mode"`
	GangMatchPolicy          string         `json:"gangMatchPolicy"`
	MinRequiredNumber        int            `json:"minRequiredNumber"`
	TotalChildrenNum         int            `json:"totalChildrenNum"`
	GangGroup                []string       `json:"gangGroup"`
	Children                 sets.String    `json:"children"`
	WaitingForBindChildren   sets.String    `json:"waitingForBindChildren"`
	BoundChildren            sets.String    `json:"boundChildren"`
	OnceResourceSatisfied    bool           `json:"onceResourceSatisfied"`
	ScheduleCycleValid       bool           `json:"scheduleCycleValid"`
	ScheduleCycle            int            `json:"scheduleCycle"`
	ChildrenScheduleRoundMap map[string]int `json:"childrenScheduleRoundMap"`
	GangFrom                 string         `json:"gangFrom"`
	HasGangInit              bool           `json:"hasGangInit"`
}

type Manager interface {
	PreFilter(context.Context, *corev1.Pod) error
	Permit(context.Context, *corev1.Pod) (time.Duration, Status)
	PostBind(context.Context, *corev1.Pod, string)
	PostFilter(context.Context, *corev1.Pod, framework.Handle, string, framework.NodeToStatusMap) (*framework.PostFilterResult, *framework.Status)
	GetCreatTime(*framework.QueuedPodInfo) time.Time
	GetGroupId(*corev1.Pod) (string, error)
	GetAllPodsFromGang(string) []*corev1.Pod
	ActivateSiblings(*corev1.Pod, *framework.CycleState)
	AllowGangGroup(*corev1.Pod, framework.Handle, string)
	Unreserve(context.Context, *framework.CycleState, *corev1.Pod, string, framework.Handle, string)
	GetGangSummary(gangId string) (*GangSummary, bool)
	GetGangSummaries() map[string]*GangSummary
	IsGangMinSatisfied(*corev1.Pod) bool
	GetChildScheduleCycle(*corev1.Pod) int
}

type PodGroupManager struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

type Status string