@@ -22,13 +22,11 @@ import io.getstream.chat.android.client.api2.model.dto.AIIndicatorClearEventDto
2222import io.getstream.chat.android.client.api2.model.dto.AIIndicatorStopEventDto
2323import io.getstream.chat.android.client.api2.model.dto.AIIndicatorUpdatedEventDto
2424import io.getstream.chat.android.client.api2.model.dto.ChatEventDto
25- import io.getstream.chat.android.client.api2.model.dto.ConnectedEventDto
2625import io.getstream.chat.android.client.api2.model.dto.ConnectingEventDto
2726import io.getstream.chat.android.client.api2.model.dto.ConnectionErrorEventDto
2827import io.getstream.chat.android.client.api2.model.dto.DisconnectedEventDto
2928import io.getstream.chat.android.client.api2.model.dto.ErrorEventDto
3029import io.getstream.chat.android.client.api2.model.dto.GeneratedEventDto
31- import io.getstream.chat.android.client.api2.model.dto.HealthEventDto
3230import io.getstream.chat.android.client.api2.model.dto.NotificationAddedToChannelEventDto
3331import io.getstream.chat.android.client.api2.model.dto.UnknownEventDto
3432import io.getstream.chat.android.client.events.AIIndicatorClearEvent
@@ -113,8 +111,10 @@ import io.getstream.chat.android.network.models.ChannelHiddenEvent as GeneratedC
113111import io.getstream.chat.android.network.models.ChannelTruncatedEvent as GeneratedChannelTruncatedEvent
114112import io.getstream.chat.android.network.models.ChannelUpdatedEvent as GeneratedChannelUpdatedEvent
115113import io.getstream.chat.android.network.models.ChannelVisibleEvent as GeneratedChannelVisibleEvent
114+ import io.getstream.chat.android.network.models.ConnectedEvent as GeneratedConnectedEvent
116115import io.getstream.chat.android.network.models.DraftDeletedEvent as GeneratedDraftDeletedEvent
117116import io.getstream.chat.android.network.models.DraftUpdatedEvent as GeneratedDraftUpdatedEvent
117+ import io.getstream.chat.android.network.models.HealthCheckEvent as GeneratedHealthCheckEvent
118118import io.getstream.chat.android.network.models.MemberAddedEvent as GeneratedMemberAddedEvent
119119import io.getstream.chat.android.network.models.MemberRemovedEvent as GeneratedMemberRemovedEvent
120120import io.getstream.chat.android.network.models.MemberUpdatedEvent as GeneratedMemberUpdatedEvent
@@ -174,12 +174,10 @@ internal class EventMapping(
174174 @Suppress(" LongMethod" )
175175 internal fun ChatEventDto.toDomain (): ChatEvent {
176176 return when (this ) {
177- is ConnectedEventDto -> toDomain()
178177 is ConnectionErrorEventDto -> toDomain()
179178 is ConnectingEventDto -> toDomain()
180179 is DisconnectedEventDto -> toDomain()
181180 is ErrorEventDto -> toDomain()
182- is HealthEventDto -> toDomain()
183181 is NotificationAddedToChannelEventDto -> toDomain()
184182 is UnknownEventDto -> toDomain()
185183 is GeneratedEventDto -> event.toDomain()
@@ -226,14 +224,15 @@ internal class EventMapping(
226224 )
227225 }
228226
229- private fun HealthEventDto.toDomain (): HealthEvent {
230- return HealthEvent (
231- type = type,
232- createdAt = created_at.date,
233- rawCreatedAt = created_at.rawDate,
234- connectionId = connection_id,
235- )
236- }
227+ /* *
228+ * Transforms the generated [GeneratedHealthCheckEvent] to [HealthEvent].
229+ */
230+ private fun GeneratedHealthCheckEvent.toDomain (): HealthEvent = HealthEvent (
231+ type = type,
232+ createdAt = createdAt.date,
233+ rawCreatedAt = createdAt.raw,
234+ connectionId = connectionId,
235+ )
237236
238237 /* *
239238 * Transforms the generated [GeneratedMessageDeletedEvent] to [MessageDeletedEvent].
@@ -507,6 +506,8 @@ internal class EventMapping(
507506 is GeneratedReactionDeletedEvent -> toDomain()
508507 is GeneratedReactionNewEvent -> toDomain()
509508 is GeneratedReactionUpdatedEvent -> toDomain()
509+ is GeneratedConnectedEvent -> toDomain()
510+ is GeneratedHealthCheckEvent -> toDomain()
510511 else -> error(" No mapping for the generated ${getWSEventType()} event" )
511512 }
512513
@@ -1339,15 +1340,16 @@ internal class EventMapping(
13391340 }
13401341
13411342 /* *
1342- * Transforms [ConnectedEventDto] to [ConnectedEvent].
1343+ * Transforms the generated [GeneratedConnectedEvent] to [ConnectedEvent]. On the v1 socket it is the
1344+ * `health.check` hello that carries `me`.
13431345 */
1344- private fun ConnectedEventDto .toDomain (): ConnectedEvent = with (domainMapping) {
1346+ private fun GeneratedConnectedEvent .toDomain (): ConnectedEvent = with (domainMapping) {
13451347 ConnectedEvent (
13461348 type = type,
1347- createdAt = created_at .date,
1348- rawCreatedAt = created_at.rawDate ,
1349+ createdAt = createdAt .date,
1350+ rawCreatedAt = createdAt.raw ,
13491351 me = me.toDomain(),
1350- connectionId = connection_id ,
1352+ connectionId = connectionId ,
13511353 )
13521354 }
13531355
0 commit comments