com.tesora.dve.db.mysql.libmy.MyOKResponse.java Source code

Introduction

Here is the source code for com.tesora.dve.db.mysql.libmy.MyOKResponse.java

Source

package com.tesora.dve.db.mysql.libmy;

/*
 * #%L
 * Tesora Inc.
 * Database Virtualization Engine
 * %%
 * Copyright (C) 2011 - 2014 Tesora Inc.
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

import java.nio.ByteOrder;

import io.netty.buffer.ByteBuf;
import io.netty.util.CharsetUtil;

import com.tesora.dve.db.mysql.common.MysqlAPIUtils;

public class MyOKResponse extends MyResponseMessage {

    public static final byte OKPKT_INDICATOR = 0;

    private long affectedRows = 0;
    private long insertId = 0;
    private int serverStatus = MyProtocolDefs.SERVER_STATUS_AUTOCOMMIT;
    private int warningCount = 0;
    private String message;

    public MyOKResponse() {
    }

    @Override
    public void marshallMessage(ByteBuf in) {
        ByteBuf cb = in.order(ByteOrder.LITTLE_ENDIAN);
        cb.writeByte(0); // field_count - spec says this is always 0
        MysqlAPIUtils.putLengthCodedLong(cb, affectedRows);
        MysqlAPIUtils.putLengthCodedLong(cb, insertId);
        cb.writeShort(serverStatus);
        cb.writeShort(warningCount);
        if (message != null && message.length() > 0) {
            cb.writeBytes(message.getBytes());
        }
    }

    public int calculateSize() {
        return 23 + ((message != null) ? message.length() : 0);
    }

    @Override
    public void unmarshallMessage(ByteBuf cb) {
        cb.readByte(); // field count - always 0
        affectedRows = MysqlAPIUtils.getLengthCodedLong(cb);
        insertId = MysqlAPIUtils.getLengthCodedLong(cb);
        serverStatus = cb.readUnsignedShort();
        warningCount = cb.readUnsignedShort();
        if (cb.isReadable())
            message = MysqlAPIUtils.readBytesAsString(cb, CharsetUtil.UTF_8);
    }

    public static long getAffectedRows(ByteBuf cb) {
        cb.readByte();
        return MysqlAPIUtils.getLengthCodedLong(cb);
    }

    public long getAffectedRows() {
        return affectedRows;
    }

    public void setAffectedRows(long affectedRows) {
        this.affectedRows = affectedRows;
    }

    public void setAffectedRows(int affectedRows) {
        this.affectedRows = (long) affectedRows;
    }

    public long getInsertId() {
        return insertId;
    }

    public void setInsertId(long insertId) {
        this.insertId = insertId;
    }

    public short getWarningCount() {
        return (short) warningCount;
    }

    public void setWarningCount(short warningCount) {
        this.warningCount = warningCount;
    }

    public String getMessage() {
        return message;
    }

    public void setMessage(String message) {
        this.message = message;
    }

    public short getServerStatus() {
        return (short) serverStatus;
    }

    public void setServerStatus(short serverStatus) {
        this.serverStatus = serverStatus;
    }

    public void setStatusInTrans(boolean inTransaction) {
        if (inTransaction)
            serverStatus = (short) (serverStatus | MyProtocolDefs.SERVER_STATUS_IN_TRANS);
        else
            serverStatus = (short) (serverStatus & ~MyProtocolDefs.SERVER_STATUS_IN_TRANS);
    }

    @Override
    public MyMessageType getMessageType() {
        return MyMessageType.OK_RESPONSE;
    }

    @Override
    public String toString() {
        return super.toString() + " affectedRows=" + affectedRows + " insertId=" + insertId + " warningCount="
                + warningCount;
    }

}