Subversion Repositories SmartDukaan

package com.smartdukaan.cron.monitored;

import java.util.Date;

import java.util.Map;

import java.util.concurrent.ConcurrentHashMap;

import java.util.concurrent.TimeUnit;

import io.micrometer.core.instrument.Counter;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.Around;

import org.aspectj.lang.annotation.Aspect;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import io.micrometer.core.instrument.MeterRegistry;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * Aspect to monitor execution of cron jobs via metrics and logging.

 */

@Aspect

@Component

public class CronJobMonitorAspect {

    private static final Logger log = LoggerFactory.getLogger(CronJobMonitorAspect.class);

    @Autowired

    private MeterRegistry meterRegistry;

    private final Map<String, Counter> failureCounters = new ConcurrentHashMap<>();

    /**

     * Intercept any method annotated with @Scheduled (i.e. cron jobs).

     */

    @Around("@annotation(org.springframework.scheduling.annotation.Scheduled)")

    public Object monitorCronJob(ProceedingJoinPoint joinPoint) throws Throwable {

        String methodName = joinPoint.getSignature().toShortString();

        log.info("Cron job {} started", methodName);

        System.out.println("Cron job " + methodName + " started");

        long startTime = System.currentTimeMillis();

        try {

            // Proceed with the scheduled method

            Object result = joinPoint.proceed();

            long elapsed = System.currentTimeMillis() - startTime;

            log.info("Cron job {} completed successfully in {} ms", methodName, elapsed);

            System.out.println("Cron job " + methodName + " completed successfully in " + elapsed + " ms");

            try {

                meterRegistry.counter("custom.cron.success.count", "method", methodName, "status", "success")

                        .increment();

                meterRegistry.timer("custom.cron.execution.time", "method", methodName, "status", "success")

                        .record(elapsed, TimeUnit.MILLISECONDS);

            } catch (Exception e) {

                log.warn("Failed to update success metrics for {}: {}", methodName, e.toString());

            }

            return result;

        } catch (Throwable ex) {

            long elapsed = System.currentTimeMillis() - startTime;

            log.error("Cron job {} failed in {} ms: {}", methodName, elapsed, ex.toString());

            System.out.println("Cron job " + methodName + " failed in " + elapsed + " ms");

            try {

                meterRegistry.counter("custom.cron.failure.count", "method", methodName, "status", "failure")

                        .increment();

                meterRegistry.timer("custom.cron.execution.time", "method", methodName, "status", "failure")

                        .record(elapsed, TimeUnit.MILLISECONDS);

            } catch (Exception e) {

                log.warn("Failed to update failure metrics for {}: {}", methodName, e.toString());

            }

            getFailureCounter(methodName).increment();

            throw ex;

        }

    }

    private Counter getFailureCounter(String methodName) {

        return failureCounters.computeIfAbsent(methodName, name ->

                Counter.builder("cron_execution_failure_count_total")

                        .description("Number of failed executions of scheduled method")

                        .tag("method", name)

                        .register(meterRegistry)

        );

    }

}