Day 88/100 commits - add event handler for IaC Apply scheduled

internal/eventbus/events/apply_iac.go

@@ -8,14 +8,14 @@ import (
 
 const IAC_APPLY_SCHEDULED EventName = "iac_apply_scheduled"
 
-type IacApplied struct {
+type IacApplyScheduled struct {
 	ChangeId  uuid.UUID
 	ProjectId uuid.UUID
 	PlanId    uuid.UUID
 	IacType   aggregates.IaCPlanType
 	Owner     string
 }
 
-func (i IacApplied) MarshalBinary() ([]byte, error) {
+func (i IacApplyScheduled) MarshalBinary() ([]byte, error) {
 	return json.Marshal(i)
 }

internal/eventbus/events/scheduled_plan.go

@@ -6,7 +6,7 @@ import (
 	"time"
 )
 
-const SCHEDULED_PLAN = "scheduled_plan"
+const SCHEDULED_PLAN EventName = "scheduled_plan"
 
 type ScheduledPlan struct {
 	ProjectId uuid.UUID

internal/handlers/event_handler_factory.go

@@ -25,7 +25,7 @@ func NewEventHandlerFactory(eventSubscriber eb.EventSubscriber, eventPublisher e
 		eventSubscriber: eventSubscriber,
 		unitOfWork:      unitOfWork,
 		eventPublisher:  eventPublisher,
-		allowedEvents:   []events.EventName{events.LEASE_LOCK, events.TRIGGERED_PLAN, events.SCHEDULED_PLAN},
+		allowedEvents:   []events.EventName{events.LEASE_LOCK, events.TRIGGERED_PLAN, events.SCHEDULED_PLAN, events.IAC_APPLY_SCHEDULED},
 	}
 }
 
@@ -43,6 +43,9 @@ func (factory *EventHandlerFactory) RegisterHandler(event events.EventName) (Eve
 	case events.SCHEDULED_PLAN:
 		factory.allowedEvents = helpers.Remove(factory.allowedEvents, event)
 		return newScheduledPlanHandler(factory.eventSubscriber, factory.unitOfWork, factory.eventPublisher)
+	case events.IAC_APPLY_SCHEDULED:
+		factory.allowedEvents = helpers.Remove(factory.allowedEvents, event)
+		return newScheduledIaCApplyHandler(factory.eventSubscriber, factory.unitOfWork)
 	}
 	return nil, MissingHandlerImplementedFactory
 }

internal/handlers/scheduled_iac_apply_handler.go

@@ -0,0 +1,54 @@
+package handlers
+
+import (
+	"encoding/json"
+	"fmt"
+	"golang.org/x/net/context"
+	eb "labraboard/internal/eventbus"
+	"labraboard/internal/eventbus/events"
+	"labraboard/internal/logger"
+	"labraboard/internal/repositories"
+)
+
+type scheduledIaCApplyHandler struct {
+	eventSubscriber eb.EventSubscriber
+	unitOfWork      *repositories.UnitOfWork
+}
+
+func newScheduledIaCApplyHandler(eventSubscriber eb.EventSubscriber, unitOfWork *repositories.UnitOfWork) (*scheduledIaCApplyHandler, error) {
+	return &scheduledIaCApplyHandler{
+		eventSubscriber,
+		unitOfWork,
+	}, nil
+}
+
+func (handler *scheduledIaCApplyHandler) Handle(ctx context.Context) {
+	log := logger.GetWitContext(ctx).With().Str("event", string(events.IAC_APPLY_SCHEDULED)).Logger()
+	locks := handler.eventSubscriber.Subscribe(events.SCHEDULED_PLAN, log.WithContext(ctx))
+	for msg := range locks {
+		var event = events.IacApplyScheduled{}
+		err := json.Unmarshal(msg, &event)
+		if err != nil {
+			log.Error().Err(fmt.Errorf("cannot handle message type %T", event))
+		}
+		log.Info().Msgf("Received message: %s", msg)
+		go handler.handle(event, log.WithContext(ctx))
+	}
+}
+
+func (handler *scheduledIaCApplyHandler) handle(event events.IacApplyScheduled, ctx context.Context) {
+	log := logger.GetWitContext(ctx).
+		With().
+		Str("changeId", event.ChangeId.String()).
+		Str("iacType", string(event.IacType)).
+		Str("planId", event.PlanId.String()).
+		Str("projectId", event.ProjectId.String()).
+		Str("owner", event.Owner).
+		Logger()
+	_, err := handler.unitOfWork.IacPlan.Get(event.PlanId, log.WithContext(ctx))
+	if err != nil {
+		log.Error().Err(err)
+		return
+	}
+
+}

internal/handlers/scheduled_plan_handler.go

@@ -24,9 +24,10 @@ func newScheduledPlanHandler(eventSubscriber eb.EventSubscriber, unitOfWork *rep
 		publisher,
 	}, nil
 }
+
 func (handler *scheduledPlanHandler) Handle(ctx context.Context) {
-	log := logger.GetWitContext(ctx).With().Str("event", events.SCHEDULED_PLAN).Logger()
-	locks := handler.eventSubscriber.Subscribe(events.SCHEDULED_PLAN, ctx)
+	log := logger.GetWitContext(ctx).With().Str("event", string(events.SCHEDULED_PLAN)).Logger()
+	locks := handler.eventSubscriber.Subscribe(events.SCHEDULED_PLAN, log.WithContext(ctx))
 	for msg := range locks {
 		var event = events.ScheduledPlan{}
 		err := json.Unmarshal(msg, &event)

internal/services/iacService.go

@@ -283,7 +283,7 @@ func (svc *IacService) ScheduleApply(projectId uuid.UUID, planId uuid.UUID, ctx
 	}
 
 	var changeId = uuid.New()
-	var event = &events.IacApplied{
+	var event = &events.IacApplyScheduled{
 		ChangeId:  changeId,
 		ProjectId: projectId,
 		PlanId:    planId,