/*
* 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.operator;
import io.airlift.configuration.Config;
import io.airlift.http.client.HttpClientConfig;
import io.airlift.units.DataSize;
import io.airlift.units.DataSize.Unit;
import io.airlift.units.Duration;
import io.airlift.units.MinDataSize;
import io.airlift.units.MinDuration;
import javax.validation.constraints.Min;
import javax.validation.constraints.NotNull;
import java.util.concurrent.TimeUnit;
public class ExchangeClientConfig
{
private DataSize maxBufferSize = new DataSize(32, Unit.MEGABYTE);
private int concurrentRequestMultiplier = 3;
private Duration minErrorDuration = new Duration(1, TimeUnit.MINUTES);
private Duration maxErrorDuration = new Duration(5, TimeUnit.MINUTES);
private DataSize maxResponseSize = new HttpClientConfig().getMaxContentLength();
private int clientThreads = 25;
@NotNull
public DataSize getMaxBufferSize()
{
return maxBufferSize;
}
@Config("exchange.max-buffer-size")
public ExchangeClientConfig setMaxBufferSize(DataSize maxBufferSize)
{
this.maxBufferSize = maxBufferSize;
return this;
}
@Min(1)
public int getConcurrentRequestMultiplier()
{
return concurrentRequestMultiplier;
}
@Config("exchange.concurrent-request-multiplier")
public ExchangeClientConfig setConcurrentRequestMultiplier(int concurrentRequestMultiplier)
{
this.concurrentRequestMultiplier = concurrentRequestMultiplier;
return this;
}
@NotNull
@MinDuration("1ms")
public Duration getMinErrorDuration()
{
return minErrorDuration;
}
@Config("exchange.min-error-duration")
public ExchangeClientConfig setMinErrorDuration(Duration minErrorDuration)
{
this.minErrorDuration = minErrorDuration;
return this;
}
@NotNull
@MinDuration("1ms")
public Duration getMaxErrorDuration()
{
return maxErrorDuration;
}
@Config("exchange.max-error-duration")
public ExchangeClientConfig setMaxErrorDuration(Duration maxErrorDuration)
{
this.maxErrorDuration = maxErrorDuration;
return this;
}
@NotNull
@MinDataSize("1MB")
public DataSize getMaxResponseSize()
{
return maxResponseSize;
}
@Config("exchange.max-response-size")
public ExchangeClientConfig setMaxResponseSize(DataSize maxResponseSize)
{
this.maxResponseSize = maxResponseSize;
return this;
}
@Min(1)
public int getClientThreads()
{
return clientThreads;
}
@Config("exchange.client-threads")
public ExchangeClientConfig setClientThreads(int clientThreads)
{
this.clientThreads = clientThreads;
return this;
}
}