Commit bcee8be3 authored by terrymanu's avatar terrymanu
Browse files

refactor ExecuteUpdateResponseUnit

parent 9dd58b12

sharding-proxy/src/main/java/io/shardingsphere/proxy/backend/common/jdbc/execute/memory/ConnectionStrictlyExecuteEngine.java

+1 −1
Original line number Diff line number Diff line
@@ -132,7 +132,7 @@ public abstract class ConnectionStrictlyExecuteEngine extends JDBCExecuteEngine 
        for (Future<Collection<ExecuteResponseUnit>> each : futureList) {

            try {

                for (ExecuteResponseUnit executeResponse : each.get()) {

                    result.getPackets().add(((ExecuteUpdateResponseUnit) executeResponse).getDatabasePacket());

                    result.getPackets().add(((ExecuteUpdateResponseUnit) executeResponse).getOkPacket());

                }

            } catch (final InterruptedException | ExecutionException ex) {

                throw new ShardingException(ex.getMessage(), ex);

sharding-proxy/src/main/java/io/shardingsphere/proxy/backend/common/jdbc/execute/response/ExecuteUpdateResponse.java

+1 −1
Original line number Diff line number Diff line
@@ -44,7 +44,7 @@ public final class ExecuteUpdateResponse implements ExecuteResponse { 
    

    public ExecuteUpdateResponse(final Collection<ExecuteResponseUnit> responseUnits) {

        for (ExecuteResponseUnit each : responseUnits) {

            packets.add(((ExecuteUpdateResponseUnit) each).getDatabasePacket());

            packets.add(((ExecuteUpdateResponseUnit) each).getOkPacket());

        }

    }

sharding-proxy/src/main/java/io/shardingsphere/proxy/backend/common/jdbc/execute/response/unit/ExecuteUpdateResponseUnit.java

+2 −2
Original line number Diff line number Diff line
@@ -17,7 +17,7 @@ 


package io.shardingsphere.proxy.backend.common.jdbc.execute.response.unit;



import io.shardingsphere.proxy.transport.common.packet.DatabasePacket;

import io.shardingsphere.proxy.transport.mysql.packet.generic.OKPacket;

import lombok.Getter;

import lombok.RequiredArgsConstructor;
@@ -30,5 +30,5 @@ import lombok.RequiredArgsConstructor; 
@Getter

public final class ExecuteUpdateResponseUnit implements ExecuteResponseUnit {

    

    private final DatabasePacket databasePacket;

    private final OKPacket okPacket;

}

sharding-proxy/src/main/java/io/shardingsphere/proxy/backend/common/jdbc/execute/stream/MemoryStrictlyExecuteEngine.java

+2 −2
Original line number Diff line number Diff line
@@ -97,10 +97,10 @@ public abstract class MemoryStrictlyExecuteEngine extends JDBCExecuteEngine { 
    }

    

    private ExecuteResponse getExecuteUpdateResponse(final ExecuteUpdateResponseUnit firstResponseUnit, final List<Future<ExecuteResponseUnit>> futureList) {

        ExecuteUpdateResponse result = new ExecuteUpdateResponse(firstResponseUnit.getDatabasePacket());

        ExecuteUpdateResponse result = new ExecuteUpdateResponse(firstResponseUnit.getOkPacket());

        for (Future<ExecuteResponseUnit> each : futureList) {

            try {

                result.getPackets().add(((ExecuteUpdateResponseUnit) each.get()).getDatabasePacket());

                result.getPackets().add(((ExecuteUpdateResponseUnit) each.get()).getOkPacket());

            } catch (final InterruptedException | ExecutionException ex) {

                throw new ShardingException(ex.getMessage(), ex);

            }