func()

in pkg/monitor/sqsevent/rebalance-recommendation-event.go [49:88]

• 39 lines of code
• 5 McCabe index (conditional complexity)

func (m SQSMonitor) rebalanceRecommendationToInterruptionEvent(event *EventBridgeEvent, message *sqs.Message) (*monitor.InterruptionEvent, error) {
	rebalanceRecDetail := &RebalanceRecommendationDetail{}
	err := json.Unmarshal(event.Detail, rebalanceRecDetail)
	if err != nil {
		return nil, err
	}

	nodeInfo, err := m.getNodeInfo(rebalanceRecDetail.InstanceID)
	if err != nil {
		return nil, err
	}
	interruptionEvent := monitor.InterruptionEvent{
		EventID:              fmt.Sprintf("rebalance-recommendation-event-%x", event.ID),
		Kind:                 monitor.RebalanceRecommendationKind,
		Monitor:              SQSMonitorKind,
		AutoScalingGroupName: nodeInfo.AsgName,
		StartTime:            event.getTime(),
		NodeName:             nodeInfo.Name,
		IsManaged:            nodeInfo.IsManaged,
		InstanceID:           nodeInfo.InstanceID,
		ProviderID:           nodeInfo.ProviderID,
		InstanceType:         nodeInfo.InstanceType,
		Description:          fmt.Sprintf("Rebalance recommendation event received. Instance %s will be cordoned at %s \n", rebalanceRecDetail.InstanceID, event.getTime()),
	}
	interruptionEvent.PostDrainTask = func(interruptionEvent monitor.InterruptionEvent, n node.Node) error {
		errs := m.deleteMessages([]*sqs.Message{message})
		if errs != nil {
			return errs[0]
		}
		return nil
	}
	interruptionEvent.PreDrainTask = func(interruptionEvent monitor.InterruptionEvent, n node.Node) error {
		err := n.TaintRebalanceRecommendation(interruptionEvent.NodeName, interruptionEvent.EventID)
		if err != nil {
			log.Err(err).Msgf("Unable to taint node with taint %s:%s", node.RebalanceRecommendationTaint, interruptionEvent.EventID)
		}
		return nil
	}
	return &interruptionEvent, nil
}