@Nonnull public static ZNRecord toZnRecord(@Nonnull TableConfig tableConfig) { ZNRecord znRecord = new ZNRecord(tableConfig.getTableName()); Map<String, String> simpleFields = new HashMap<>(); simpleFields.put(TABLE_NAME_KEY, tableConfig._tableName); simpleFields.put(TABLE_TYPE_KEY, tableConfig._tableType.toString()); try { simpleFields.put(VALIDATION_CONFIG_KEY, JsonUtils.objectToString(tableConfig._validationConfig)); simpleFields.put(TENANT_CONFIG_KEY, JsonUtils.objectToString(tableConfig._tenantConfig)); simpleFields.put(INDEXING_CONFIG_KEY, JsonUtils.objectToString(tableConfig._indexingConfig)); simpleFields.put(CUSTOM_CONFIG_KEY, JsonUtils.objectToString(tableConfig._customConfig)); if (tableConfig._quotaConfig != null) { simpleFields.put(QUOTA_CONFIG_KEY, JsonUtils.objectToString(tableConfig._quotaConfig)); } if (tableConfig._taskConfig != null) { simpleFields.put(TASK_CONFIG_KEY, JsonUtils.objectToString(tableConfig._taskConfig)); } if (tableConfig._routingConfig != null) { simpleFields.put(ROUTING_CONFIG_KEY, JsonUtils.objectToString(tableConfig._routingConfig)); } } catch (IOException e) { throw new RuntimeException(e); } znRecord.setSimpleFields(simpleFields); return znRecord; }
/** * Get relay message attached for the given instance. * * @param instance * @return null if no message for the instance */ public Message getRelayMessage(String instance) { Map<String, String> messageInfo = _record.getMapField(instance); if (messageInfo != null) { String id = messageInfo.get(Attributes.RELAY_MSG_ID.name()); if (id == null) { id = messageInfo.get(Attributes.MSG_ID.name()); if (id == null) { return null; } } ZNRecord record = new ZNRecord(id); record.setSimpleFields(messageInfo); return new Message(record); } return null; }
/** * Get relay message attached for the given instance. * * @param instance * @return null if no message for the instance */ public Message getRelayMessage(String instance) { Map<String, String> messageInfo = _record.getMapField(instance); if (messageInfo != null) { String id = messageInfo.get(Attributes.RELAY_MSG_ID.name()); if (id == null) { id = messageInfo.get(Attributes.MSG_ID.name()); if (id == null) { return null; } } ZNRecord record = new ZNRecord(id); record.setSimpleFields(messageInfo); return new Message(record); } return null; }
metaRecord.setSimpleFields(value.getRecord().getSimpleFields()); records.add(metaRecord);
metaRecord.setSimpleFields(value.getRecord().getSimpleFields()); records.add(metaRecord);
bucketizedRecord.setSimpleFields(record.getSimpleFields());
bucketizedRecord.setSimpleFields(record.getSimpleFields());
private void setNodes(ZkHelixPropertyStore<ZNRecord> store, char c, boolean needTimestamp) { char[] data = new char[bufSize]; for (int i = 0; i < bufSize; i++) { data[i] = c; } Map<String, String> map = new TreeMap<String, String>(); for (int i = 0; i < mapNr; i++) { map.put("key_" + i, new String(data)); } for (int i = 0; i < firstLevelNr; i++) { for (int j = 0; j < secondLevelNr; j++) { String nodeId = getNodeId(i, j); ZNRecord record = new ZNRecord(nodeId); record.setSimpleFields(map); if (needTimestamp) { long now = System.currentTimeMillis(); record.setSimpleField("SetTimestamp", Long.toString(now)); } String key = getSecondLevelKey(i, j); store.set(key, record, AccessOption.PERSISTENT); } } }
ZNRecord record = new ZNRecord(provider.getReportName()); if (report != null) { record.setSimpleFields(report);
ZNRecord record = new ZNRecord(provider.getReportName()); if (report != null) { record.setSimpleFields(report);
private IdealState generateNewIdealState(String resourceName, IdealState currentIdealState, ZNRecord newMapping) { IdealState newIdealState = new IdealState(resourceName); newIdealState.getRecord().setSimpleFields(currentIdealState.getRecord().getSimpleFields()); newIdealState.setRebalanceMode(currentIdealState.getRebalanceMode()); newIdealState.getRecord().setListFields(newMapping.getListFields()); return newIdealState; }
private IdealState generateNewIdealState(String resourceName, IdealState currentIdealState, ZNRecord newMapping) { IdealState newIdealState = new IdealState(resourceName); newIdealState.getRecord().setSimpleFields(currentIdealState.getRecord().getSimpleFields()); newIdealState.setRebalanceMode(currentIdealState.getRebalanceMode()); newIdealState.getRecord().setListFields(newMapping.getListFields()); return newIdealState; }
metaRecord.setSimpleFields(value.getRecord().getSimpleFields()); success = _baseDataAccessor.set(path, metaRecord, options); if (success) {
@Override public ZNRecord update(ZNRecord currentData) { CurrentState curState = null; if (currentData == null) { curState = new CurrentState(_lastCurState.getId()); // copy all simple fields settings and overwrite session-id to current session curState.getRecord().setSimpleFields(_lastCurState.getRecord().getSimpleFields()); curState.setSessionId(_curSessionId); } else { curState = new CurrentState(currentData); } for (String partitionName : _lastCurState.getPartitionStateMap().keySet()) { // carry-over only when current-state not exist if (curState.getState(partitionName) == null) { curState.setState(partitionName, _initState); } } return curState.getRecord(); }
metaRecord.setSimpleFields(value.getRecord().getSimpleFields()); success = _baseDataAccessor.set(path, metaRecord, options); if (success) {
@Override public ZNRecord update(ZNRecord currentData) { CurrentState curState = null; if (currentData == null) { curState = new CurrentState(_lastCurState.getId()); // copy all simple fields settings and overwrite session-id to current session curState.getRecord().setSimpleFields(_lastCurState.getRecord().getSimpleFields()); curState.setSessionId(_curSessionId); } else { curState = new CurrentState(currentData); } for (String partitionName : _lastCurState.getPartitionStateMap().keySet()) { // carry-over only when current-state not exist if (curState.getState(partitionName) == null) { curState.setState(partitionName, _initState); } } return curState.getRecord(); }
private void setNodes(HelixZkClient zkClient, String root, char c, boolean needTimestamp) { char[] data = new char[bufSize]; for (int i = 0; i < bufSize; i++) { data[i] = c; } Map<String, String> map = new TreeMap<String, String>(); for (int i = 0; i < mapNr; i++) { map.put("key_" + i, new String(data)); } for (int i = 0; i < firstLevelNr; i++) { String firstLevelKey = getFirstLevelKey(i); for (int j = 0; j < secondLevelNr; j++) { String nodeId = getNodeId(i, j); ZNRecord record = new ZNRecord(nodeId); record.setSimpleFields(map); if (needTimestamp) { long now = System.currentTimeMillis(); record.setSimpleField("SetTimestamp", Long.toString(now)); } String key = getSecondLevelKey(i, j); try { zkClient.writeData(root + key, record); } catch (ZkNoNodeException e) { zkClient.createPersistent(root + firstLevelKey, true); zkClient.createPersistent(root + key, record); } } } }
metaRecord.setSimpleFields(lastCurState.getRecord().getSimpleFields()); DataUpdater<ZNRecord> metaRecordUpdater = new CurStateCarryOverUpdater(_sessionId, initState, new CurrentState(metaRecord));
StringRepresentation getConfigs(ConfigScopeProperty scopeProperty, String... keys) throws Exception { StringRepresentation representation = null; ZkClient zkClient = ResourceUtil.getAttributeFromCtx(getContext(), ResourceUtil.ContextKey.ZKCLIENT); ClusterSetup setupTool = new ClusterSetup(zkClient); HelixAdmin admin = setupTool.getClusterManagementTool(); ZNRecord record = new ZNRecord(scopeProperty + " Config"); HelixConfigScope scope = new HelixConfigScopeBuilder(scopeProperty, keys).build(); List<String> configKeys = admin.getConfigKeys(scope); Map<String, String> configs = admin.getConfig(scope, configKeys); record.setSimpleFields(configs); representation = new StringRepresentation(ClusterRepresentationUtil.ZNRecordToJson(record), MediaType.APPLICATION_JSON); return representation; }
newIdealState.getRecord().setSimpleFields(idealState.getRecord().getSimpleFields()); newIdealState.setPartitionState("0", "localhost_12918", "ONLINE"); newIdealState.setPartitionState("0", "localhost_12919", "ONLINE");