org.mariadb.r2dbc.message.client.ChangeSchemaPacket Maven / Gradle / Ivy

Go to download
// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2020-2024 MariaDB Corporation Ab

package org.mariadb.r2dbc.message.client;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import java.nio.charset.StandardCharsets;
import org.mariadb.r2dbc.message.ClientMessage;
import org.mariadb.r2dbc.message.Context;
import reactor.core.publisher.Mono;

/**
 * see COM_INIT_DB https://mariadb.com/kb/en/com_init_db/ COM_INIT_DB is used to specify the default
 * schema for the connection.
 */
public final class ChangeSchemaPacket implements ClientMessage {
  private final String schema;

  /**
   * Constructor
   *
   * @param schema new default schema
   */
  public ChangeSchemaPacket(String schema) {
    this.schema = schema;
  }

  @Override
  public Mono encode(Context context, ByteBufAllocator allocator) {
    ByteBuf buf = allocator.ioBuffer();
    buf.writeByte(0x02);
    buf.writeCharSequence(this.schema, StandardCharsets.UTF_8);
    return Mono.just(buf);
  }
}