com.netflix.metacat.thrift.CatalogThriftEventHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metacat-thrift Show documentation
metacat-thrift
There is a newer version: 1.3.1
/*
 * Copyright 2016 Netflix, Inc.
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *        http://www.apache.org/licenses/LICENSE-2.0
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package com.netflix.metacat.thrift;

import com.netflix.metacat.common.MetacatContext;
import com.netflix.metacat.common.util.MetacatContextManager;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.server.ServerContext;
import org.apache.thrift.server.TServerEventHandler;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class CatalogThriftEventHandler implements TServerEventHandler {
    private static final Logger log = LoggerFactory.getLogger(CatalogThriftEventHandler.class);

    @Override
    public ServerContext createContext(TProtocol input, TProtocol output) {
        String userName = "metacat-thrift-interface";
        String clientAppName = null; //requestContext.getHeaderString(MetacatContext.HEADER_KEY_CLIENT_APP_NAME);
        String clientHost = null; //requestContext.getHeaderString("X-Forwarded-For");
        String jobId = null; //requestContext.getHeaderString(MetacatContext.HEADER_KEY_JOB_ID);
        String dataTypeContext = MetacatContext.DATA_TYPE_CONTEXTS.hive.name();
        long requestThreadId = Thread.currentThread().getId();

        TTransport transport = input.getTransport();
        if (transport instanceof TSocket) {
            TSocket thriftSocket = (TSocket) transport;
            clientHost = thriftSocket.getSocket().getInetAddress().getHostAddress();
        }

        CatalogServerContext context = new CatalogServerContext(userName, clientAppName, clientHost, jobId,
                dataTypeContext, requestThreadId);
        MetacatContextManager.setContext(context);
        return context;
    }

    @Override
    public void deleteContext(ServerContext serverContext, TProtocol input, TProtocol output) {
        validateRequest((CatalogServerContext) serverContext);
        MetacatContextManager.removeContext();
    }

    @Override
    public void preServe() {
        // nothing to do
    }

    @Override
    public void processContext(ServerContext serverContext, TTransport inputTransport, TTransport outputTransport) {
        validateRequest((CatalogServerContext) serverContext);
    }

    protected void validateRequest(CatalogServerContext serverContext) {
        long requestThreadId = serverContext.requestThreadId;
        if (requestThreadId != Thread.currentThread().getId()) {
            throw new IllegalStateException("Expect all processing to happen on the same thread as the request thread");
        }
    }

    public static class CatalogServerContext extends MetacatContext implements ServerContext {
        public final long requestThreadId;

        public CatalogServerContext(String userName, String clientAppName, String clientId, String jobId,
                String dataTypeContext, long requestThreadId) {
            super(userName, clientAppName, clientId, jobId, dataTypeContext);
            this.requestThreadId = requestThreadId;
        }
    }
}