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

Go to download
/*
 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.ClusterManager
import orbit.server.pipeline.PipelineContext
import orbit.shared.exception.InvalidNodeId
import orbit.shared.net.Message

class VerifyStep(
    private val clusterManager: ClusterManager
) : PipelineStep {
    override suspend fun onInbound(context: PipelineContext, msg: Message) {
        val source = msg.source

        checkNotNull(source) { "Source should not be null at this point" }

        if (clusterManager.getNode(source) == null) {
            throw InvalidNodeId(source)
        }

        context.next(msg)
    }
}