public static KeyValueSetProto convertFromMap(Map<String, String> map) { return new KeyValueSet(map).getProto(); }
public static KeyValueSetProto convertFromMap(Map<String, String> map) { return new KeyValueSet(map).getProto(); }
public TableProto getProto() { TableProto.Builder builder = TableProto.newBuilder(); builder.setDataFormat(dataFormat); builder.setParams(propertySet.getProto()); return builder.build(); }
public TableProto getProto() { TableProto.Builder builder = TableProto.newBuilder(); builder.setDataFormat(dataFormat); builder.setParams(options.getProto()); return builder.build(); }
private static PlanProto.PersistentStoreNode.Builder buildPersistentStoreBuilder(PersistentStoreNode node, int [] childIds) { PlanProto.PersistentStoreNode.Builder persistentStoreBuilder = PlanProto.PersistentStoreNode.newBuilder(); if (childIds.length > 0) { // Simple create table may not have any children. This should be improved at TAJO-1589. persistentStoreBuilder.setChildSeq(childIds[0]); } persistentStoreBuilder.setStorageType(node.getStorageType()); if (node.hasOptions()) { persistentStoreBuilder.setTableProperties(node.getOptions().getProto()); } return persistentStoreBuilder; }
private static PlanProto.PersistentStoreNode.Builder buildPersistentStoreBuilder(PersistentStoreNode node, int [] childIds) { PlanProto.PersistentStoreNode.Builder persistentStoreBuilder = PlanProto.PersistentStoreNode.newBuilder(); if (childIds.length > 0) { // Simple create table may not have any children. This should be improved at TAJO-1589. persistentStoreBuilder.setChildSeq(childIds[0]); } persistentStoreBuilder.setStorageType(node.getStorageType()); if (node.hasOptions()) { persistentStoreBuilder.setTableProperties(node.getOptions().getProto()); } return persistentStoreBuilder; }
public Map<String, String> updateSessionVariables(final Map<String, String> variables) { NettyClientBase client = getTajoMasterConnection(); checkSessionAndGet(client); BlockingInterface tajoMasterService = client.getStub(); KeyValueSet keyValueSet = new KeyValueSet(); keyValueSet.putAll(variables); UpdateSessionVariableRequest request = UpdateSessionVariableRequest.newBuilder() .setSessionId(sessionId) .setSessionVars(keyValueSet.getProto()).build(); SessionUpdateResponse response; try { response = tajoMasterService.updateSessionVariables(null, request); } catch (ServiceException e) { throw new RuntimeException(e); } ensureOk(response.getState()); updateSessionVarsCache(ProtoUtil.convertToMap(response.getSessionVars())); return Collections.unmodifiableMap(sessionVarsCache); }
public Map<String, String> updateSessionVariables(final Map<String, String> variables) { NettyClientBase client = getTajoMasterConnection(); checkSessionAndGet(client); BlockingInterface tajoMasterService = client.getStub(); KeyValueSet keyValueSet = new KeyValueSet(); keyValueSet.putAll(variables); UpdateSessionVariableRequest request = UpdateSessionVariableRequest.newBuilder() .setSessionId(sessionId) .setSessionVars(keyValueSet.getProto()).build(); SessionUpdateResponse response; try { response = tajoMasterService.updateSessionVariables(null, request); } catch (ServiceException e) { throw new RuntimeException(e); } ensureOk(response.getState()); updateSessionVarsCache(ProtoUtil.convertToMap(response.getSessionVars())); return Collections.unmodifiableMap(sessionVarsCache); }
@Override public SessionProto getProto() { SessionProto.Builder builder = SessionProto.newBuilder(); builder.setSessionId(getSessionId()); builder.setUsername(getUserName()); builder.setCurrentDatabase(getCurrentDatabase()); builder.setLastAccessTime(lastAccessTime); KeyValueSet variables = new KeyValueSet(); synchronized (sessionVariables) { variables.putAll(this.sessionVariables); builder.setVariables(variables.getProto()); return builder.build(); } }
@Override public SessionProto getProto() { SessionProto.Builder builder = SessionProto.newBuilder(); builder.setSessionId(getSessionId()); builder.setUsername(getUserName()); builder.setCurrentDatabase(getCurrentDatabase()); builder.setLastAccessTime(lastAccessTime); KeyValueSet variables = new KeyValueSet(); synchronized (sessionVariables) { variables.putAll(this.sessionVariables); builder.setVariables(variables.getProto()); return builder.build(); } }
UpdateSessionVariableRequest request = UpdateSessionVariableRequest.newBuilder() .setSessionId(sessionId) .setSessionVars(keyValueSet.getProto()).build();
UpdateSessionVariableRequest request = UpdateSessionVariableRequest.newBuilder() .setSessionId(sessionId) .setSessionVars(keyValueSet.getProto()).build();
@Override public KeyValueSetResponse getAllSessionVariables(RpcController controller, TajoIdProtos.SessionIdProto request) throws ServiceException { try { String sessionId = request.getId(); KeyValueSet keyValueSet = new KeyValueSet(); keyValueSet.putAll(context.getSessionManager().getAllVariables(sessionId)); return KeyValueSetResponse.newBuilder() .setState(OK) .setValue(keyValueSet.getProto()) .build(); } catch (Throwable t) { printStackTraceIfError(LOG, t); return KeyValueSetResponse.newBuilder() .setState(returnError(t)) .build(); } }
@Override public KeyValueSetResponse getAllSessionVariables(RpcController controller, TajoIdProtos.SessionIdProto request) throws ServiceException { try { String sessionId = request.getId(); KeyValueSet keyValueSet = new KeyValueSet(); keyValueSet.putAll(context.getSessionManager().getAllVariables(sessionId)); return KeyValueSetResponse.newBuilder() .setState(OK) .setValue(keyValueSet.getProto()) .build(); } catch (Throwable t) { printStackTraceIfError(LOG, t); return KeyValueSetResponse.newBuilder() .setState(returnError(t)) .build(); } }
@Override public LogicalNode visitCreateIndex(SerializeContext context, LogicalPlan plan, LogicalPlan.QueryBlock block, CreateIndexNode node, Stack<LogicalNode> stack) throws TajoException { super.visitCreateIndex(context, plan, block, node, new Stack<>()); PlanProto.CreateIndexNode.Builder createIndexBuilder = PlanProto.CreateIndexNode.newBuilder(); int [] childIds = registerGetChildIds(context, node); createIndexBuilder.setChildSeq(childIds[0]); createIndexBuilder.setIndexName(node.getIndexName()); createIndexBuilder.setIndexMethod(node.getIndexMethod()); createIndexBuilder.setIndexPath(node.getIndexPath().toString()); for (SortSpec sortSpec : node.getKeySortSpecs()) { createIndexBuilder.addKeySortSpecs(sortSpec.getProto()); } createIndexBuilder.setTargetRelationSchema(node.getTargetRelationSchema().getProto()); createIndexBuilder.setIsUnique(node.isUnique()); createIndexBuilder.setIsClustered(node.isClustered()); if (node.hasOptions()) { createIndexBuilder.setIndexProperties(node.getOptions().getProto()); } createIndexBuilder.setIsExternal(node.isExternal()); PlanProto.LogicalNode.Builder nodeBuilder = createNodeBuilder(context, node); nodeBuilder.setCreateIndex(createIndexBuilder); context.treeBuilder.addNodes(nodeBuilder); return node; }
@Override public LogicalNode visitCreateIndex(SerializeContext context, LogicalPlan plan, LogicalPlan.QueryBlock block, CreateIndexNode node, Stack<LogicalNode> stack) throws TajoException { super.visitCreateIndex(context, plan, block, node, new Stack<LogicalNode>()); PlanProto.CreateIndexNode.Builder createIndexBuilder = PlanProto.CreateIndexNode.newBuilder(); int [] childIds = registerGetChildIds(context, node); createIndexBuilder.setChildSeq(childIds[0]); createIndexBuilder.setIndexName(node.getIndexName()); createIndexBuilder.setIndexMethod(node.getIndexMethod()); createIndexBuilder.setIndexPath(node.getIndexPath().toString()); for (SortSpec sortSpec : node.getKeySortSpecs()) { createIndexBuilder.addKeySortSpecs(sortSpec.getProto()); } createIndexBuilder.setTargetRelationSchema(node.getTargetRelationSchema().getProto()); createIndexBuilder.setIsUnique(node.isUnique()); createIndexBuilder.setIsClustered(node.isClustered()); if (node.hasOptions()) { createIndexBuilder.setIndexProperties(node.getOptions().getProto()); } createIndexBuilder.setIsExternal(node.isExternal()); PlanProto.LogicalNode.Builder nodeBuilder = createNodeBuilder(context, node); nodeBuilder.setCreateIndex(createIndexBuilder); context.treeBuilder.addNodes(nodeBuilder); return node; }
builder.setParams(properties.getProto());
@Override public SessionUpdateResponse updateSessionVariables(RpcController controller, UpdateSessionVariableRequest request) throws ServiceException { SessionUpdateResponse.Builder builder = SessionUpdateResponse.newBuilder(); try { String sessionId = request.getSessionId().getId(); for (KeyValueProto kv : request.getSessionVars().getKeyvalList()) { context.getSessionManager().setVariable(sessionId, kv.getKey(), kv.getValue()); } for (String unsetVariable : request.getUnsetVariablesList()) { context.getSessionManager().removeVariable(sessionId, unsetVariable); } builder.setState(OK); builder.setSessionVars(new KeyValueSet(context.getSessionManager().getAllVariables(sessionId)).getProto()); return builder.build(); } catch (Throwable t) { printStackTraceIfError(LOG, t); builder.setState(returnError(t)); return builder.build(); } }
@Override public SessionUpdateResponse updateSessionVariables(RpcController controller, UpdateSessionVariableRequest request) throws ServiceException { SessionUpdateResponse.Builder builder = SessionUpdateResponse.newBuilder(); try { String sessionId = request.getSessionId().getId(); for (KeyValueProto kv : request.getSessionVars().getKeyvalList()) { context.getSessionManager().setVariable(sessionId, kv.getKey(), kv.getValue()); } for (String unsetVariable : request.getUnsetVariablesList()) { context.getSessionManager().removeVariable(sessionId, unsetVariable); } builder.setState(OK); builder.setSessionVars(new KeyValueSet(context.getSessionManager().getAllVariables(sessionId)).getProto()); return builder.build(); } catch (Throwable t) { printStackTraceIfError(LOG, t); builder.setState(returnError(t)); return builder.build(); } }
builder.setParams(properties.getProto());