Examples with ActionStatus org.smartdata.actions.ActionStatus used on opensource projects

Example 6 with ActionStatus

use of org.smartdata.actions.ActionStatus in project SSM by Intel-bigdata.

the class TestCacheFile method testCacheFile.

@Test
public void testCacheFile() throws IOException {
    dfs.mkdirs(new Path("/fileTestA"));
    String[] args = { "/fileTestA" };
    CacheFileAction cacheAction = new CacheFileAction();
    cacheAction.setContext(smartContext);
    cacheAction.setDfsClient(dfsClient);
    cacheAction.init(args);
    ActionStatus actionStatus = cacheAction.getActionStatus();
    try {
        Assert.assertEquals(false, cacheAction.isCached(args[0]));
        cacheAction.run();
        Assert.assertEquals(true, cacheAction.isCached(args[0]));
        Assert.assertTrue(actionStatus.isFinished());
        Assert.assertTrue(actionStatus.isSuccessful());
        System.out.println("Cache action running time : " + StringUtils.formatTime(actionStatus.getRunningTime()));
        Assert.assertEquals(1.0f, actionStatus.getPercentage(), 0.00001f);
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
}

Example 7 with ActionStatus

use of org.smartdata.actions.ActionStatus in project SSM by Intel-bigdata.

the class WriteFileAction method execute.

@Override
protected void execute() {
    ActionStatus actionStatus = getActionStatus();
    actionStatus.begin();
    try {
        final OutputStream out = dfsClient.create(filePath, true);
        // generate random data with given length
        byte[] buffer = new byte[bufferSize];
        new Random().nextBytes(buffer);
        // write to HDFS
        for (int pos = 0; pos < length; pos += bufferSize) {
            int writeLength = pos + bufferSize < length ? bufferSize : length - pos;
            out.write(buffer, 0, writeLength);
        }
        out.close();
        actionStatus.setSuccessful(true);
    } catch (IOException e) {
        actionStatus.setSuccessful(false);
        resultOut.println("WriteFile Action fails!\n" + e.getMessage());
    } finally {
        actionStatus.end();
    }
}

Example 8 with ActionStatus

use of org.smartdata.actions.ActionStatus in project SSM by Intel-bigdata.

the class CheckStorageAction method execute.

@Override
protected void execute() {
    ActionStatus actionStatus = getActionStatus();
    actionStatus.begin();
    try {
        HdfsFileStatus fileStatus = dfsClient.getFileInfo(fileName);
        long length = fileStatus.getLen();
        BlockLocation[] blockLocations = dfsClient.getBlockLocations(fileName, 0, length);
        for (BlockLocation blockLocation : blockLocations) {
            StringBuilder hosts = new StringBuilder();
            hosts.append("{");
            for (String host : blockLocation.getHosts()) {
                hosts.append(host + " ");
            }
            hosts.append("}");
            this.resultOut.println(hosts);
        }
        actionStatus.setSuccessful(true);
    } catch (Exception e) {
        actionStatus.setSuccessful(false);
        throw new RuntimeException(e);
    } finally {
        actionStatus.end();
    }
}

Example 9 with ActionStatus

use of org.smartdata.actions.ActionStatus in project SSM by Intel-bigdata.

the class CacheFileAction method execute.

protected void execute() {
    ActionStatus actionStatus = getActionStatus();
    actionStatus.begin();
    try {
        addActionEvent(fileName);
        executeCacheAction(fileName);
        actionStatus.setSuccessful(true);
    } catch (Exception e) {
        actionStatus.setSuccessful(false);
        throw new RuntimeException(e);
    } finally {
        actionStatus.end();
    }
}

Example 10 with ActionStatus

use of org.smartdata.actions.ActionStatus in project SSM by Intel-bigdata.

the class ReadFileAction method execute.

@Override
protected void execute() {
    ActionStatus actionStatus = getActionStatus();
    actionStatus.begin();
    try {
        HdfsFileStatus fileStatus = dfsClient.getFileInfo(filePath);
        if (fileStatus == null) {
            resultOut.println("ReadFile Action fails, file doesn't exist!");
        }
        DFSInputStream dfsInputStream = dfsClient.open(filePath);
        byte[] buffer = new byte[bufferSize];
        // read from HDFS
        while (dfsInputStream.read(buffer, 0, bufferSize) != -1) {
        }
        dfsInputStream.close();
        actionStatus.setSuccessful(true);
    } catch (IOException e) {
        actionStatus.setSuccessful(false);
        resultOut.println("ReadFile Action fails!\n" + e.getMessage());
    } finally {
        actionStatus.end();
    }
}