catapult/internal/service/core/service.go

package core

import (
	"context"
	"fmt"
	"log"
	"sync"
	"time"

	"git.kundeng.us/phoenix/textsender-models/tx0/message/scheduling"
	"git.kundeng.us/phoenix/textsender-models/tx0/token"

	"git.kundeng.us/phoenix/catapult/internal/app"
	"git.kundeng.us/phoenix/catapult/internal/service"
)

type Service struct {
	wg     sync.WaitGroup
	ctx    context.Context
	cancel context.CancelFunc
	App    *app.App
	token  *token.Login
}

func NewService(application *app.App) *Service {
	ctx, cancel := context.WithCancel(context.Background())
	return &Service{
		ctx:    ctx,
		cancel: cancel,
		App:    application,
	}
}

func (s *Service) Start() {
	log.Println("Starting service...")

	// Start multiple background workers
	s.wg.Add(3)
	go s.worker1()
	go s.worker2()
	go s.healthChecker()
}

func (s *Service) worker1() {
	defer s.wg.Done()

	ticker := time.NewTicker(5 * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-s.ctx.Done():
			log.Println("Worker 1 shutting down...")
			return
		case <-ticker.C:
			// Do some work
			now := time.Now()
			log.Println("Worker 1: Processing...")
			if s.token == nil {
				catapultAuth := service.Auth{Application: s.App}
				if token, err := catapultAuth.GetToken(); err != nil {
					fmt.Println("Error:", err)
				} else {
					fmt.Println("Access token:", token.AccessToken)
					s.token = token
				}
			}

			queue := service.Queue{Application: s.App, Token: s.token}
			if item, exists, err := queue.GetQueue(); err != nil {
				fmt.Println("Error:", err)
			} else {
				if *exists {
					fmt.Println("Id:", item.Id)
					fmt.Println("Created:", item.Created)
					fmt.Println("Scheduled:", item.Scheduled)
					fmt.Println("Status:", item.Status)
					fmt.Println("User Id:", item.UserId)
					if scheduledMessageValid(item, now) {
						fmt.Println("Scheduled Message can be sent")
					} else {
						fmt.Println("Invalid scheduled message")
					}
				} else {
					fmt.Println("Empty queue")
				}
			}
		}
	}
}

func (s *Service) worker2() {
	defer s.wg.Done()

	ticker := time.NewTicker(10 * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-s.ctx.Done():
			log.Println("Worker 2 shutting down...")
			return
		case <-ticker.C:
			// Do some work
			log.Println("Worker 2: Processing...")
		}
	}
}

func (s *Service) healthChecker() {
	defer s.wg.Done()

	ticker := time.NewTicker(30 * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-s.ctx.Done():
			log.Println("Health checker shutting down...")
			return
		case <-ticker.C:
			log.Println("Health check: Service is healthy")
		}
	}
}

func (s *Service) Stop() {
	log.Println("Shutting down service...")
	s.cancel()
	s.wg.Wait()
	log.Println("Service stopped gracefully")
}

func scheduledMessageValid(schMsg *scheduling.ScheduledMessage, now time.Time) bool {
	if schMsg.Scheduled.Before(now) {
		return false
	} else {
		return true
	}
}