check reg time

isd-sgcu · Jul 22, 2024 · 7630af8 · 7630af8
1 parent b8e3a80
commit 7630af8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/internal/group/group.handler.go b/internal/group/group.handler.go
@@ -339,6 +339,9 @@ func (h *handlerImpl) checkRegTime() bool {
 	if nowGMTPlus7.Before(h.regConf.RpkmStart) {
 		h.log.Named("checkRegTime").Warn("Forbidden: Registration hasn't started")
 		return false
+	} else if nowGMTPlus7.After(h.regConf.RpkmEnd) {
+		h.log.Named("checkRegTime").Warn("Forbidden: Registration has ended")
+		return false
 	}
 
 	return true

diff --git a/internal/selection/selection.handler.go b/internal/selection/selection.handler.go
@@ -283,6 +283,9 @@ func (h *handlerImpl) checkRegTime() bool {
 	if nowGMTPlus7.Before(h.regConf.RpkmStart) {
 		h.log.Named("checkRegTime").Warn("Forbidden: Registration hasn't started")
 		return false
+	} else if nowGMTPlus7.After(h.regConf.RpkmEnd) {
+		h.log.Named("checkRegTime").Warn("Forbidden: Registration has ended")
+		return false
 	}
 
 	return true