com.julienviet.pgclient.impl.codec.encoder.message.Parse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of reactive-pg-client Show documentation
The reactive Postgres client
There is a newer version: 0.6.0
/*
 * Copyright (C) 2017 Julien Viet
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package com.julienviet.pgclient.impl.codec.encoder.message;

import com.julienviet.pgclient.impl.codec.decoder.message.ErrorResponse;
import com.julienviet.pgclient.impl.codec.decoder.message.ParseComplete;
import com.julienviet.pgclient.impl.codec.encoder.OutboundMessage;
import com.julienviet.pgclient.impl.codec.util.Util;
import io.netty.buffer.ByteBuf;

import java.util.Arrays;
import java.util.Objects;

import static com.julienviet.pgclient.impl.codec.encoder.message.type.MessageType.PARSE;

/**
 * 
 * The message contains a textual SQL query string.
 *
 * 
 * The response is either {@link ParseComplete} or {@link ErrorResponse}
 *
 * @author Emad Alblueshi
 */

public class Parse implements OutboundMessage {

  private final String query;
  private long statement;
  private int[] paramDataTypes;

  public Parse(String query) {
    this.query = query;
  }

  public Parse setStatement(long statement) {
    this.statement = statement;
    return this;
  }

  public Parse setParamDataTypes(int[] paramDataTypes) {
    this.paramDataTypes = paramDataTypes;
    return this;
  }

  public long getStatement() {
    return statement;
  }

  public String getQuery() {
    return query;
  }

  public int[] getParamDataTypes() {
    return paramDataTypes;
  }


  @Override
  public boolean equals(Object o) {
    if (this == o) return true;
    if (o == null || getClass() != o.getClass()) return false;
    Parse parse = (Parse) o;
    return Objects.equals(query, parse.query) &&
      Objects.equals(statement, parse.statement) &&
      Arrays.equals(paramDataTypes, parse.paramDataTypes);
  }

  private static void encode(long statement, String query, int[] paramDataTypes, ByteBuf out) {
    int pos = out.writerIndex();
    out.writeByte(PARSE);
    out.writeInt(0);
    if(statement == 0) {
      out.writeByte(0);
    } else {
      out.writeLong(statement);
    }
    Util.writeCStringUTF8(out, query);
    // no parameter data types (OIDs)
    if(paramDataTypes == null) {
      out.writeShort(0);
    } else {
      // Parameter data types (OIDs)
      out.writeShort(paramDataTypes.length);
      for (int paramDataType : paramDataTypes) {
        out.writeInt(paramDataType);
      }
    }
    out.setInt(pos + 1, out.writerIndex() - pos - 1);
  }

  @Override
  public void encode(ByteBuf out) {
    encode(statement, query, paramDataTypes, out);
  }

  @Override
  public int hashCode() {
    return Objects.hash(query, statement, paramDataTypes);
  }

  @Override
  public String toString() {
    return "Parse{" +
      "query='" + query + '\'' +
      ", statement='" + statement + '\'' +
      ", paramDataTypes=" + Arrays.toString(paramDataTypes) +
      '}';
  }
}