ebean/src/main/java/io/ebeaninternal/server/query/CQueryDraftSupport.java

package io.ebeaninternal.server.query;

import java.util.Map;

/**
 * Support 'asDraft' queries.
 */
class CQueryDraftSupport {

  /**
   * The mapping of base tables to their matching 'draft' table.
   */
  private final Map<String, String> tableMap;

  CQueryDraftSupport(Map<String, String> tableMap) {
    this.tableMap = tableMap;
  }

  /**
   * Return the draft table associated to the base table.
   * <p>
   * This returns null for entities that are not draftable and in that case
   * the usual base table is used.
   */
  String getDraftTable(String table) {
    return tableMap.get(table);
  }
}