BW for lineage master Thrift services API.

Alluxio · May 16, 2017 · d5e178a · d5e178a
1 parent 0995507
commit d5e178a
Show file tree

Hide file tree

Showing 14 changed files with 4,456 additions and 350 deletions.
diff --git a/core/client/fs/src/main/java/alluxio/client/lineage/LineageMasterClient.java b/core/client/fs/src/main/java/alluxio/client/lineage/LineageMasterClient.java
@@ -15,7 +15,12 @@
 import alluxio.Constants;
 import alluxio.job.CommandLineJob;
 import alluxio.thrift.AlluxioService;
+import alluxio.thrift.CreateLineageTOptions;
+import alluxio.thrift.DeleteLineageTOptions;
+import alluxio.thrift.GetLineageInfoListTOptions;
 import alluxio.thrift.LineageMasterClientService;
+import alluxio.thrift.ReinitializeFileTOptions;
+import alluxio.thrift.ReportLostFileTOptions;
 import alluxio.wire.LineageInfo;
 import alluxio.wire.ThriftUtils;
 import alluxio.wire.TtlAction;
@@ -82,7 +87,8 @@ public synchronized long createLineage(final List<String> inputFiles,
       @Override
       public Long call() throws TException {
         return mClient.createLineage(inputFiles, outputFiles,
-            ThriftUtils.toThrift(job.generateCommandLineJobInfo()));
+            ThriftUtils.toThrift(job.generateCommandLineJobInfo()), new CreateLineageTOptions())
+            .getId();
       }
     });
   }
@@ -99,7 +105,7 @@ public synchronized boolean deleteLineage(final long lineageId, final boolean ca
     return retryRPC(new RpcCallable<Boolean>() {
       @Override
       public Boolean call() throws TException {
-        return mClient.deleteLineage(lineageId, cascade);
+        return mClient.deleteLineage(lineageId, cascade, new DeleteLineageTOptions()).isSuccess();
       }
     });
   }
@@ -118,8 +124,8 @@ public synchronized long reinitializeFile(final String path, final long blockSiz
     return retryRPC(new RpcCallable<Long>() {
       @Override
       public Long call() throws TException {
-        return mClient.reinitializeFile(path, blockSizeBytes, ttl,
+        return mClient.reinitializeFile(path, blockSizeBytes, ttl, ThriftUtils.toThrift(ttlAction),
-            ThriftUtils.toThrift(ttlAction));
+            new ReinitializeFileTOptions()).getId();
       }
     });
   }
@@ -134,7 +140,8 @@ public synchronized List<LineageInfo> getLineageInfoList() throws IOException {
       @Override
       public List<LineageInfo> call() throws TException {
         List<LineageInfo> result = new ArrayList<>();
-        for (alluxio.thrift.LineageInfo lineageInfo : mClient.getLineageInfoList()) {
+        for (alluxio.thrift.LineageInfo lineageInfo : mClient
+            .getLineageInfoList(new GetLineageInfoListTOptions()).getLineageInfoList()) {
           result.add(ThriftUtils.fromThrift(lineageInfo));
         }
         return result;
@@ -151,7 +158,7 @@ public synchronized void reportLostFile(final String path) throws IOException {
     retryRPC(new RpcCallable<Void>() {
       @Override
       public Void call() throws TException {
-        mClient.reportLostFile(path);
+        mClient.reportLostFile(path, new ReportLostFileTOptions());
         return null;
       }
     });

diff --git a/core/common/src/main/java/alluxio/thrift/CreateLineageTOptions.java b/core/common/src/main/java/alluxio/thrift/CreateLineageTOptions.java