(ignite) branch master updated: IGNITE-26259 Use MessageSerializer for GridQueryCancelRequest (#12291)

[shishkovilja]

This is an automated email from the ASF dual-hosted git repository.

shishkovilja pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/ignite.git


The following commit(s) were added to refs/heads/master by this push:
     new c9bbb2f97e8 IGNITE-26259 Use MessageSerializer for 
GridQueryCancelRequest (#12291)
c9bbb2f97e8 is described below

commit c9bbb2f97e851971482ec547b92f09a6a6592f29
Author: Denis <116185428+denisp...@users.noreply.github.com>
AuthorDate: Fri Sep 12 20:17:33 2025 +1000

    IGNITE-26259 Use MessageSerializer for GridQueryCancelRequest (#12291)
---
 .../communication/GridIoMessageFactory.java        |  3 +-
 .../twostep/messages/GridQueryCancelRequest.java   | 53 ++++------------------
 2 files changed, 11 insertions(+), 45 deletions(-)

diff --git 
a/modules/core/src/main/java/org/apache/ignite/internal/managers/communication/GridIoMessageFactory.java
 
b/modules/core/src/main/java/org/apache/ignite/internal/managers/communication/GridIoMessageFactory.java
index c7dbc3a899c..b2873d83312 100644
--- 
a/modules/core/src/main/java/org/apache/ignite/internal/managers/communication/GridIoMessageFactory.java
+++ 
b/modules/core/src/main/java/org/apache/ignite/internal/managers/communication/GridIoMessageFactory.java
@@ -38,6 +38,7 @@ import 
org.apache.ignite.internal.codegen.GridDhtTxOnePhaseCommitAckRequestSeria
 import org.apache.ignite.internal.codegen.GridIntListSerializer;
 import org.apache.ignite.internal.codegen.GridJobCancelRequestSerializer;
 import org.apache.ignite.internal.codegen.GridJobSiblingsRequestSerializer;
+import org.apache.ignite.internal.codegen.GridQueryCancelRequestSerializer;
 import org.apache.ignite.internal.codegen.GridQueryFailResponseSerializer;
 import org.apache.ignite.internal.codegen.GridQueryKillRequestSerializer;
 import org.apache.ignite.internal.codegen.GridQueryKillResponseSerializer;
@@ -316,7 +317,7 @@ public class GridIoMessageFactory implements 
MessageFactoryProvider {
         factory.register((short)102, CacheVersionedValue::new, new 
CacheVersionedValueSerializer());
         factory.register((short)103, GridCacheRawVersionedEntry::new);
         factory.register((short)104, GridCacheVersionEx::new, new 
GridCacheVersionExSerializer());
-        factory.register((short)106, GridQueryCancelRequest::new);
+        factory.register((short)106, GridQueryCancelRequest::new, new 
GridQueryCancelRequestSerializer());
         factory.register((short)107, GridQueryFailResponse::new, new 
GridQueryFailResponseSerializer());
         factory.register((short)108, GridQueryNextPageRequest::new, new 
GridQueryNextPageRequestSerializer());
         factory.register((short)109, GridQueryNextPageResponse::new, new 
GridQueryNextPageResponseSerializer());
diff --git 
a/modules/core/src/main/java/org/apache/ignite/internal/processors/query/h2/twostep/messages/GridQueryCancelRequest.java
 
b/modules/core/src/main/java/org/apache/ignite/internal/processors/query/h2/twostep/messages/GridQueryCancelRequest.java
index 203b4eaf4fa..52ea52119de 100644
--- 
a/modules/core/src/main/java/org/apache/ignite/internal/processors/query/h2/twostep/messages/GridQueryCancelRequest.java
+++ 
b/modules/core/src/main/java/org/apache/ignite/internal/processors/query/h2/twostep/messages/GridQueryCancelRequest.java
@@ -17,17 +17,16 @@
 
 package org.apache.ignite.internal.processors.query.h2.twostep.messages;
 
-import java.nio.ByteBuffer;
+import org.apache.ignite.internal.Order;
 import org.apache.ignite.internal.util.typedef.internal.S;
 import org.apache.ignite.plugin.extensions.communication.Message;
-import org.apache.ignite.plugin.extensions.communication.MessageReader;
-import org.apache.ignite.plugin.extensions.communication.MessageWriter;
 
 /**
  * Cancel map part of query request.
  */
 public class GridQueryCancelRequest implements Message {
     /** */
+    @Order(value = 0, method = "queryRequestId")
     private long qryReqId;
 
     /**
@@ -51,50 +50,16 @@ public class GridQueryCancelRequest implements Message {
         return qryReqId;
     }
 
-    /** {@inheritDoc} */
-    @Override public String toString() {
-        return S.toString(GridQueryCancelRequest.class, this);
-    }
-
-    /** {@inheritDoc} */
-    @Override public boolean writeTo(ByteBuffer buf, MessageWriter writer) {
-        writer.setBuffer(buf);
-
-        if (!writer.isHeaderWritten()) {
-            if (!writer.writeHeader(directType()))
-                return false;
-
-            writer.onHeaderWritten();
-        }
-
-        switch (writer.state()) {
-            case 0:
-                if (!writer.writeLong(qryReqId))
-                    return false;
-
-                writer.incrementState();
-
-        }
-
-        return true;
+    /**
+     * @param qryReqId Query request ID.
+     */
+    public void queryRequestId(long qryReqId) {
+        this.qryReqId = qryReqId;
     }
 
     /** {@inheritDoc} */
-    @Override public boolean readFrom(ByteBuffer buf, MessageReader reader) {
-        reader.setBuffer(buf);
-
-        switch (reader.state()) {
-            case 0:
-                qryReqId = reader.readLong();
-
-                if (!reader.isLastRead())
-                    return false;
-
-                reader.incrementState();
-
-        }
-
-        return true;
+    @Override public String toString() {
+        return S.toString(GridQueryCancelRequest.class, this);
     }
 
     /** {@inheritDoc} */