|
1 | 1 | package io.redlink.more.studymanager.utils; |
2 | 2 |
|
3 | | -import io.redlink.more.studymanager.model.scheduler.Event; |
4 | | -import io.redlink.more.studymanager.model.scheduler.RecurrenceRule; |
5 | | -import io.redlink.more.studymanager.model.scheduler.RelativeDate; |
6 | | -import io.redlink.more.studymanager.model.scheduler.RelativeEvent; |
7 | | -import io.redlink.more.studymanager.model.scheduler.RelativeRecurrenceRule; |
8 | | -import io.redlink.more.studymanager.model.scheduler.ScheduleEvent; |
9 | 3 | import org.apache.commons.lang3.Range; |
10 | 4 |
|
11 | 5 | import java.nio.ByteBuffer; |
12 | 6 | import java.nio.charset.StandardCharsets; |
13 | 7 | import java.security.MessageDigest; |
14 | 8 | import java.time.Instant; |
15 | | -import java.util.ArrayList; |
16 | 9 | import java.util.Collections; |
17 | | -import java.util.Comparator; |
18 | 10 | import java.util.List; |
19 | 11 | import java.util.Objects; |
20 | 12 | import java.util.stream.IntStream; |
21 | 13 |
|
22 | 14 | public class RandomSchedulerUtils { |
23 | 15 | public static final String OBSERVATION_SCHEDULE_SEED_KEY = "observation_schedule_seed"; |
24 | 16 |
|
25 | | - public static Long generateSeedFromSchedule(ScheduleEvent schedule) { |
26 | | - return generateSeedFromSchedule(schedule, null); |
27 | | - } |
28 | | - |
29 | | - public static Long generateSeedFromSchedule(ScheduleEvent schedule, String userId) { |
30 | | - if (schedule == null || schedule.getRandomization() == null || !schedule.getRandomization().state()) { |
31 | | - return null; |
32 | | - } |
33 | | - |
34 | | - if (Objects.equals(schedule.getType(), Event.TYPE)) { |
35 | | - return generateSeedForEvent((Event) schedule, userId); |
36 | | - } else if (Objects.equals(schedule.getType(), RelativeEvent.TYPE)) { |
37 | | - return generateSeedForRelativeEvent((RelativeEvent) schedule, userId); |
38 | | - } |
39 | | - |
40 | | - return null; |
41 | | - } |
42 | | - |
43 | | - private static Long generateSeedForEvent(Event event, String userId) { |
44 | | - Long windowStart = event.getDateStart() != null ? event.getDateStart().getEpochSecond() : null; |
45 | | - Long windowEnd = event.getDateEnd() != null ? event.getDateEnd().getEpochSecond() : null; |
46 | | - Integer duration = event.getRandomization() != null ? event.getRandomization().duration() : null; |
47 | | - |
48 | | - RecurrenceRule r = event.getRRule(); |
49 | | - List<String> byDay = null; |
50 | | - if (r != null && r.getByDay() != null) { |
51 | | - byDay = new ArrayList<>(r.getByDay()); |
52 | | - byDay.sort(Comparator.naturalOrder()); |
53 | | - } |
54 | | - |
55 | | - // Only include userId if provided; tests for generateSeedFromSchedule(event) expect a seed |
56 | | - // derived purely from the schedule definition. |
57 | | - if (userId != null) { |
58 | | - return stableHash64( |
59 | | - userId, |
60 | | - event.getType(), |
61 | | - windowStart, |
62 | | - windowEnd, |
63 | | - duration, |
64 | | - r != null ? r.getFreq() : null, |
65 | | - r != null && r.getUntil() != null ? r.getUntil().getEpochSecond() : null, |
66 | | - r != null ? r.getCount() : null, |
67 | | - r != null ? r.getInterval() : null, |
68 | | - byDay, |
69 | | - r != null ? r.getByMonth() : null, |
70 | | - r != null ? r.getByMonthDay() : null, |
71 | | - r != null ? r.getBySetPos() : null |
72 | | - ); |
73 | | - } |
74 | | - |
75 | | - return stableHash64( |
76 | | - event.getType(), |
77 | | - windowStart, |
78 | | - windowEnd, |
79 | | - duration, |
80 | | - r != null ? r.getFreq() : null, |
81 | | - r != null && r.getUntil() != null ? r.getUntil().getEpochSecond() : null, |
82 | | - r != null ? r.getCount() : null, |
83 | | - r != null ? r.getInterval() : null, |
84 | | - byDay, |
85 | | - r != null ? r.getByMonth() : null, |
86 | | - r != null ? r.getByMonthDay() : null, |
87 | | - r != null ? r.getBySetPos() : null |
88 | | - ); |
89 | | - } |
90 | | - |
91 | | - private static long generateSeedForRelativeEvent(RelativeEvent event, String userId) { |
92 | | - RelativeDate start = event.getDtstart(); |
93 | | - RelativeDate end = event.getDtend(); |
94 | | - Integer duration = event.getRandomization() != null ? event.getRandomization().duration() : null; |
95 | | - |
96 | | - RelativeRecurrenceRule rr = event.getRrrule(); |
97 | | - |
98 | | - Integer startHour = start != null && start.getTime() != null ? start.getHours() : null; |
99 | | - Integer startMinute = start != null && start.getTime() != null ? start.getMinutes() : null; |
100 | | - Integer endHour = end != null && end.getTime() != null ? end.getHours() : null; |
101 | | - Integer endMinute = end != null && end.getTime() != null ? end.getMinutes() : null; |
102 | | - |
103 | | - // Represent Duration deterministically via (value, unit) |
104 | | - Long startOffsetValue = start != null && start.getOffset() != null ? (long) start.getOffset().getValue() : null; |
105 | | - String startOffsetUnit = start != null && start.getOffset() != null && start.getOffset().getUnit() != null ? start.getOffset().getUnit().name() : null; |
106 | | - |
107 | | - Long endOffsetValue = end != null && end.getOffset() != null ? (long) end.getOffset().getValue() : null; |
108 | | - String endOffsetUnit = end != null && end.getOffset() != null && end.getOffset().getUnit() != null ? end.getOffset().getUnit().name() : null; |
109 | | - |
110 | | - Long freqValue = rr != null && rr.getFrequency() != null ? (long) rr.getFrequency().getValue() : null; |
111 | | - String freqUnit = rr != null && rr.getFrequency() != null && rr.getFrequency().getUnit() != null ? rr.getFrequency().getUnit().name() : null; |
112 | | - |
113 | | - Long endAfterValue = rr != null && rr.getEndAfter() != null ? (long) rr.getEndAfter().getValue() : null; |
114 | | - String endAfterUnit = rr != null && rr.getEndAfter() != null && rr.getEndAfter().getUnit() != null ? rr.getEndAfter().getUnit().name() : null; |
115 | | - |
116 | | - if (userId != null) { |
117 | | - return stableHash64( |
118 | | - userId, |
119 | | - event.getType(), |
120 | | - startHour, |
121 | | - startMinute, |
122 | | - endHour, |
123 | | - endMinute, |
124 | | - startOffsetValue, |
125 | | - startOffsetUnit, |
126 | | - endOffsetValue, |
127 | | - endOffsetUnit, |
128 | | - freqValue, |
129 | | - freqUnit, |
130 | | - endAfterValue, |
131 | | - endAfterUnit, |
132 | | - duration |
133 | | - ); |
134 | | - } |
135 | | - |
136 | | - return stableHash64( |
137 | | - event.getType(), |
138 | | - startHour, |
139 | | - startMinute, |
140 | | - endHour, |
141 | | - endMinute, |
142 | | - startOffsetValue, |
143 | | - startOffsetUnit, |
144 | | - endOffsetValue, |
145 | | - endOffsetUnit, |
146 | | - freqValue, |
147 | | - freqUnit, |
148 | | - endAfterValue, |
149 | | - endAfterUnit, |
150 | | - duration |
151 | | - ); |
152 | | - } |
153 | 17 |
|
154 | 18 | /** |
155 | 19 | * Stable 64-bit hash derived from SHA-256. Deterministic across JVMs/platforms. |
|
0 commit comments