Add SetLocation method to crontab

crontab.go

@@ -17,6 +17,7 @@ type Crontab struct {
 	ticker *time.Ticker
 	jobs   []*job
 	sync.RWMutex
+	location *time.Location
 }
 
 // job in cron table
@@ -62,6 +63,13 @@ func new(t time.Duration) *Crontab {
 	return c
 }
 
+// SetLocation to run cron on specific timezone.
+//
+// Returns error if provided location is not valid
+func (c *Crontab) SetLocation(loc *time.Location) {
+	c.location = loc
+}
+
 // AddJob to cron table
 //
 // Returns error if:
@@ -153,6 +161,10 @@ func (c *Crontab) RunAll() {
 
 // RunScheduled jobs
 func (c *Crontab) runScheduled(t time.Time) {
+	if c.location != nil {
+		t = t.In(c.location)
+	}
+
 	tick := getTick(t)
 	c.RLock()
 	defer c.RUnlock()

crontab_test.go

@@ -1,7 +1,9 @@
 package crontab
 
-import "testing"
-import "time"
+import (
+	"testing"
+	"time"
+)
 
 // TestSchedule parse the crontab syntax and compare number of target min/hour/days/month with expected ones
 func TestSchedule(t *testing.T) {