Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.facebook.presto.spark;
import com.facebook.presto.common.RuntimeStats;
import com.facebook.presto.common.transaction.TransactionId;
import com.facebook.presto.server.SessionContext;
import com.facebook.presto.spark.accesscontrol.PrestoSparkAuthenticatorProvider;
import com.facebook.presto.spark.accesscontrol.PrestoSparkCredentialsProvider;
import com.facebook.presto.spark.classloader_interface.PrestoSparkSession;
import com.facebook.presto.spi.function.SqlFunctionId;
import com.facebook.presto.spi.function.SqlInvokedFunction;
import com.facebook.presto.spi.security.Identity;
import com.facebook.presto.spi.security.TokenAuthenticator;
import com.facebook.presto.spi.session.ResourceEstimates;
import com.facebook.presto.spi.tracing.Tracer;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import javax.annotation.Nullable;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import static java.util.Objects.requireNonNull;
public class PrestoSparkSessionContext
implements SessionContext
{
private final Identity identity;
private final String catalog;
private final String schema;
private final String source;
private final String userAgent;
private final String clientInfo;
private final Set clientTags;
private final String timeZoneId;
private final String language;
private final Map systemProperties;
private final Map> catalogSessionProperties;
private final Optional traceToken;
private final RuntimeStats runtimeStats = new RuntimeStats();
public static PrestoSparkSessionContext createFromSessionInfo(
PrestoSparkSession prestoSparkSession,
Set credentialsProviders,
Set authenticatorProviders)
{
ImmutableMap.Builder extraCredentials = ImmutableMap.builder();
extraCredentials.putAll(prestoSparkSession.getExtraCredentials());
credentialsProviders.forEach(provider -> extraCredentials.putAll(provider.getCredentials()));
ImmutableMap.Builder extraTokenAuthenticators = ImmutableMap.builder();
authenticatorProviders.forEach(provider -> extraTokenAuthenticators.putAll(provider.getTokenAuthenticators()));
return new PrestoSparkSessionContext(
new Identity(
prestoSparkSession.getUser(),
prestoSparkSession.getPrincipal(),
ImmutableMap.of(), // presto on spark does not support role management
extraCredentials.build(),
extraTokenAuthenticators.build(),
Optional.empty(),
Optional.empty()),
prestoSparkSession.getCatalog().orElse(null),
prestoSparkSession.getSchema().orElse(null),
prestoSparkSession.getSource().orElse(null),
prestoSparkSession.getUserAgent().orElse(null),
prestoSparkSession.getClientInfo().orElse(null),
prestoSparkSession.getClientTags(),
prestoSparkSession.getTimeZoneId().orElse(null),
prestoSparkSession.getLanguage().orElse(null),
prestoSparkSession.getSystemProperties(),
prestoSparkSession.getCatalogSessionProperties(),
prestoSparkSession.getTraceToken());
}
public PrestoSparkSessionContext(
Identity identity,
String catalog,
String schema,
String source,
String userAgent,
String clientInfo,
Set clientTags,
String timeZoneId,
String language,
Map systemProperties,
Map> catalogSessionProperties,
Optional traceToken)
{
this.identity = requireNonNull(identity, "identity is null");
this.catalog = catalog;
this.schema = schema;
this.source = source;
this.userAgent = userAgent;
this.clientInfo = clientInfo;
this.clientTags = ImmutableSet.copyOf(requireNonNull(clientTags, "clientTags is null"));
this.timeZoneId = timeZoneId;
this.language = language;
this.systemProperties = ImmutableMap.copyOf(requireNonNull(systemProperties, "systemProperties is null"));
this.catalogSessionProperties = ImmutableMap.copyOf(requireNonNull(catalogSessionProperties, "catalogSessionProperties is null"));
this.traceToken = requireNonNull(traceToken, "traceToken is null");
}
@Override
public Identity getIdentity()
{
return identity;
}
@Nullable
@Override
public String getCatalog()
{
return catalog;
}
@Nullable
@Override
public String getSchema()
{
return schema;
}
@Nullable
@Override
public String getSource()
{
return source;
}
@Override
public String getRemoteUserAddress()
{
return "localhost";
}
@Nullable
@Override
public String getUserAgent()
{
return userAgent;
}
@Nullable
@Override
public String getClientInfo()
{
return clientInfo;
}
@Override
public Set getClientTags()
{
return clientTags;
}
@Override
public ResourceEstimates getResourceEstimates()
{
// presto on spark does not use resource groups
return new ResourceEstimates(Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty());
}
@Nullable
@Override
public String getTimeZoneId()
{
return timeZoneId;
}
@Nullable
@Override
public String getLanguage()
{
return language;
}
@Override
public Optional getTracer()
{
return Optional.empty();
}
@Override
public Map getSystemProperties()
{
return systemProperties;
}
@Override
public Map> getCatalogSessionProperties()
{
return catalogSessionProperties;
}
@Override
public Map getPreparedStatements()
{
// presto on spark does not support prepared statements
return ImmutableMap.of();
}
@Override
public Optional getTransactionId()
{
// presto on spark does not support explicit transaction management
return Optional.empty();
}
@Override
public Optional getTraceToken()
{
return traceToken;
}
@Override
public boolean supportClientTransaction()
{
return false;
}
@Override
public Map getSessionFunctions()
{
// presto on spark does not support session functions
return ImmutableMap.of();
}
@Override
public RuntimeStats getRuntimeStats()
{
return runtimeStats;
}
}
