From 2ef5eb6cd080dafd63041e6c83af12a5906b204e Mon Sep 17 00:00:00 2001
From: redkale <redkale@qq.com>
Date: Thu, 2 Feb 2023 15:02:43 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96client?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../org/redkale/net/client/ClientCodec.java   |  40 ++++--
 .../redkale/net/client/ClientResponse.java    |  17 ++-
 .../org/redkale/net/sncp/SncpClientCodec.java |  54 ++++++--
 .../redkale/net/sncp/SncpClientResult.java    | 129 ++++--------------
 .../java/org/redkale/net/sncp/SncpHeader.java |  28 ++--
 5 files changed, 124 insertions(+), 144 deletions(-)

diff --git a/src/main/java/org/redkale/net/client/ClientCodec.java b/src/main/java/org/redkale/net/client/ClientCodec.java
index 26b2f48db..1bccf8ade 100644
--- a/src/main/java/org/redkale/net/client/ClientCodec.java
+++ b/src/main/java/org/redkale/net/client/ClientCodec.java
@@ -40,8 +40,7 @@ public abstract class ClientCodec<R extends ClientRequest, P> implements Complet
         this.connection = connection;
     }
 
-    //返回true: array会clear, 返回false: buffer会clear
-    public abstract boolean decodeMessages(ByteBuffer buffer, ByteArray array);
+    public abstract void decodeMessages(ByteBuffer buffer, ByteArray array);
 
     @Override
     public final void completed(Integer count, ByteBuffer attachment) {
@@ -63,15 +62,20 @@ public abstract class ClientCodec<R extends ClientRequest, P> implements Complet
     private void decodeResponse(ByteBuffer buffer) {
         AsyncConnection channel = connection.channel;
         connection.currRespIterator = null;
-        if (decodeMessages(buffer, readArray)) { //成功了
+        decodeMessages(buffer, readArray);
+        if (!respResults.isEmpty()) { //存在解析结果
             connection.currRespIterator = null;
             readArray.clear();
             for (ClientResponse<R, P> cr : respResults) {
-                ClientFuture<R, P> respFuture = connection.pollRespFuture(cr.getRequestid());
-                if (respFuture != null) {
-                    responseComplete(respFuture, cr.message, cr.exc);
+                if (cr.isError()) {
+                    connection.dispose(null);
+                } else {
+                    ClientFuture<R, P> respFuture = connection.pollRespFuture(cr.getRequestid());
+                    if (respFuture != null) {
+                        responseComplete(respFuture, cr.message, cr.exc);
+                    }
+                    respPool.accept(cr);
                 }
-                respPool.accept(cr);
             }
             respResults.clear();
 
@@ -95,7 +99,7 @@ public abstract class ClientCodec<R extends ClientRequest, P> implements Complet
             R request = respFuture.request;
             WorkThread workThread = null;
             try {
-                if (!request.isCompleted()) {
+                if (request != null && !request.isCompleted()) {
                     if (exc == null) {
                         connection.sendHalfWrite(exc);
                         //request没有发送完，respFuture需要再次接收
@@ -104,8 +108,10 @@ public abstract class ClientCodec<R extends ClientRequest, P> implements Complet
                         connection.sendHalfWrite(exc);
                     }
                 }
-                workThread = request.workThread;
-                request.workThread = null;
+                if (request != null) {
+                    workThread = request.workThread;
+                    request.workThread = null;
+                }
                 connection.respWaitingCounter.decrement();
                 if (connection.isAuthenticated()) {
                     connection.client.incrRespDoneCounter();
@@ -119,13 +125,17 @@ public abstract class ClientCodec<R extends ClientRequest, P> implements Complet
                 }
                 if (exc != null) {
                     workThread.runWork(() -> {
-                        Traces.currTraceid(request.traceid);
+                        if (request != null) {
+                            Traces.currTraceid(request.traceid);
+                        }
                         respFuture.completeExceptionally(exc);
                     });
                 } else {
-                    final Object rs = request.respTransfer == null ? message : request.respTransfer.apply(message);
+                    final Object rs = request == null || request.respTransfer == null ? message : request.respTransfer.apply(message);
                     workThread.runWork(() -> {
-                        Traces.currTraceid(request.traceid);
+                        if (request != null) {
+                            Traces.currTraceid(request.traceid);
+                        }
                         ((ClientFuture) respFuture).complete(rs);
                     });
                 }
@@ -169,6 +179,10 @@ public abstract class ClientCodec<R extends ClientRequest, P> implements Complet
         this.respResults.add(respPool.get().set(request, exc));
     }
 
+    public void occurError(R request, Throwable exc) {
+        this.respResults.add(new ClientResponse.ClientErrorResponse<>(request, exc));
+    }
+
     @Override
     public String toString() {
         return JsonConvert.root().convertTo(this);
diff --git a/src/main/java/org/redkale/net/client/ClientResponse.java b/src/main/java/org/redkale/net/client/ClientResponse.java
index 829ef7a69..53cceeb08 100644
--- a/src/main/java/org/redkale/net/client/ClientResponse.java
+++ b/src/main/java/org/redkale/net/client/ClientResponse.java
@@ -21,7 +21,7 @@ import java.io.Serializable;
  */
 public class ClientResponse<R extends ClientRequest, P> {
 
-    protected R request;
+    protected R request; //服务端返回一个不存在的requestid，可能为null
 
     protected P message;
 
@@ -101,4 +101,19 @@ public class ClientResponse<R extends ClientRequest, P> {
         return "{\"message\":" + message + "}";
     }
 
+    boolean isError() {
+        return false;
+    }
+
+    static class ClientErrorResponse<R extends ClientRequest, P> extends ClientResponse<R, P> {
+
+        public ClientErrorResponse(R request, Throwable exc) {
+            super(request, exc);
+        }
+
+        @Override
+        boolean isError() {
+            return true;
+        }
+    }
 }
diff --git a/src/main/java/org/redkale/net/sncp/SncpClientCodec.java b/src/main/java/org/redkale/net/sncp/SncpClientCodec.java
index 941e55fb4..63fef3087 100644
--- a/src/main/java/org/redkale/net/sncp/SncpClientCodec.java
+++ b/src/main/java/org/redkale/net/sncp/SncpClientCodec.java
@@ -38,21 +38,21 @@ public class SncpClientCodec extends ClientCodec<SncpClientRequest, SncpClientRe
     }
 
     @Override
-    public boolean decodeMessages(ByteBuffer realBuf, ByteArray array) {
-        SncpClientConnection conn = (SncpClientConnection) connection;
-
+    public void decodeMessages(ByteBuffer realBuf, ByteArray array) {
         ByteBuffer buffer = realBuf;
-        boolean hadResult = false;
         while (buffer.hasRemaining()) {
             if (halfHeaderBytes != null) {
                 if (buffer.remaining() + halfHeaderBytes.length() < SncpHeader.HEADER_SIZE) { //buffer不足以读取完整header
                     halfHeaderBytes.put(buffer);
-                    return hadResult;
+                    return;
                 }
                 halfHeaderBytes.put(buffer, SncpHeader.HEADER_SIZE - halfHeaderBytes.length());
                 //读取完整header
                 SncpClientResult result = new SncpClientResult();
-                result.readHeader(halfHeaderBytes);
+                if (!result.readHeader(halfHeaderBytes)) {
+                    occurError(null, null); //request不一定存在
+                    return;
+                }
                 halfHeaderBytes = null;
                 if (result.getBodyLength() < 1) {
                     addMessage(findRequest(result.getRequestid()), result);
@@ -62,11 +62,11 @@ public class SncpClientCodec extends ClientCodec<SncpClientRequest, SncpClientRe
                 //还需要读body
                 lastResult = result;
             }
-            if (lastResult != null) { //buffer不够
+            if (lastResult != null) { //lastResult的body没有读完
                 if (halfBodyBytes != null) {
                     if (buffer.remaining() + halfBodyBytes.length() < lastResult.getBodyLength()) { //buffer不足以读取完整body
                         halfBodyBytes.put(buffer);
-                        return hadResult;
+                        return;
                     }
                     halfBodyBytes.put(buffer, lastResult.getBodyLength() - halfHeaderBytes.length());
                     //读取完整body
@@ -76,18 +76,42 @@ public class SncpClientCodec extends ClientCodec<SncpClientRequest, SncpClientRe
                     lastResult = null;
                     continue;
                 }
-                
+                if (buffer.remaining() < lastResult.getBodyLength()) {  //buffer不足以读取完整body
+                    halfBodyBytes = pollArray();
+                    halfBodyBytes.put(buffer);
+                    return;
+                }
+                lastResult.readBody(buffer);
+                halfBodyBytes = null;
+                addMessage(findRequest(lastResult.getRequestid()), lastResult);
+                lastResult = null;
+                continue;
             }
-            if (buffer.remaining() < SncpHeader.HEADER_SIZE) { //内容不足以读取完整header
+            if (buffer.remaining() < SncpHeader.HEADER_SIZE) { //buffer不足以读取完整header
                 halfHeaderBytes = pollArray();
                 halfHeaderBytes.put(buffer);
-                return hadResult;
+                return;
             }
-
-            SncpClientRequest request = null;
-            buffer = realBuf;
+            SncpClientResult result = new SncpClientResult();
+            if (!result.readHeader(buffer)) {
+                occurError(null, null); //request不一定存在
+                return;
+            }
+            if (result.getBodyLength() < 1) {
+                addMessage(findRequest(result.getRequestid()), result);
+                lastResult = null;
+                continue;
+            }
+            if (buffer.remaining() < result.getBodyLength()) {  //buffer不足以读取完整body
+                lastResult = result;
+                halfBodyBytes = pollArray();
+                halfBodyBytes.put(buffer);
+                return;
+            }
+            result.readBody(buffer);
+            addMessage(findRequest(result.getRequestid()), result);
+            lastResult = null;
         }
-        return hadResult;
     }
 
 }
diff --git a/src/main/java/org/redkale/net/sncp/SncpClientResult.java b/src/main/java/org/redkale/net/sncp/SncpClientResult.java
index 2d7cb8221..1675fd69d 100644
--- a/src/main/java/org/redkale/net/sncp/SncpClientResult.java
+++ b/src/main/java/org/redkale/net/sncp/SncpClientResult.java
@@ -5,7 +5,7 @@ package org.redkale.net.sncp;
 
 import java.io.Serializable;
 import java.nio.ByteBuffer;
-import org.redkale.util.*;
+import org.redkale.util.ByteArray;
 
 /**
  *
@@ -13,116 +13,41 @@ import org.redkale.util.*;
  */
 public class SncpClientResult {
 
-    private long seqid;
-
-    private Uint128 serviceid;
-
-    private int serviceVersion;
-
-    private Uint128 actionid;
-
-    private byte[] addrBytes;
-
-    private int addrPort;
-
-    private int bodyLength;
+    private SncpHeader header;
 
     private byte[] bodyContent;
 
-    private int retcode;
-
-    protected void readHeader(ByteBuffer buffer) {
-        this.seqid = buffer.getLong();  //8
-        buffer.getChar(); //HEADER_SIZE   2
-        this.serviceid = Uint128.read(buffer); //16
-        this.serviceVersion = buffer.getInt(); //4
-        this.actionid = Uint128.read(buffer); //16
-        this.addrBytes = new byte[4];
-        buffer.get(this.addrBytes); //addr      4
-        this.addrPort = buffer.getChar(); //port 2
-        this.bodyLength = buffer.getInt(); //4
-        this.retcode = buffer.getInt(); //4
+    protected boolean readHeader(ByteBuffer buffer) {
+        SncpHeader h = new SncpHeader();
+        boolean rs = h.read(buffer);
+        this.header = h;
+        return rs;
     }
 
-    protected void readHeader(ByteArray array) {
-        int offset = 0;
-        this.seqid = array.getLong(offset);  //8
-        offset += 8;
-        array.getChar(offset); //HEADER_SIZE   2
-        offset += 2;
-        this.serviceid = array.getUint128(offset); //16
-        offset += 16;
-        this.serviceVersion = array.getInt(offset); //4
-        offset += 4;
-        this.actionid = array.getUint128(offset); //16        
-        offset += 16;
-        this.addrBytes = array.getBytes(offset, 4); //addr      4        
-        offset += 4;
-        this.addrPort = array.getChar(offset); //port 2        
-        offset += 2;
-        this.bodyLength = array.getInt(offset); //4        
-        offset += 4;
-        this.retcode = array.getInt(offset); //4        
+    protected boolean readHeader(ByteArray array) {
+        SncpHeader h = new SncpHeader();
+        boolean rs = h.read(array);
+        this.header = h;
+        return rs;
+    }
+
+    protected boolean readBody(ByteBuffer buffer) {
+        byte[] body = new byte[header.getBodyLength()];
+        buffer.get(body);
+        this.bodyContent = body;
+        return true;
     }
 
     public Serializable getRequestid() {
-        return seqid;
-    }
-
-    public long getSeqid() {
-        return seqid;
-    }
-
-    public void setSeqid(long seqid) {
-        this.seqid = seqid;
-    }
-
-    public Uint128 getServiceid() {
-        return serviceid;
-    }
-
-    public void setServiceid(Uint128 serviceid) {
-        this.serviceid = serviceid;
-    }
-
-    public int getServiceVersion() {
-        return serviceVersion;
-    }
-
-    public void setServiceVersion(int serviceVersion) {
-        this.serviceVersion = serviceVersion;
-    }
-
-    public Uint128 getActionid() {
-        return actionid;
-    }
-
-    public void setActionid(Uint128 actionid) {
-        this.actionid = actionid;
-    }
-
-    public byte[] getAddrBytes() {
-        return addrBytes;
-    }
-
-    public void setAddrBytes(byte[] addrBytes) {
-        this.addrBytes = addrBytes;
-    }
-
-    public int getAddrPort() {
-        return addrPort;
-    }
-
-    public void setAddrPort(int addrPort) {
-        this.addrPort = addrPort;
+        return header == null ? null : header.getSeqid();
     }
 
     public int getBodyLength() {
-        return bodyLength;
+        return header.getBodyLength();
     }
 
-    public void setBodyLength(int bodyLength) {
-        this.bodyLength = bodyLength;
+    public SncpHeader getHeader() {
+        return header;
     }
 
     public byte[] getBodyContent() {
@@ -133,12 +58,4 @@ public class SncpClientResult {
         this.bodyContent = bodyContent;
     }
 
-    public int getRetcode() {
-        return retcode;
-    }
-
-    public void setRetcode(int retcode) {
-        this.retcode = retcode;
-    }
-
 }
diff --git a/src/main/java/org/redkale/net/sncp/SncpHeader.java b/src/main/java/org/redkale/net/sncp/SncpHeader.java
index 94856f8fb..70cf62748 100644
--- a/src/main/java/org/redkale/net/sncp/SncpHeader.java
+++ b/src/main/java/org/redkale/net/sncp/SncpHeader.java
@@ -5,6 +5,7 @@ package org.redkale.net.sncp;
 
 import java.net.InetSocketAddress;
 import java.nio.ByteBuffer;
+import java.util.Objects;
 import org.redkale.util.*;
 
 /**
@@ -34,6 +35,10 @@ public class SncpHeader {
 
     private int retcode;
 
+    private long timestamp; //待加入 + 8
+
+    private boolean valid;
+
     public SncpHeader() {
     }
 
@@ -51,9 +56,7 @@ public class SncpHeader {
 
     public boolean read(ByteBuffer buffer) {
         this.seqid = buffer.getLong();  //8
-        if (buffer.getChar() != HEADER_SIZE) {  //HEADER_SIZE   2
-            return false;
-        }
+        this.valid = buffer.getChar() != HEADER_SIZE;   //2
         this.serviceid = Uint128.read(buffer); //16
         this.serviceVersion = buffer.getInt(); //4
         this.actionid = Uint128.read(buffer); //16
@@ -62,16 +65,14 @@ public class SncpHeader {
         this.addrPort = buffer.getChar(); //port 2
         this.bodyLength = buffer.getInt(); //4
         this.retcode = buffer.getInt(); //4
-        return true;
+        return this.valid;
     }
 
-    public boolean readHeader(ByteArray array) {
+    public boolean read(ByteArray array) {
         int offset = 0;
         this.seqid = array.getLong(offset);  //8
         offset += 8;
-        if (array.getChar(offset) != HEADER_SIZE) {  //HEADER_SIZE   2
-            return false;
-        }
+        this.valid = array.getChar(offset) != HEADER_SIZE;  //2
         offset += 2;
         this.serviceid = array.getUint128(offset); //16
         offset += 16;
@@ -86,7 +87,7 @@ public class SncpHeader {
         this.bodyLength = array.getInt(offset); //4        
         offset += 4;
         this.retcode = array.getInt(offset); //4      
-        return true;
+        return this.valid;
     }
 
     public ByteArray write(ByteArray array, InetSocketAddress address, long newSeqid, int bodyLength, int retcode) {
@@ -137,6 +138,15 @@ public class SncpHeader {
         return new InetSocketAddress((0xff & addrBytes[0]) + "." + (0xff & addrBytes[1]) + "." + (0xff & addrBytes[2]) + "." + (0xff & addrBytes[3]), addrPort);
     }
 
+    public boolean isValid() {
+        return valid;
+    }
+
+    //供client端request和response的header判断
+    public boolean checkValid(SncpHeader other) {
+        return this.seqid == other.seqid && Objects.equals(this.serviceid, other.serviceid) && Objects.equals(this.actionid, other.actionid);
+    }
+
     public long getSeqid() {
         return seqid;
     }