org.postgresql.adba.communication.network.ExecuteResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pgadba Show documentation

ADBA implementation for PostgreSQL

The newest version!

package org.postgresql.adba.communication.network;

import java.io.IOException;

import org.postgresql.adba.communication.BeFrame;
import org.postgresql.adba.communication.NetworkReadContext;
import org.postgresql.adba.communication.NetworkResponse;
import org.postgresql.adba.communication.packets.CommandComplete;
import org.postgresql.adba.communication.packets.DataRow;

/**
 * Execute {@link NetworkResponse}.
 * 
 * @author Daniel Sagenschneider
 */
public class ExecuteResponse extends AbstractPortalResponse {

  public ExecuteResponse(Portal portal) {
    super(portal);
  }

  @Override
  public NetworkResponse read(NetworkReadContext context) throws IOException {
    BeFrame frame = context.getBeFrame();
    switch (frame.getTag()) {

      case DATA_ROW:
        if (!portal.getQuery().isCanceled()) {
          DataRow dataRow = new DataRow(frame.getPayload(), portal.getQuery().getRowDescription().getDescriptions(),
              portal.nextRowNumber(), portal.getQuery());
          portal.addDataRow(dataRow);
        }
        return this;

      case COMMAND_COMPLETE:
        CommandComplete complete = new CommandComplete(frame.getPayload());
        portal.commandComplete(complete, context.getSocketChannel());
        return this;

      case READY_FOR_QUERY:
        return null;

      default:
        throw new IllegalStateException("Invalid tag '" + frame.getTag() + "' for " + getClass().getSimpleName());
    }
  }

}