From 1dbbc486ff9de1dc697884a08d071bb84224a76a Mon Sep 17 00:00:00 2001
From: Tom Morelly <tom.morelly@clearroute.io>
Date: Fri, 20 Sep 2024 09:02:22 +1000
Subject: [PATCH] feat(monitor): introduce MonitorWithStatus

---
 README.md   |  1 +
 executor.go | 10 ++++++++++
 monitor.go  |  8 ++++++++
 3 files changed, 19 insertions(+)

diff --git a/README.md b/README.md
index 4a1de758..b97ff29b 100644
--- a/README.md
+++ b/README.md
@@ -151,6 +151,7 @@ The provided NewLogger uses the standard library's log package.
 ### Metrics
 Metrics may be collected from the execution of each job.
 - [**Monitor**](https://pkg.go.dev/github.com/go-co-op/gocron/v2#Monitor):
+- [**MonitorWithStatus**](https://pkg.go.dev/github.com/go-co-op/gocron/v2#MonitorWithStatus) (includes the Status of a Job)
 A monitor can be used to collect metrics for each job from a scheduler.
   - Implementations: [go-co-op monitors](https://github.com/go-co-op?q=-monitor&type=all&language=&sort=)
     (don't see what you need? request on slack to get a repo created to contribute it!)
diff --git a/executor.go b/executor.go
index ae346b1e..3296122b 100644
--- a/executor.go
+++ b/executor.go
@@ -49,6 +49,8 @@ type executor struct {
 	locker Locker
 	// monitor for reporting metrics
 	monitor Monitor
+	// monitor for reporting metrics
+	MonitorWithStatus MonitorWithStatus
 }
 
 type jobIn struct {
@@ -401,9 +403,11 @@ func (e *executor) runJob(j internalJob, jIn jobIn) {
 	if err != nil {
 		_ = callJobFuncWithParams(j.afterJobRunsWithError, j.id, j.name, err)
 		e.incrementJobCounter(j, Fail)
+		e.recordJobTimingWithStatus(startTime, time.Now(), j, Fail)
 	} else {
 		_ = callJobFuncWithParams(j.afterJobRuns, j.id, j.name)
 		e.incrementJobCounter(j, Success)
+		e.recordJobTimingWithStatus(startTime, time.Now(), j, Success)
 	}
 }
 
@@ -426,6 +430,12 @@ func (e *executor) recordJobTiming(start time.Time, end time.Time, j internalJob
 	}
 }
 
+func (e *executor) recordJobTimingWithStatus(start time.Time, end time.Time, j internalJob, status JobStatus) {
+	if e.MonitorWithStatus != nil {
+		e.MonitorWithStatus.RecordJobTimingWithStatus(start, end, j.id, j.name, j.tags, status)
+	}
+}
+
 func (e *executor) incrementJobCounter(j internalJob, status JobStatus) {
 	if e.monitor != nil {
 		e.monitor.IncrementJob(j.id, j.name, j.tags, status)
diff --git a/monitor.go b/monitor.go
index d3c5bbd9..68e7ae19 100644
--- a/monitor.go
+++ b/monitor.go
@@ -26,3 +26,11 @@ type Monitor interface {
 	// to handle instantiating and recording the value
 	RecordJobTiming(startTime, endTime time.Time, id uuid.UUID, name string, tags []string)
 }
+
+// MonitorWithStatus extends RecordJobTiming with the job status.
+type MonitorWithStatus interface {
+	Monitor
+	// RecordJobTimingWithStatus will provide details about the job, its status and the timing and expects the underlying implementation
+	// to handle instantiating and recording the value
+	RecordJobTimingWithStatus(startTime, endTime time.Time, id uuid.UUID, name string, tags []string, status JobStatus)
+}