From 3ed9f510ff34fff50317d8aca6650b9b183f8d82 Mon Sep 17 00:00:00 2001 From: Samuel Pua Date: Wed, 28 Sep 2022 00:40:32 +0800 Subject: [PATCH] Fix(booking-job): Handle sql query builder when no running task --- .../ktmtrainbot/backgroundbookingjob.go | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/backend/internal/ktmtrainbot/backgroundbookingjob.go b/backend/internal/ktmtrainbot/backgroundbookingjob.go index 1faac2f..cad0890 100644 --- a/backend/internal/ktmtrainbot/backgroundbookingjob.go +++ b/backend/internal/ktmtrainbot/backgroundbookingjob.go @@ -54,11 +54,19 @@ func (env *Env) BackgroundJobRunner() { } var jobToDo Booking - err := tx.Model(&jobToDo). - Where("user_id NOT IN (?)", userIDArr). - Where("status = ?", "pending"). - Preload("User"). - First(&jobToDo).Error + var err error + if len(userIDArr) > 0 { + err = tx.Model(&jobToDo). + Where("user_id NOT IN (?)", userIDArr). + Where("status = ?", "pending"). + Preload("User"). + First(&jobToDo).Error + } else { + err = tx.Model(&jobToDo). + Where("status = ?", "pending"). + Preload("User"). + First(&jobToDo).Error + } // if no jobs pending found if err != nil {