com.netflix.metacat.main.services.SessionProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metacat-main Show documentation
metacat-main
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.main.services;

import com.facebook.presto.Session;
import com.facebook.presto.metadata.SessionPropertyManager;
import com.facebook.presto.spi.security.Identity;
import com.google.inject.Inject;
import com.netflix.metacat.common.MetacatContext;
import com.netflix.metacat.common.QualifiedName;
import com.netflix.metacat.common.util.MetacatContextManager;
import com.netflix.metacat.main.connector.MetacatConnectorManager;

import java.util.Optional;

import static com.facebook.presto.spi.type.TimeZoneKey.UTC_KEY;
import static java.util.Locale.ENGLISH;

public class SessionProvider {
    @Inject
    MetacatConnectorManager metacatConnectorManager;
    @Inject
    SessionPropertyManager sessionPropertyManager;

    public Session getSession(QualifiedName name) {
        if (name.isDatabaseDefinition() || name.isTableDefinition() || name.isPartitionDefinition() || name.isViewDefinition()) {
            return getSession(name.getCatalogName(), name.getDatabaseName());
        } else {
            return getSession(name.getCatalogName());
        }
    }

    private Session getSession(String catalogName) {
        return getSession(catalogName, "%");
    }

    private Session getSession(String catalogName, String databaseName) {
        String source = metacatConnectorManager.getCatalogConfig(catalogName).getType();
        MetacatContext context = MetacatContextManager.getContext();
        return Session.builder(sessionPropertyManager)
                .setIdentity(new Identity(context.getUserName(), Optional.empty()))
                .setRemoteUserAddress(context.getClientId())
                .setSource(source)
                .setCatalog(catalogName)
                .setSchema(databaseName)
                .setTimeZoneKey(UTC_KEY)
                .setLocale(ENGLISH)
                .build();
    }
}