com.facebook.presto.jdbc.internal.client.ClientSession Maven / Gradle / Ivy
/*
* 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.jdbc.internal.client;
import com.facebook.presto.jdbc.internal.spi.type.TimeZoneKey;
import com.facebook.presto.jdbc.internal.guava.collect.ImmutableMap;
import com.facebook.presto.jdbc.internal.guava.collect.ImmutableSet;
import com.facebook.presto.jdbc.internal.airlift.units.Duration;
import java.net.URI;
import java.nio.charset.CharsetEncoder;
import java.util.Locale;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Optional;
import java.util.Set;
import static com.facebook.presto.jdbc.internal.guava.base.MoreObjects.toStringHelper;
import static com.facebook.presto.jdbc.internal.guava.base.Preconditions.checkArgument;
import static java.nio.charset.StandardCharsets.US_ASCII;
import static java.util.Objects.requireNonNull;
public class ClientSession
{
private final URI server;
private final String user;
private final String source;
private final Optional traceToken;
private final Set clientTags;
private final String clientInfo;
private final String catalog;
private final String schema;
private final String path;
private final TimeZoneKey timeZone;
private final Locale locale;
private final Map resourceEstimates;
private final Map properties;
private final Map preparedStatements;
private final String transactionId;
private final Duration clientRequestTimeout;
public static Builder builder(ClientSession clientSession)
{
return new Builder(clientSession);
}
public static ClientSession stripTransactionId(ClientSession session)
{
return ClientSession.builder(session)
.withoutTransactionId()
.build();
}
public ClientSession(
URI server,
String user,
String source,
Optional traceToken,
Set clientTags,
String clientInfo,
String catalog,
String schema,
String path,
String timeZoneId,
Locale locale,
Map resourceEstimates,
Map properties,
Map preparedStatements,
String transactionId,
Duration clientRequestTimeout)
{
this.server = requireNonNull(server, "server is null");
this.user = user;
this.source = source;
this.traceToken = requireNonNull(traceToken, "traceToken is null");
this.clientTags = ImmutableSet.copyOf(requireNonNull(clientTags, "clientTags is null"));
this.clientInfo = clientInfo;
this.catalog = catalog;
this.schema = schema;
this.path = path;
this.locale = locale;
this.timeZone = TimeZoneKey.getTimeZoneKey(timeZoneId);
this.transactionId = transactionId;
this.resourceEstimates = ImmutableMap.copyOf(requireNonNull(resourceEstimates, "resourceEstimates is null"));
this.properties = ImmutableMap.copyOf(requireNonNull(properties, "properties is null"));
this.preparedStatements = ImmutableMap.copyOf(requireNonNull(preparedStatements, "preparedStatements is null"));
this.clientRequestTimeout = clientRequestTimeout;
for (String clientTag : clientTags) {
checkArgument(!clientTag.contains(","), "client tag cannot contain ','");
}
// verify that resource estimates are valid
CharsetEncoder charsetEncoder = US_ASCII.newEncoder();
for (Entry entry : resourceEstimates.entrySet()) {
checkArgument(!entry.getKey().isEmpty(), "Resource name is empty");
checkArgument(entry.getKey().indexOf('=') < 0, "Resource name must not contain '=': %s", entry.getKey());
checkArgument(charsetEncoder.canEncode(entry.getKey()), "Resource name is not US_ASCII: %s", entry.getKey());
}
// verify the properties are valid
for (Entry entry : properties.entrySet()) {
checkArgument(!entry.getKey().isEmpty(), "Session property name is empty");
checkArgument(entry.getKey().indexOf('=') < 0, "Session property name must not contain '=': %s", entry.getKey());
checkArgument(charsetEncoder.canEncode(entry.getKey()), "Session property name is not US_ASCII: %s", entry.getKey());
checkArgument(charsetEncoder.canEncode(entry.getValue()), "Session property value is not US_ASCII: %s", entry.getValue());
}
}
public URI getServer()
{
return server;
}
public String getUser()
{
return user;
}
public String getSource()
{
return source;
}
public Optional getTraceToken()
{
return traceToken;
}
public Set getClientTags()
{
return clientTags;
}
public String getClientInfo()
{
return clientInfo;
}
public String getCatalog()
{
return catalog;
}
public String getSchema()
{
return schema;
}
public String getPath()
{
return path;
}
public TimeZoneKey getTimeZone()
{
return timeZone;
}
public Locale getLocale()
{
return locale;
}
public Map getResourceEstimates()
{
return resourceEstimates;
}
public Map getProperties()
{
return properties;
}
public Map getPreparedStatements()
{
return preparedStatements;
}
public String getTransactionId()
{
return transactionId;
}
public boolean isDebug()
{
return false;
}
public Duration getClientRequestTimeout()
{
return clientRequestTimeout;
}
@Override
public String toString()
{
return toStringHelper(this)
.add("server", server)
.add("user", user)
.add("clientTags", clientTags)
.add("clientInfo", clientInfo)
.add("catalog", catalog)
.add("schema", schema)
.add("path", path)
.add("traceToken", traceToken.orElse(null))
.add("timeZone", timeZone)
.add("locale", locale)
.add("properties", properties)
.add("transactionId", transactionId)
.omitNullValues()
.toString();
}
public static final class Builder
{
private URI server;
private String user;
private String source;
private Optional traceToken;
private Set clientTags;
private String clientInfo;
private String catalog;
private String schema;
private String path;
private TimeZoneKey timeZone;
private Locale locale;
private Map resourceEstimates;
private Map properties;
private Map preparedStatements;
private String transactionId;
private Duration clientRequestTimeout;
private Builder(ClientSession clientSession)
{
requireNonNull(clientSession, "clientSession is null");
server = clientSession.getServer();
user = clientSession.getUser();
source = clientSession.getSource();
traceToken = clientSession.getTraceToken();
clientTags = clientSession.getClientTags();
clientInfo = clientSession.getClientInfo();
catalog = clientSession.getCatalog();
schema = clientSession.getSchema();
path = clientSession.getPath();
timeZone = clientSession.getTimeZone();
locale = clientSession.getLocale();
resourceEstimates = clientSession.getResourceEstimates();
properties = clientSession.getProperties();
preparedStatements = clientSession.getPreparedStatements();
transactionId = clientSession.getTransactionId();
clientRequestTimeout = clientSession.getClientRequestTimeout();
}
public Builder withCatalog(String catalog)
{
this.catalog = requireNonNull(catalog, "catalog is null");
return this;
}
public Builder withSchema(String schema)
{
this.schema = requireNonNull(schema, "schema is null");
return this;
}
public Builder withPath(String path)
{
this.path = requireNonNull(path, "path is null");
return this;
}
public Builder withProperties(Map properties)
{
this.properties = requireNonNull(properties, "properties is null");
return this;
}
public Builder withPreparedStatements(Map preparedStatements)
{
this.preparedStatements = requireNonNull(preparedStatements, "preparedStatements is null");
return this;
}
public Builder withTransactionId(String transactionId)
{
this.transactionId = requireNonNull(transactionId, "transactionId is null");
return this;
}
public Builder withoutTransactionId()
{
this.transactionId = null;
return this;
}
public ClientSession build()
{
return new ClientSession(
server,
user,
source,
traceToken,
clientTags,
clientInfo,
catalog,
schema,
path,
timeZone.getId(),
locale,
resourceEstimates,
properties,
preparedStatements,
transactionId,
clientRequestTimeout);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy