org.mariadb.r2dbc.ExceptionFactory Maven / Gradle / Ivy

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

package org.mariadb.r2dbc;

import io.r2dbc.spi.*;
import org.mariadb.r2dbc.message.ServerMessage;
import org.mariadb.r2dbc.message.server.ErrorPacket;
import reactor.core.publisher.SynchronousSink;

public final class ExceptionFactory {

  public static final ExceptionFactory INSTANCE = new ExceptionFactory("");
  private final String sql;

  private ExceptionFactory(String sql) {
    this.sql = sql;
  }

  public static ExceptionFactory withSql(String sql) {
    return new ExceptionFactory(sql);
  }

  public static R2dbcException createException(ErrorPacket error, String sql) {
    return createException(error.message(), error.sqlState(), error.errorCode(), sql);
  }

  public static R2dbcException createException(
      String message, String sqlState, int errorCode, String sql) {

    if ("70100".equals(sqlState) || errorCode == 3024) { // ER_QUERY_INTERRUPTED
      return new R2dbcTimeoutException(message, sqlState, errorCode, sql);
    }

    String sqlClass = sqlState.substring(0, 2);
    switch (sqlClass) {
      case "0A":
      case "22":
      case "26":
      case "2F":
      case "20":
      case "42":
      case "XA":
        return new R2dbcBadGrammarException(message, sqlState, errorCode, sql);
      case "25":
      case "28":
        return new R2dbcPermissionDeniedException(message, sqlState, errorCode, sql);
      case "21":
      case "23":
        return new R2dbcDataIntegrityViolationException(message, sqlState, errorCode, sql);
      case "H1":
      case "08":
        return new R2dbcNonTransientResourceException(message, sqlState, errorCode, sql);
      case "40":
        return new R2dbcRollbackException(message, sqlState, errorCode, sql);
    }

    return new R2dbcTransientResourceException(message, sqlState, errorCode, sql);
  }

  public R2dbcException createParsingException(String message) {
    return new R2dbcNonTransientResourceException(message, "H1000", 9000, this.sql);
  }

  public R2dbcException createParsingException(String message, Throwable cause) {
    return new R2dbcNonTransientResourceException(message, "H1000", 9000, this.sql, cause);
  }

  public R2dbcException createConnectionErrorException(String message) {
    return new R2dbcNonTransientResourceException(message, "08000", 9000, this.sql);
  }

  public R2dbcException createConnectionErrorException(String message, Throwable cause) {
    return new R2dbcNonTransientResourceException(
        message + " : " + cause.getMessage(), "08000", 9000, this.sql, cause);
  }

  public R2dbcException createException(String message, String sqlState, int errorCode) {
    return ExceptionFactory.createException(message, sqlState, errorCode, this.sql);
  }

  public R2dbcException from(ErrorPacket err) {
    return createException(err, this.sql);
  }

  public void handleErrorResponse(ServerMessage message, SynchronousSink sink) {
    if (message instanceof ErrorPacket) {
      sink.error(createException((ErrorPacket) message, this.sql));
    } else {
      sink.next(message);
    }
  }
}