orbit.server.pipeline.step.IdentityStep.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of orbit-server Show documentation

Orbit is a system to make building highly scalable realtime services easier.

There is a newer version: 2.0.0a19

/*
 Copyright (C) 2015 - 2019 Electronic Arts Inc.  All rights reserved.
 This file is part of the Orbit Project .
 See license in LICENSE.
 */

package orbit.server.pipeline.step

import orbit.server.mesh.LocalNodeInfo
import orbit.server.pipeline.PipelineContext
import orbit.shared.net.Message
import java.util.concurrent.atomic.AtomicLong

class IdentityStep(
    private val localNodeInfo: LocalNodeInfo
) : PipelineStep {
    private val messageCounter = AtomicLong(0)

    override suspend fun onOutbound(context: PipelineContext, msg: Message) {
        msg.copy(
            messageId = msg.messageId ?: messageCounter.incrementAndGet(),
            source = msg.source ?: localNodeInfo.info.id
        ).also {
            context.next(it)
        }
    }
}