private List<String> getFileName(TaskInfo taskInfo) { Labels labels = taskInfo.getLabels(); for (Label label : labels.getLabelsList()) { //Aurora client driver uses keyname - "org.apache.aurora.metadata.fileName".. if (label.getKey().endsWith("fileName")) { List<String> files = Arrays.asList(label.getValue().split(FILE_DELIMITER)); for (int i = 0; i < files.size(); i++) files.set(i, files.get(i).trim()); return files; } } log.warn("error reading fileName from taskInfo"); return null; }
private Set<UUID> getTaskConfigs() { final Collection<Protos.TaskInfo> taskInfos = stateStore.fetchTasks(); final Set<UUID> activeConfigs = new HashSet<>(); try { for (Protos.TaskInfo taskInfo : taskInfos) { final Protos.Labels labels = taskInfo.getLabels(); for(Protos.Label label : labels.getLabelsList()) { if (label.getKey().equals(PersistentOfferRequirementProvider.CONFIG_TARGET_KEY)) { activeConfigs.add(UUID.fromString(label.getValue())); } } } } catch (Exception e) { LOGGER.error("Failed to fetch configurations from taskInfos.", e); } return activeConfigs; }
for (Protos.Label l : taskInfo.getLabels().getLabelsList()) { taskIdLabel.put(l.getKey(), l.getValue());
public CassandraDaemonTask move( CassandraDaemonTask currentDaemon, CassandraTaskExecutor executor) { final List<Protos.Label> labelsList = currentDaemon.getTaskInfo().getLabels().getLabelsList(); String configName = ""; for (Protos.Label label : labelsList) { if ("config_target".equals(label.getKey())) { configName = label.getValue(); } } if (StringUtils.isBlank(configName)) { throw new IllegalStateException("Task should have 'config_target'"); } CassandraDaemonTask replacementDaemon = new CassandraDaemonTask( currentDaemon.getName(), configName, executor, currentDaemon.getConfig(), capabilities, currentDaemon.getData().replacing(currentDaemon.getData().getHostname())); Protos.TaskInfo taskInfo = Protos.TaskInfo.newBuilder(replacementDaemon.getTaskInfo()) .setTaskId(currentDaemon.getTaskInfo().getTaskId()) .build(); return new CassandraDaemonTask(taskInfo); }
private static Protos.TaskInfo updateConfigLabel(String configName, Protos.TaskInfo taskInfo) { final Protos.Labels.Builder labelsBuilder = Protos.Labels.newBuilder(); final Protos.Labels labels = taskInfo.getLabels(); for (Protos.Label label : labels.getLabelsList()) { final String key = label.getKey(); if (!CONFIG_TARGET_KEY.equals(key)) { labelsBuilder.addLabels(label); } } labelsBuilder.addLabels(Protos.Label.newBuilder() .setKey(CONFIG_TARGET_KEY) .setValue(configName)); return Protos.TaskInfo.newBuilder(taskInfo) .clearLabels() .setLabels(labelsBuilder.build()) .build(); }
private Optional<String> getTaskConfig(CassandraDaemonTask task) { final Protos.TaskInfo taskInfo = task.getTaskInfo(); if (!taskInfo.hasLabels() || CollectionUtils.isEmpty(taskInfo.getLabels().getLabelsList())) { return Optional.empty(); } for (Protos.Label label : taskInfo.getLabels().getLabelsList()) { final String key = label.getKey(); if (PersistentOfferRequirementProvider.CONFIG_TARGET_KEY.equals(key)) { return Optional.ofNullable(label.getValue()); } } return Optional.empty(); }
private static String getConfigName(Protos.TaskInfo taskInfo) { for (Protos.Label label : taskInfo.getLabels().getLabelsList()) { if (label.getKey().equals("config_target")) { return label.getValue(); } } return null; }
/** * @see LabelReader#LabelReader(String, Labels) */ @SuppressWarnings("checkstyle:MultipleStringLiterals") public TaskLabelReader(TaskInfo taskInfo) { reader = new LabelReader(String.format("Task %s", taskInfo.getName()), taskInfo.getLabels()); }
/** * @see LabelWriter#LabelWriter(Labels) */ public TaskLabelWriter(TaskInfo taskInfo) { writer = new LabelWriter(taskInfo.getLabels()); }