fix select shift

lynxzp · Jan 4, 2022 · 2f36124 · 2f36124
1 parent 1c49006
commit 2f36124
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/collector/sources/pkg/shift/shift.go b/collector/sources/pkg/shift/shift.go
@@ -38,7 +38,7 @@ func GetCurrentShift() int {
 	//	return 3
 	//}
 	//return 0
-	t := time.Time{}
+	t := time.Now()
 	for i := range config.Cfg.Shifts {
 		if t.AfterOrEqual(config.Cfg.Shifts[i].Start) && t.BeforeOrEqual(config.Cfg.Shifts[i].Finish) {
 			return config.Cfg.Shifts[i].Number

diff --git a/collector/sources/pkg/time/time.go b/collector/sources/pkg/time/time.go
@@ -3,6 +3,7 @@ package time
 import (
 	"log"
 	"strconv"
+	systemT "time"
 )
 
 type Time struct {
@@ -49,3 +50,11 @@ func (t1 *Time) AfterOrEqual(t2 Time) bool {
 	}
 	return false
 }
+
+func Now() Time {
+	ownT := Time{}
+	sysT := systemT.Now()
+	ownT.hours = sysT.Hour()
+	ownT.minutes = sysT.Minute()
+	return ownT
+}