orbit.server.service.ServerAuthInterceptor.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.service

import io.grpc.Context
import io.grpc.Contexts
import io.grpc.Metadata
import io.grpc.ServerCall
import io.grpc.ServerCallHandler
import io.grpc.ServerInterceptor
import orbit.shared.mesh.Namespace
import orbit.shared.mesh.NodeId
import orbit.shared.mesh.NodeKey
import orbit.shared.proto.Headers

class ServerAuthInterceptor : ServerInterceptor {
    companion object Keys {
        val NODE_KEY = Context.key(Headers.NODE_KEY_NAME)!!
        val NAMESPACE = Context.key(Headers.NAMESPACE_NAME)!!
        val NODE_ID = Context.key("nodeId")!!
    }

    override fun  interceptCall(
        call: ServerCall?,
        headers: Metadata,
        next: ServerCallHandler
    ): ServerCall.Listener {

        val nodeKey = headers.get(Metadata.Key.of(Headers.NODE_KEY_NAME, Metadata.ASCII_STRING_MARSHALLER))
        val namespace = headers.get(Metadata.Key.of(Headers.NAMESPACE_NAME, Metadata.ASCII_STRING_MARSHALLER))

        val context = Context.current().let {
            if (namespace != null) it.withValue(NAMESPACE, namespace) else it
        }.let {
            if (nodeKey != null) it.withValue(NODE_KEY, nodeKey) else it
        }.let {
            if (namespace != null && nodeKey != null) it.withValue(NODE_ID, NodeId(nodeKey, namespace)) else it
        }

        return Contexts.interceptCall(context, call, headers, next)
    }
}