List of usage examples for org.apache.hadoop.io DataOutputBuffer DataOutputBuffer
public DataOutputBuffer()
From source file:org.apache.mahout.matrix.MatrixTest.java
License:Apache License
public void testMatrixWritable() throws IOException { Matrix m = matrixFactory(new double[][] { { 1, 3, 4 }, { 5, 2, 3 }, { 1, 4, 2 } }); DataOutputBuffer out = new DataOutputBuffer(); m.write(out);//from w ww .j ava 2 s . c om out.close(); DataInputStream in = new DataInputStream(new ByteArrayInputStream(out.getData())); Matrix m2 = m.like(); m2.readFields(in); in.close(); assertEquals("row size", m.size()[ROW], m2.size()[ROW]); assertEquals("col size", m.size()[COL], m2.size()[COL]); }
From source file:org.apache.mahout.matrix.TestMatrixView.java
License:Apache License
public void testMatrixWritable() throws IOException { DataOutputBuffer out = new DataOutputBuffer(); test.write(out);/*from www . j ava2s . co m*/ out.close(); DataInputStream in = new DataInputStream(new ByteArrayInputStream(out.getData())); Matrix m2 = test.clone(); m2.readFields(in); in.close(); assertEquals("row size", test.size()[ROW], m2.size()[ROW]); assertEquals("col size", test.size()[COL], m2.size()[COL]); }
From source file:org.apache.mahout.matrix.TestVectorWritable.java
License:Apache License
private static void doTest(Vector writable) throws Exception { for (int i = 0; i < cardinality; i++) { writable.set(i, i);/*from w ww. j av a2s . com*/ } DataOutputBuffer out = new DataOutputBuffer(); writable.write(out); out.close(); DataInputStream in = new DataInputStream(new ByteArrayInputStream(out.getData())); writable.readFields(in); in.close(); assertEquals(cardinality, writable.size()); for (int i = 0; i < cardinality; i++) { assertEquals((double) i, writable.get(i)); } in = new DataInputStream(new ByteArrayInputStream(out.getData())); writable.readFields(in); in.close(); assertEquals(cardinality, writable.size()); for (int i = 0; i < cardinality; i++) { assertEquals((double) i, writable.get(i)); } }
From source file:org.apache.metron.maas.service.Client.java
License:Apache License
/** * Main run function for the client//from w w w.j av a 2 s . c o m * @return true if application completed successfully * @throws IOException * @throws YarnException */ public boolean run() throws IOException, YarnException { LOG.info("Running Client"); yarnClient.start(); YarnClusterMetrics clusterMetrics = yarnClient.getYarnClusterMetrics(); LOG.info("Got Cluster metric info from ASM" + ", numNodeManagers=" + clusterMetrics.getNumNodeManagers()); List<NodeReport> clusterNodeReports = yarnClient.getNodeReports(NodeState.RUNNING); LOG.info("Got Cluster node info from ASM"); for (NodeReport node : clusterNodeReports) { LOG.info("Got node report from ASM for" + ", nodeId=" + node.getNodeId() + ", nodeAddress" + node.getHttpAddress() + ", nodeRackName" + node.getRackName() + ", nodeNumContainers" + node.getNumContainers()); } QueueInfo queueInfo = yarnClient.getQueueInfo(this.amQueue); LOG.info("Queue info" + ", queueName=" + queueInfo.getQueueName() + ", queueCurrentCapacity=" + queueInfo.getCurrentCapacity() + ", queueMaxCapacity=" + queueInfo.getMaximumCapacity() + ", queueApplicationCount=" + queueInfo.getApplications().size() + ", queueChildQueueCount=" + queueInfo.getChildQueues().size()); List<QueueUserACLInfo> listAclInfo = yarnClient.getQueueAclsInfo(); for (QueueUserACLInfo aclInfo : listAclInfo) { for (QueueACL userAcl : aclInfo.getUserAcls()) { LOG.info("User ACL Info for Queue" + ", queueName=" + aclInfo.getQueueName() + ", userAcl=" + userAcl.name()); } } if (domainId != null && domainId.length() > 0 && toCreateDomain) { prepareTimelineDomain(); } // Get a new application id YarnClientApplication app = yarnClient.createApplication(); GetNewApplicationResponse appResponse = app.getNewApplicationResponse(); // TODO get min/max resource capabilities from RM and change memory ask if needed // If we do not have min/max, we may not be able to correctly request // the required resources from the RM for the app master // Memory ask has to be a multiple of min and less than max. // Dump out information about cluster capability as seen by the resource manager int maxMem = appResponse.getMaximumResourceCapability().getMemory(); LOG.info("Max mem capabililty of resources in this cluster " + maxMem); // A resource ask cannot exceed the max. if (amMemory > maxMem) { LOG.info("AM memory specified above max threshold of cluster. Using max value." + ", specified=" + amMemory + ", max=" + maxMem); amMemory = maxMem; } int maxVCores = appResponse.getMaximumResourceCapability().getVirtualCores(); LOG.info("Max virtual cores capabililty of resources in this cluster " + maxVCores); if (amVCores > maxVCores) { LOG.info("AM virtual cores specified above max threshold of cluster. " + "Using max value." + ", specified=" + amVCores + ", max=" + maxVCores); amVCores = maxVCores; } // set the application name ApplicationSubmissionContext appContext = app.getApplicationSubmissionContext(); ApplicationId appId = appContext.getApplicationId(); appContext.setKeepContainersAcrossApplicationAttempts(keepContainers); appContext.setApplicationName(appName); if (attemptFailuresValidityInterval >= 0) { appContext.setAttemptFailuresValidityInterval(attemptFailuresValidityInterval); } // set local resources for the application master // local files or archives as needed // In this scenario, the jar file for the application master is part of the local resources Map<String, LocalResource> localResources = new HashMap<String, LocalResource>(); LOG.info("Copy App Master jar from local filesystem and add to local environment"); // Copy the application master jar to the filesystem // Create a local resource to point to the destination jar path FileSystem fs = FileSystem.get(conf); Path ajPath = addToLocalResources(fs, appMasterJar, appMasterJarPath, appId.toString(), localResources, null); // Set the log4j properties if needed if (!log4jPropFile.isEmpty()) { addToLocalResources(fs, log4jPropFile, log4jPath, appId.toString(), localResources, null); } // Set the necessary security tokens as needed //amContainer.setContainerTokens(containerToken); // Set the env variables to be setup in the env where the application master will be run LOG.info("Set the environment for the application master"); Map<String, String> env = new HashMap<String, String>(); // put location of shell script into env // using the env info, the application master will create the correct local resource for the // eventual containers that will be launched to execute the shell scripts if (domainId != null && domainId.length() > 0) { env.put(Constants.TIMELINEDOMAIN, domainId); } // Add AppMaster.jar location to classpath // At some point we should not be required to add // the hadoop specific classpaths to the env. // It should be provided out of the box. // For now setting all required classpaths including // the classpath to "." for the application jar StringBuilder classPathEnv = new StringBuilder(Environment.CLASSPATH.$$()) .append(ApplicationConstants.CLASS_PATH_SEPARATOR).append("./*"); for (String c : conf.getStrings(YarnConfiguration.YARN_APPLICATION_CLASSPATH, YarnConfiguration.DEFAULT_YARN_CROSS_PLATFORM_APPLICATION_CLASSPATH)) { classPathEnv.append(ApplicationConstants.CLASS_PATH_SEPARATOR); classPathEnv.append(c.trim()); } classPathEnv.append(ApplicationConstants.CLASS_PATH_SEPARATOR).append("./log4j.properties"); // add the runtime classpath needed for tests to work if (conf.getBoolean(YarnConfiguration.IS_MINI_YARN_CLUSTER, false)) { classPathEnv.append(':'); classPathEnv.append(System.getProperty("java.class.path")); } env.put("CLASSPATH", classPathEnv.toString()); // Set the necessary command to execute the application master Vector<CharSequence> vargs = new Vector<CharSequence>(30); // Set java executable command LOG.info("Setting up app master command"); vargs.add(Environment.JAVA_HOME.$$() + "/bin/java"); // Set Xmx based on am memory size vargs.add("-Xmx" + amMemory + "m"); // Set class name vargs.add(appMasterMainClass); // Set params for Application Master vargs.add(ApplicationMaster.AMOptions.toArgs(ApplicationMaster.AMOptions.ZK_QUORUM.of(zkQuorum), ApplicationMaster.AMOptions.ZK_ROOT.of(zkRoot), ApplicationMaster.AMOptions.APP_JAR_PATH.of(ajPath.toString()))); if (null != nodeLabelExpression) { appContext.setNodeLabelExpression(nodeLabelExpression); } for (Map.Entry<String, String> entry : shellEnv.entrySet()) { vargs.add("--shell_env " + entry.getKey() + "=" + entry.getValue()); } vargs.add("1>" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/AppMaster.stdout"); vargs.add("2>" + ApplicationConstants.LOG_DIR_EXPANSION_VAR + "/AppMaster.stderr"); // Get final commmand StringBuilder command = new StringBuilder(); for (CharSequence str : vargs) { command.append(str).append(" "); } LOG.info("Completed setting up app master command " + command.toString()); List<String> commands = new ArrayList<String>(); commands.add(command.toString()); // Set up the container launch context for the application master ContainerLaunchContext amContainer = ContainerLaunchContext.newInstance(localResources, env, commands, null, null, null); // Set up resource type requirements // For now, both memory and vcores are supported, so we set memory and // vcores requirements Resource capability = Resource.newInstance(amMemory, amVCores); appContext.setResource(capability); // Service data is a binary blob that can be passed to the application // Not needed in this scenario // Setup security tokens if (UserGroupInformation.isSecurityEnabled()) { // Note: Credentials class is marked as LimitedPrivate for HDFS and MapReduce Credentials credentials = new Credentials(); String tokenRenewer = conf.get(YarnConfiguration.RM_PRINCIPAL); if (tokenRenewer == null || tokenRenewer.length() == 0) { throw new IOException("Can't get Master Kerberos principal for the RM to use as renewer"); } // For now, only getting tokens for the default file-system. final Token<?> tokens[] = fs.addDelegationTokens(tokenRenewer, credentials); if (tokens != null) { for (Token<?> token : tokens) { LOG.info("Got dt for " + fs.getUri() + "; " + token); } } DataOutputBuffer dob = new DataOutputBuffer(); credentials.writeTokenStorageToStream(dob); ByteBuffer fsTokens = ByteBuffer.wrap(dob.getData(), 0, dob.getLength()); amContainer.setTokens(fsTokens); } appContext.setAMContainerSpec(amContainer); // Set the priority for the application master Priority pri = Priority.newInstance(amPriority); appContext.setPriority(pri); // Set the queue to which this application is to be submitted in the RM appContext.setQueue(amQueue); // Submit the application to the applications manager // SubmitApplicationResponse submitResp = applicationsManager.submitApplication(appRequest); // Ignore the response as either a valid response object is returned on success // or an exception thrown to denote some form of a failure LOG.info("Submitting application to ASM"); yarnClient.submitApplication(appContext); // Monitor the application return monitorApplication(appId); }
From source file:org.apache.metron.maas.service.yarn.YarnUtils.java
License:Apache License
public ByteBuffer tokensFromCredentials(Credentials credentials) throws IOException { // Note: Credentials, Token, UserGroupInformation, DataOutputBuffer class // are marked as LimitedPrivate credentials = credentials == null ? UserGroupInformation.getCurrentUser().getCredentials() : credentials; DataOutputBuffer dob = new DataOutputBuffer(); credentials.writeTokenStorageToStream(dob); // Now remove the AM->RM token so that containers cannot access it. Iterator<Token<?>> iter = credentials.getAllTokens().iterator(); LOG.info("Executing with tokens:"); while (iter.hasNext()) { Token<?> token = iter.next(); LOG.info(token);/*from w ww. j av a2 s . c o m*/ if (token.getKind().equals(AMRMTokenIdentifier.KIND_NAME)) { iter.remove(); } } return ByteBuffer.wrap(dob.getData(), 0, dob.getLength()); }
From source file:org.apache.mrql.FlinkJsonSplitter.java
License:Apache License
FlinkJsonSplitter(String[] tags, FSDataInputStream fsin, long start, long end) { in_memory = false;//from w ww.j a v a 2 s. c o m this.tags = tags; this.fsin = fsin; this.end = end; this.buffer = new DataOutputBuffer(); try { fsin.seek(start); this.start = (start == 0) ? start : sync(start); fsin.seek(this.start); scanner = new JSONLex(fsin); } catch (IOException e) { System.err.println("*** Cannot parse the data split: " + fsin); } }
From source file:org.apache.mrql.FlinkJsonSplitter.java
License:Apache License
FlinkJsonSplitter(String[] tags, String file) { in_memory = true;//from w w w . j a v a2 s. c o m try { in = new FileInputStream(file); } catch (Exception e) { throw new Error("Cannot open the file: " + file); } ; this.tags = tags; this.buffer = new DataOutputBuffer(); scanner = new JSONLex(in); }
From source file:org.apache.mrql.FlinkXMLSplitter.java
License:Apache License
FlinkXMLSplitter(String[] tags, FSDataInputStream fsin, long start, long end) { in_memory = false;/* w w w . j a v a 2 s . co m*/ this.tags = tags; this.fsin = fsin; this.start = start; this.end = end; pos = start; this.buffer = new DataOutputBuffer(); try { fsin.seek(start); } catch (IOException e) { System.err.println("*** Cannot parse the data split: " + fsin); } }
From source file:org.apache.mrql.FlinkXMLSplitter.java
License:Apache License
FlinkXMLSplitter(String[] tags, String file) { in_memory = true;/*from w w w.j a va 2 s .c o m*/ try { in = new BufferedReader(new InputStreamReader(new FileInputStream(file)), 100000); } catch (Exception e) { throw new Error("Cannot open the file: " + file); } ; this.tags = tags; this.buffer = new DataOutputBuffer(); }
From source file:org.apache.mrql.JsonFormatParser.java
License:Apache License
public void open(String file) { try {/*from w w w. ja va 2s . c om*/ splitter = new JsonSplitter(tags, file, new DataOutputBuffer()); } catch (Exception e) { throw new Error(e); } }