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.
io.trino.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 io.trino.client;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import io.airlift.units.Duration;
import java.net.URI;
import java.nio.charset.CharsetEncoder;
import java.time.ZoneId;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Optional;
import java.util.Set;
import static com.google.common.base.MoreObjects.toStringHelper;
import static com.google.common.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 Optional principal;
private final Optional user;
private final Optional authorizationUser;
private final String source;
private final Optional traceToken;
private final Set clientTags;
private final String clientInfo;
private final Optional catalog;
private final Optional schema;
private final List path;
private final ZoneId timeZone;
private final Locale locale;
private final Map resourceEstimates;
private final Map properties;
private final Map preparedStatements;
private final Map roles;
private final Map extraCredentials;
private final String transactionId;
private final Duration clientRequestTimeout;
private final boolean compressionDisabled;
private Optional encoding;
public static Builder builder()
{
return new Builder();
}
public static Builder builder(ClientSession clientSession)
{
return new Builder(clientSession);
}
public static ClientSession stripTransactionId(ClientSession session)
{
return ClientSession.builder(session)
.transactionId(null)
.build();
}
private ClientSession(
URI server,
Optional principal,
Optional user,
Optional authorizationUser,
String source,
Optional traceToken,
Set clientTags,
String clientInfo,
Optional catalog,
Optional schema,
List path,
ZoneId timeZone,
Locale locale,
Map resourceEstimates,
Map properties,
Map preparedStatements,
Map roles,
Map extraCredentials,
String transactionId,
Duration clientRequestTimeout,
boolean compressionDisabled,
Optional encoding)
{
this.server = requireNonNull(server, "server is null");
this.principal = requireNonNull(principal, "principal is null");
this.user = requireNonNull(user, "user is null");
this.authorizationUser = requireNonNull(authorizationUser, "authorizationUser is null");
this.source = requireNonNull(source, "source is null");
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 = ImmutableList.copyOf(requireNonNull(path, "path is null"));
this.locale = locale;
this.timeZone = requireNonNull(timeZone, "timeZone is null");
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.roles = ImmutableMap.copyOf(requireNonNull(roles, "roles is null"));
this.extraCredentials = ImmutableMap.copyOf(requireNonNull(extraCredentials, "extraCredentials is null"));
this.clientRequestTimeout = clientRequestTimeout;
this.compressionDisabled = compressionDisabled;
this.encoding = requireNonNull(encoding, "encoding is null");
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());
}
// verify the extra credentials are valid
for (Entry entry : extraCredentials.entrySet()) {
checkArgument(!entry.getKey().isEmpty(), "Credential name is empty");
checkArgument(entry.getKey().indexOf('=') < 0, "Credential name must not contain '=': %s", entry.getKey());
checkArgument(charsetEncoder.canEncode(entry.getKey()), "Credential name is not US_ASCII: %s", entry.getKey());
checkArgument(charsetEncoder.canEncode(entry.getValue()), "Credential value is not US_ASCII: %s", entry.getValue());
}
}
public URI getServer()
{
return server;
}
public Optional getPrincipal()
{
return principal;
}
public Optional getUser()
{
return user;
}
public Optional getAuthorizationUser()
{
return authorizationUser;
}
public String getSource()
{
return source;
}
public Optional getTraceToken()
{
return traceToken;
}
public Set getClientTags()
{
return clientTags;
}
public String getClientInfo()
{
return clientInfo;
}
public Optional getCatalog()
{
return catalog;
}
public Optional getSchema()
{
return schema;
}
public List getPath()
{
return path;
}
public ZoneId getTimeZone()
{
return timeZone;
}
public Locale getLocale()
{
return locale;
}
public Map getResourceEstimates()
{
return resourceEstimates;
}
public Map getProperties()
{
return properties;
}
public Map getPreparedStatements()
{
return preparedStatements;
}
/**
* Returns the map of catalog name -> selected role
*/
public Map getRoles()
{
return roles;
}
public Map getExtraCredentials()
{
return extraCredentials;
}
public String getTransactionId()
{
return transactionId;
}
public boolean isDebug()
{
return false;
}
public Duration getClientRequestTimeout()
{
return clientRequestTimeout;
}
public boolean isCompressionDisabled()
{
return compressionDisabled;
}
public Optional getEncoding()
{
return encoding;
}
@Override
public String toString()
{
return toStringHelper(this)
.add("server", server)
.add("principal", principal)
.add("user", user)
.add("authorizationUser", authorizationUser)
.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)
.add("source", source)
.add("resourceEstimates", resourceEstimates)
.add("clientRequestTimeout", clientRequestTimeout)
.add("compressionDisabled", compressionDisabled)
.add("encoding", encoding)
.omitNullValues()
.toString();
}
public static final class Builder
{
private URI server;
private Optional principal = Optional.empty();
private Optional user = Optional.empty();
private Optional authorizationUser = Optional.empty();
private String source;
private Optional traceToken = Optional.empty();
private Set clientTags = ImmutableSet.of();
private String clientInfo;
private String catalog;
private String schema;
private List path = ImmutableList.of();
private ZoneId timeZone;
private Locale locale;
private Map resourceEstimates = ImmutableMap.of();
private Map properties = ImmutableMap.of();
private Map preparedStatements = ImmutableMap.of();
private Map roles = ImmutableMap.of();
private Map credentials = ImmutableMap.of();
private String transactionId;
private Duration clientRequestTimeout;
private boolean compressionDisabled;
private Optional encoding = Optional.empty();
private Builder() {}
private Builder(ClientSession clientSession)
{
requireNonNull(clientSession, "clientSession is null");
server = clientSession.getServer();
principal = clientSession.getPrincipal();
user = clientSession.getUser();
authorizationUser = clientSession.getAuthorizationUser();
source = clientSession.getSource();
traceToken = clientSession.getTraceToken();
clientTags = clientSession.getClientTags();
clientInfo = clientSession.getClientInfo();
catalog = clientSession.getCatalog().orElse(null);
schema = clientSession.getSchema().orElse(null);
path = clientSession.getPath();
timeZone = clientSession.getTimeZone();
locale = clientSession.getLocale();
resourceEstimates = clientSession.getResourceEstimates();
properties = clientSession.getProperties();
preparedStatements = clientSession.getPreparedStatements();
roles = clientSession.getRoles();
credentials = clientSession.getExtraCredentials();
transactionId = clientSession.getTransactionId();
clientRequestTimeout = clientSession.getClientRequestTimeout();
compressionDisabled = clientSession.isCompressionDisabled();
encoding = clientSession.getEncoding();
}
public Builder server(URI server)
{
this.server = server;
return this;
}
public Builder user(Optional user)
{
this.user = user;
return this;
}
public Builder authorizationUser(Optional authorizationUser)
{
this.authorizationUser = authorizationUser;
return this;
}
public Builder principal(Optional principal)
{
this.principal = principal;
return this;
}
public Builder source(String source)
{
this.source = source;
return this;
}
public Builder traceToken(Optional traceToken)
{
this.traceToken = traceToken;
return this;
}
public Builder clientTags(Set clientTags)
{
this.clientTags = clientTags;
return this;
}
public Builder clientInfo(String clientInfo)
{
this.clientInfo = clientInfo;
return this;
}
public Builder catalog(String catalog)
{
this.catalog = catalog;
return this;
}
public Builder schema(String schema)
{
this.schema = schema;
return this;
}
public Builder path(List path)
{
this.path = path;
return this;
}
public Builder timeZone(ZoneId timeZone)
{
this.timeZone = timeZone;
return this;
}
public Builder locale(Locale locale)
{
this.locale = locale;
return this;
}
public Builder resourceEstimates(Map resourceEstimates)
{
this.resourceEstimates = resourceEstimates;
return this;
}
public Builder properties(Map properties)
{
this.properties = properties;
return this;
}
public Builder roles(Map roles)
{
this.roles = roles;
return this;
}
public Builder credentials(Map credentials)
{
this.credentials = credentials;
return this;
}
public Builder preparedStatements(Map preparedStatements)
{
this.preparedStatements = preparedStatements;
return this;
}
public Builder transactionId(String transactionId)
{
this.transactionId = transactionId;
return this;
}
public Builder clientRequestTimeout(Duration clientRequestTimeout)
{
this.clientRequestTimeout = clientRequestTimeout;
return this;
}
public Builder compressionDisabled(boolean compressionDisabled)
{
this.compressionDisabled = compressionDisabled;
return this;
}
public Builder encoding(Optional encoding)
{
this.encoding = encoding;
return this;
}
public ClientSession build()
{
return new ClientSession(
server,
principal,
user,
authorizationUser,
source,
traceToken,
clientTags,
clientInfo,
Optional.ofNullable(catalog),
Optional.ofNullable(schema),
path,
timeZone,
locale,
resourceEstimates,
properties,
preparedStatements,
roles,
credentials,
transactionId,
clientRequestTimeout,
compressionDisabled,
encoding);
}
}
}
