/** * @param fsCommand wire type * @return proto representation of given wire type */ public static alluxio.grpc.FileSystemCommand toProto(FileSystemCommand fsCommand) { return alluxio.grpc.FileSystemCommand.newBuilder() .setCommandType(toProto(fsCommand.getCommandType())) .setCommandOptions(FileSystemCommandOptions.newBuilder() .setPersistOptions(PersistCommandOptions.newBuilder().addAllPersistFiles( fsCommand.getCommandOptions().getPersistOptions().getFilesToPersist().stream() .map(GrpcUtils::toProto).collect(Collectors.toList())))) .build(); }
/** * Tests the {@link FileSystemMaster#workerHeartbeat} method. */ @Test public void workerHeartbeat() throws Exception { long blockId = createFileWithSingleBlock(ROOT_FILE_URI); long fileId = mFileSystemMaster.getFileId(ROOT_FILE_URI); mFileSystemMaster.scheduleAsyncPersistence(ROOT_FILE_URI); FileSystemCommand command = mFileSystemMaster .workerHeartbeat(mWorkerId1, Lists.newArrayList(fileId), WorkerHeartbeatContext.defaults()); assertEquals(alluxio.wire.CommandType.Persist, command.getCommandType()); assertEquals(0, command.getCommandOptions().getPersistOptions().getFilesToPersist().size()); }