How to use

DBRouter

in

org.killbill.billing.util.entity.dao

@Inject
public AuditLogCacheLoader(final IDBI dbi, @Named(MAIN_RO_IDBI_NAMED) final IDBI roDbi) {
  super();
  this.dbRouter = new DBRouter<AuditSqlDao>(dbi, roDbi, AuditSqlDao.class);
}

@Override
public Boolean recordsWithTrackingIdExist(final UUID subscriptionId, final String trackingId, final InternalTenantContext context) {
  return dbRouter.onDemand(false).recordsWithTrackingIdExist(subscriptionId, trackingId, context) != null;
}

@Override
public void updateNodeInfo(final String nodeName, final String nodeInfo) {
  dbRouter.inTransaction(false, new TransactionCallback<Void>() {
    @Override
    public Void inTransaction(final Handle handle, final TransactionStatus status) throws Exception {
      final NodeInfoSqlDao sqlDao = handle.attach(NodeInfoSqlDao.class);
      final Date updateDate = clock.getUTCNow().toDate();
      sqlDao.updateNodeInfo(nodeName, nodeInfo, updateDate);
      return null;
    }
  });
}

  public C onDemand(final boolean requestedRO) {
    if (shouldUseRODBI(requestedRO)) {
      return roOnDemand;
    } else {
      return onDemand;
    }
  }
}

  public C onDemand(final boolean requestedRO) {
    if (shouldUseRODBI(requestedRO)) {
      return roOnDemand;
    } else {
      return onDemand;
    }
  }
}

  public List<DefaultColumnInfo> getColumnInfoList(@Nullable final String schemaName) {
    return dbRouter.onDemand(true).getSchemaInfo(schemaName);
  }
}

@Inject
public DefaultNodeInfoDao(final IDBI dbi, @Named(MAIN_RO_IDBI_NAMED) final IDBI roDbi, final Clock clock) {
  this.dbRouter = new DBRouter<NodeInfoSqlDao>(dbi, roDbi, NodeInfoSqlDao.class);
  this.clock = clock;
}

@Override
public List<NodeInfoModelDao> getAll() {
  return dbRouter.inTransaction(true, new TransactionCallback<List<NodeInfoModelDao>>() {
    @Override
    public List<NodeInfoModelDao> inTransaction(final Handle handle, final TransactionStatus status) throws Exception {
      final NodeInfoSqlDao sqlDao = handle.attach(NodeInfoSqlDao.class);
      return sqlDao.getAll();
    }
  });
}

@Override
public void record(final Iterable<RolledUpUsageModelDao> usages, final InternalCallContext context) {
  dbRouter.onDemand(false).create(usages, context);
}

@Inject
public DefaultRolledUpUsageDao(final IDBI dbi, @Named(MAIN_RO_IDBI_NAMED) final IDBI roDbi) {
  this.dbRouter = new DBRouter<RolledUpUsageSqlDao>(dbi, roDbi, RolledUpUsageSqlDao.class);
}

  @Override
  public NodeInfoModelDao getByNodeName(final String nodeName) {
    return dbRouter.inTransaction(true, new TransactionCallback<NodeInfoModelDao>() {
      @Override
      public NodeInfoModelDao inTransaction(final Handle handle, final TransactionStatus status) throws Exception {
        final NodeInfoSqlDao sqlDao = handle.attach(NodeInfoSqlDao.class);
        return sqlDao.getByNodeName(nodeName);
      }
    });
  }
}

@Override
public Long retrieveHistoryTargetRecordId(@Nullable final Long recordId, final TableName tableName) {
  return dbRouter.onDemand(true).getHistoryTargetRecordId(recordId, tableName.getTableName());
}

@Inject
public DatabaseSchemaDao(final IDBI dbi, @Named(MAIN_RO_IDBI_NAMED) final IDBI roDbi) {
  this.dbRouter = new DBRouter<DatabaseSchemaSqlDao>(dbi, roDbi, DatabaseSchemaSqlDao.class);
}

@Override
public void create(final BroadcastModelDao broadcastModelDao) {
  dbRouter.inTransaction(false, new TransactionCallback<Void>() {
    @Override
    public Void inTransaction(final Handle handle, final TransactionStatus status) throws Exception {
      final BroadcastSqlDao sqlDao = handle.attach(BroadcastSqlDao.class);
      sqlDao.create(broadcastModelDao);
      return null;
    }
  });
}

@Override
protected Session doReadSession(final Serializable sessionId) {
  // Shiro should not pass us a null sessionId, but be safe...
  if (sessionId == null) {
    return null;
  }
  final String sessionIdString = sessionId.toString();
  final SessionModelDao sessionModelDao = dbRouter.onDemand(true).read(sessionIdString);
  if (sessionModelDao == null) {
    return null;
  }
  return toSession(sessionModelDao);
}

@Inject
public JDBCSessionDao(final IDBI dbi, @Named(MAIN_RO_IDBI_NAMED) final IDBI roDbi) {
  this.dbRouter = new DBRouter<JDBCSessionSqlDao>(dbi, roDbi, JDBCSessionSqlDao.class);
}

@Override
public List<BroadcastModelDao> getLatestEntriesFrom(final Long recordId) {
  return dbRouter.inTransaction(true, new TransactionCallback<List<BroadcastModelDao>>() {
    @Override
    public List<BroadcastModelDao> inTransaction(final Handle handle, final TransactionStatus status) throws Exception {
      final BroadcastSqlDao sqlDao = handle.attach(BroadcastSqlDao.class);
      return sqlDao.getLatestEntries(recordId);
    }
  });
}

private List<AuditLog> doGetAuditLogsForId(final TableName tableName, final UUID objectId, final AuditLevel auditLevel, final InternalTenantContext context) {
  final Long recordId = dbRouter.onDemand(true).getRecordIdFromObject(objectId.toString(), tableName.getTableName());
  if (recordId == null) {
    return ImmutableList.<AuditLog>of();
  } else {
    return getAuditLogsForRecordId(tableName, objectId, recordId, auditLevel, context);
  }
}

@Inject
public DefaultBroadcastDao(final IDBI dbi, @Named(MAIN_RO_IDBI_NAMED) final IDBI roDbi) {
  this.dbRouter = new DBRouter<BroadcastSqlDao>(dbi, roDbi, BroadcastSqlDao.class);
}

@Override
public void delete(final String nodeName) {
  dbRouter.inTransaction(false, new TransactionCallback<Void>() {
    @Override
    public Void inTransaction(final Handle handle, final TransactionStatus status) throws Exception {
      final NodeInfoSqlDao sqlDao = handle.attach(NodeInfoSqlDao.class);
      sqlDao.delete(nodeName);
      return null;
    }
  });
}