Introduce fragment result cache manager

presto-main/src/main/java/com/facebook/presto/operator/FileFragmentResultCacheConfig.java

@@ -0,0 +1,120 @@
+/*
+ * 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 com.facebook.airlift.configuration.Config;
+import com.facebook.airlift.configuration.ConfigDescription;
+import io.airlift.units.DataSize;
+import io.airlift.units.Duration;
+import io.airlift.units.MinDataSize;
+import io.airlift.units.MinDuration;
+
+import javax.validation.constraints.Min;
+
+import java.net.URI;
+
+import static io.airlift.units.DataSize.Unit.GIGABYTE;
+import static java.util.concurrent.TimeUnit.DAYS;
+
+public class FileFragmentResultCacheConfig
+{
+    private boolean cachingEnabled;
+    private URI baseDirectory;
+    private boolean blockEncodingCompressionEnabled;
+
+    private int maxCachedEntries = 10_000;
+    private Duration cacheTtl = new Duration(2, DAYS);
+    private DataSize maxInFlightSize = new DataSize(1, GIGABYTE);
+
+    public boolean isCachingEnabled()
+    {
+        return cachingEnabled;
+    }
+
+    @Config("fragment-result-cache.enabled")
+    @ConfigDescription("Enable fragment result caching")
+    public FileFragmentResultCacheConfig setCachingEnabled(boolean cachingEnabled)
+    {
+        this.cachingEnabled = cachingEnabled;
+        return this;
+    }
+
+    public URI getBaseDirectory()
+    {
+        return baseDirectory;
+    }
+
+    @Config("fragment-result-cache.base-directory")
+    @ConfigDescription("Base URI to cache data")
+    public FileFragmentResultCacheConfig setBaseDirectory(URI baseDirectory)
+    {
+        this.baseDirectory = baseDirectory;
+        return this;
+    }
+
+    public boolean isBlockEncodingCompressionEnabled()
+    {
+        return blockEncodingCompressionEnabled;
+    }
+
+    @Config("fragment-result-cache.block-encoding-compression-enabled")
+    @ConfigDescription("Enable compression for block encoding")
+    public FileFragmentResultCacheConfig setBlockEncodingCompressionEnabled(boolean blockEncodingCompressionEnabled)
+    {
+        this.blockEncodingCompressionEnabled = blockEncodingCompressionEnabled;
+        return this;
+    }
+
+    @Min(0)
+    public int getMaxCachedEntries()
+    {
+        return maxCachedEntries;
+    }
+
+    @Config("fragment-result-cache.max-cached-entries")
+    @ConfigDescription("Number of entries allowed in cache")
+    public FileFragmentResultCacheConfig setMaxCachedEntries(int maxCachedEntries)
+    {
+        this.maxCachedEntries = maxCachedEntries;
+        return this;
+    }
+
+    @MinDuration("0s")
+    public Duration getCacheTtl()
+    {
+        return cacheTtl;
+    }
+
+    @Config("fragment-result-cache.cache-ttl")
+    @ConfigDescription("Time-to-live for a cache entry")
+    public FileFragmentResultCacheConfig setCacheTtl(Duration cacheTtl)
+    {
+        this.cacheTtl = cacheTtl;
+        return this;
+    }
+
+    @MinDataSize("0B")
+    public DataSize getMaxInFlightSize()
+    {
+        return maxInFlightSize;
+    }
+
+    @Config("fragment-result-cache.max-in-flight-size")
+    @ConfigDescription("Maximum size of pages in memory waiting to be flushed")
+    public FileFragmentResultCacheConfig setMaxInFlightSize(DataSize maxInFlightSize)
+    {
+        this.maxInFlightSize = maxInFlightSize;
+        return this;
+    }
+}