Skip to content

Commit bd27205

Browse files
authored
IGNITE-27226 Use MessageSerializer for QueryStartRequest (#12548)
1 parent ae0028d commit bd27205

File tree

3 files changed

+89
-201
lines changed

3 files changed

+89
-201
lines changed

modules/calcite/src/main/java/org/apache/ignite/internal/processors/query/calcite/exec/ExecutionServiceImpl.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -882,7 +882,7 @@ private void onMessage(UUID nodeId, final QueryStartRequest msg) {
882882
);
883883

884884
final BaseQueryContext qctx = createQueryContext(
885-
msg.appAttrs() == null ? Contexts.empty() : Contexts.of(new SessionContextImpl(msg.appAttrs())),
885+
msg.applicationAttributes() == null ? Contexts.empty() : Contexts.of(new SessionContextImpl(msg.applicationAttributes())),
886886
msg.schema());
887887

888888
QueryPlan qryPlan = queryPlanCache().queryPlan(
@@ -903,7 +903,7 @@ private void onMessage(UUID nodeId, final QueryStartRequest msg) {
903903
msg.fragmentDescription(),
904904
handler,
905905
qry.createMemoryTracker(memoryTracker, cfg.getQueryMemoryQuota()),
906-
createIoTracker(nodeId, msg.originatingQryId()),
906+
createIoTracker(nodeId, msg.originatingQueryId()),
907907
msg.timeout(),
908908
Commons.parametersMap(msg.parameters()),
909909
msg.queryTransactionEntries()

modules/calcite/src/main/java/org/apache/ignite/internal/processors/query/calcite/message/MessageType.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import org.apache.ignite.internal.codegen.QueryBatchAcknowledgeMessageSerializer;
2828
import org.apache.ignite.internal.codegen.QueryBatchMessageSerializer;
2929
import org.apache.ignite.internal.codegen.QueryCloseMessageSerializer;
30+
import org.apache.ignite.internal.codegen.QueryStartRequestSerializer;
3031
import org.apache.ignite.internal.codegen.QueryStartResponseSerializer;
3132
import org.apache.ignite.internal.codegen.QueryTxEntrySerializer;
3233
import org.apache.ignite.internal.processors.query.calcite.metadata.ColocationGroup;
@@ -39,7 +40,7 @@
3940
*/
4041
public enum MessageType {
4142
/** */
42-
QUERY_START_REQUEST(300, QueryStartRequest::new),
43+
QUERY_START_REQUEST(300, QueryStartRequest::new, new QueryStartRequestSerializer()),
4344

4445
/** */
4546
QUERY_START_RESPONSE(301, QueryStartResponse::new, new QueryStartResponseSerializer()),

0 commit comments

Comments
 (0)