Command complete information has been enriched

Author: marat-gainullin

src/main/java/com/github/pgasync/io/backend/CommandCompleteDecoder.java

@@ -54,7 +54,14 @@ public byte getMessageId() {
     @Override
     public CommandComplete read(ByteBuffer buffer, Charset encoding) {
         String tag = getCString(buffer, encoding);
-        return new CommandComplete(tag);
+        int affectedRows;
+        if (tag.contains("INSERT") || tag.contains("UPDATE") || tag.contains("DELETE")) {
+            String[] parts = tag.split(" ");
+            affectedRows = Integer.parseInt(parts[parts.length - 1]);
+        } else {
+            affectedRows = 0;
+        }
+        return new CommandComplete(tag, affectedRows);
     }
 
 }

src/main/java/com/github/pgasync/message/backend/CommandComplete.java

@@ -17,19 +17,20 @@
 import com.github.pgasync.message.Message;
 
 /**
- * @author  Antti Laisi
+ * @author Antti Laisi
  */
 public class CommandComplete implements Message {
 
+    private final String tag;
     private final int affectedRows;
 
-    public CommandComplete(String tag) {
-        if (tag.contains("INSERT") || tag.contains("UPDATE") || tag.contains("DELETE")) {
-            String[] parts = tag.split(" ");
-            affectedRows = Integer.parseInt(parts[parts.length - 1]);
-        } else {
-            affectedRows = 0;
-        }
+    public CommandComplete(String tag, int affectedRows) {
+        this.tag = tag;
+        this.affectedRows = affectedRows;
+    }
+
+    public String getTag() {
+        return tag;
     }
 
     public int getAffectedRows() {
@@ -38,6 +39,9 @@ public int getAffectedRows() {
 
     @Override
     public String toString() {
-        return String.format("CommandComplete(affectedRows=%d)", affectedRows);
+        return "CommandComplete(" +
+                "tag='" + tag + '\'' +
+                ", affectedRows=" + affectedRows +
+                ')';
     }
 }