Add units for Alluxio metrics
@@ -19,6 +19,8 @@ import alluxio.client.quota.CacheScope;
import com.facebook.presto.hive.HiveFileContext;
import com.google.common.collect.ImmutableMap;
+import static com.facebook.presto.common.RuntimeUnit.BYTE;
+import static com.facebook.presto.common.RuntimeUnit.NANO;
import static com.facebook.presto.common.RuntimeUnit.NONE;
import static java.util.Objects.requireNonNull;
@@ -43,6 +45,7 @@ public class PrestoCacheContext
}
return context;
}
+
private PrestoCacheContext(HiveFileContext hiveFileContext)
{
this.hiveFileContext = requireNonNull(hiveFileContext, "hiveFileContext is null");
@@ -51,8 +54,17 @@ public class PrestoCacheContext
@Override
public void incrementCounter(String name, StatsUnit unit, long value)
{
+ switch (unit) {
+ case BYTE:
+ hiveFileContext.incrementCounter(name, BYTE, value);
+ break;
+ case NANO:
+ hiveFileContext.incrementCounter(name, NANO, value);
+ break;
+ default:
hiveFileContext.incrementCounter(name, NONE, value);
}
+ }
public HiveFileContext getHiveFileContext()
{
