@Override public Boolean apply(TaskInstance t) { return t.isAsyncTask(); } });
@Override public void run() { log.trace("Task {} need commit", task.taskName()); state.needCommit(); resume(); } }, commitMs, commitMs, TimeUnit.MILLISECONDS);
/** * Two {@link KafkaCheckpointLogKey}s are equal iff their grouperFactory class, taskName and type are equal. */ @Override public int hashCode() { int result = grouperFactoryClassName.hashCode(); result = 31 * result + taskName.hashCode(); result = 31 * result + type.hashCode(); return result; }
private final static TaskName getStandbyTaskName(TaskName activeTaskName, int replicaNum) { return new TaskName(STANDBY_TASKNAME_PREFIX.concat(TASKNAME_SEPARATOR) .concat(activeTaskName.getTaskName()) .concat(TASKNAME_SEPARATOR) .concat(String.valueOf(replicaNum))); } }
public static TaskName getTaskName(int partitionId) { return new TaskName("Partition " + partitionId); }
@Override public void serialize(TaskName taskName, JsonGenerator jsonGenerator, SerializerProvider provider) throws IOException, JsonProcessingException { jsonGenerator.writeObject(taskName.toString()); } }
@Test public void testClientResponseForHeartbeatAlive() throws IOException { client.setHttpOutput("{\"alive\": true}"); ContainerHeartbeatResponse response = client.requestHeartbeat(); Assert.assertTrue(response.isAlive()); }
@Test public void testClientResponseOnBadRequest() throws IOException { client.shouldThrowException(true); ContainerHeartbeatResponse response = client.requestHeartbeat(); Assert.assertFalse(response.isAlive()); }
private void populateContainerLocalityMappings() { Map<String, Map<String, String>> allMappings = localityManager.readContainerLocality(); for (String containerId: containers.keySet()) { if (allMappings.containsKey(containerId)) { localityMappings.put(containerId, allMappings.get(containerId).get(SetContainerHostMapping.HOST_KEY)); } else { localityMappings.put(containerId, null); } } }
SamzaContainerStatus getContainerStatus() { if (container != null) return container.getStatus(); return null; } }
@Test public void testClientResponseForHeartbeatDead() throws IOException { client.setHttpOutput("{\"alive\": false}"); ContainerHeartbeatResponse response = client.requestHeartbeat(); Assert.assertFalse(response.isAlive()); }
@Override public Boolean apply(TaskInstance t) { return t.isAsyncTask(); } });
@Override public Boolean apply(TaskInstance t) { return t.isAsyncTask(); } });
@Override public Boolean apply(TaskInstance t) { return t.isAsyncTask(); } });
@Override public Boolean apply(TaskInstance t) { return t.isAsyncTask(); } });