Skip to content

Commit 2a3672d

Browse files
committed
addressing sourcery-ai
1 parent 14f5b52 commit 2a3672d

File tree

3 files changed

+30
-2
lines changed

3 files changed

+30
-2
lines changed

gateway-ha/src/main/java/io/trino/gateway/ha/config/ClusterSchedulerConfiguration.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,13 @@
1818
import jakarta.annotation.Nullable;
1919
import jakarta.annotation.PostConstruct;
2020
import jakarta.annotation.PreDestroy;
21+
import org.slf4j.Logger;
22+
import org.slf4j.LoggerFactory;
2123

2224
public class ClusterSchedulerConfiguration
2325
{
26+
private static final Logger log = LoggerFactory.getLogger(ClusterSchedulerConfiguration.class);
27+
2428
private final ClusterScheduler scheduler;
2529

2630
@Inject
@@ -45,7 +49,7 @@ public void stop()
4549
scheduler.close();
4650
}
4751
catch (Exception e) {
48-
// Ignore
52+
log.error("Exception occurred while shutting down ClusterScheduler", e);
4953
}
5054
}
5155
}

gateway-ha/src/main/java/io/trino/gateway/ha/config/ScheduleConfiguration.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import com.fasterxml.jackson.annotation.JsonProperty;
1717
import io.airlift.units.Duration;
1818

19+
import java.util.ArrayList;
1920
import java.util.List;
2021

2122
import static java.util.Objects.requireNonNull;
@@ -25,7 +26,7 @@ public class ScheduleConfiguration
2526
private boolean enabled;
2627
private Duration checkInterval = new Duration(5, java.util.concurrent.TimeUnit.MINUTES);
2728
private String timezone = "GMT"; // Default to GMT if not specified
28-
private List<ClusterSchedule> schedules;
29+
private List<ClusterSchedule> schedules = new ArrayList<>();
2930

3031
@JsonProperty
3132
public boolean isEnabled()

gateway-ha/src/test/java/io/trino/gateway/ha/scheduler/TestClusterScheduler.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,6 +199,29 @@ void testClusterNotFoundInBackendManager()
199199
verify(backendManager, atLeastOnce()).getBackendByName(CLUSTER_NAME);
200200
}
201201

202+
@Test
203+
void testSchedulerDoesNotTriggerWhenDisabled()
204+
{
205+
// Setup test cluster with scheduling disabled
206+
when(scheduleConfig.isEnabled()).thenReturn(false);
207+
when(scheduleConfig.getTimezone()).thenReturn(TEST_TIMEZONE.toString());
208+
209+
// Initialize the scheduler with disabled configuration
210+
scheduler = new ClusterScheduler(backendManager, scheduleConfig);
211+
scheduler.start();
212+
213+
// Time within the cron schedule (9 AM - 5 PM)
214+
ZonedDateTime testTime = ZonedDateTime.of(2025, 9, 29, 10, 0, 0, 0, TEST_TIMEZONE);
215+
216+
// Execute
217+
scheduler.checkAndUpdateClusterStatus(testTime);
218+
219+
// Verify no actions are taken when scheduler is disabled
220+
verify(backendManager, never()).activateBackend(anyString());
221+
verify(backendManager, never()).deactivateBackend(anyString());
222+
verify(backendManager, never()).getBackendByName(anyString());
223+
}
224+
202225
@Test
203226
void testSchedulerWithDifferentTimezones()
204227
{

0 commit comments

Comments
 (0)