io.ifar.skidroad.jdbi.JDBILogFileDAOHelper Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of skid-road-jdbi Show documentation
Show all versions of skid-road-jdbi Show documentation
Provides JDBI-based LogFileTracker implementation
package io.ifar.skidroad.jdbi;
import com.google.common.collect.Iterators;
import io.ifar.skidroad.LogFile;
import io.ifar.skidroad.tracking.LogFileState;
import org.joda.time.DateTime;
import org.skife.jdbi.v2.ResultIterator;
import java.net.URI;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
/**
* Provides abstraction to gloss over JDBILogFileDAO vs. JDBILogFileDAOWithArraySupport.
*
* Note that methods which return iterators will fetch all results into memory when used
* without JDBILogFileDAOWithArraySupport. Not advised for production use.
*
* TODO: consolidate duplicate code
*/
public class JDBILogFileDAOHelper {
public static Class bestDefaultDAOForDriver(String jdbcDriverClass) {
if ("org.postgresql.Driver".equals(jdbcDriverClass)) {
//supports arrays
return DefaultPostgresJDBILogFileDAO.class;
} else {
return DefaultJDBILogFileDAO.class;
}
}
public static ResultIterator findByOwnerAndState(JDBILogFileDAO dao, URI owner, Set states) {
if (dao instanceof JDBILogFileDAOWithArraySupport) {
Set stateStrings = new HashSet<>(states.size());
for (LogFileState state : states) {
stateStrings.add(state.toString());
}
return ((JDBILogFileDAOWithArraySupport)dao).findByOwnerAndState(
owner.toString(),
stateStrings);
} else if (states.size() == 1) {
return dao.findByOwnerAndState(owner.toString(), states.iterator().next().toString());
} else {
Iterator accumulator = null;
for (LogFileState state : states) {
ArrayList fetched = new ArrayList<>();
Iterators.addAll(fetched, dao.findByOwnerAndState(owner.toString(), state.toString()));
accumulator = (accumulator == null) ? fetched.iterator() : Iterators.concat(accumulator, fetched.iterator());
}
final Iterator result = accumulator == null ? Iterators.emptyIterator() : accumulator;
return new ResultIterator() {
@Override
public void close() {
// no op for this one.
}
@Override
public boolean hasNext() {
return result.hasNext();
}
@Override
public LogFile next() {
return result.next();
}
@Override
public void remove() {
throw new UnsupportedOperationException();
}
};
}
}
public static ResultIterator listLogFilesByOwnerAndDateAndState(JDBILogFileDAO dao, URI owner,
Set states, DateTime startDate, DateTime endDate) {
if (dao instanceof JDBILogFileDAOWithArraySupport) {
Set stateNames = new HashSet<>();
for (LogFileState state : states) {
stateNames.add(state.name());
}
return ((JDBILogFileDAOWithArraySupport) dao).listLogFilesByOwnerAndDateAndState(stateNames,
owner.toString(), startDate, endDate);
} else if (states.size() == 1) {
return dao.listLogFilesByOwnerAndDateAndState(states.iterator().next().name(), owner.toString(), startDate, endDate);
} else {
//May be the case that it isn't possible to hold multiple database cursors open at once:
//java.sql.SQLException: invalid cursor state: identified cursor is not open
//Therefore drain each iterator fully. (Alternatively could implement smart iterator that doesn't issue
//next query until prior query exhausted...)
Iterator accumulator = Iterators.emptyIterator();
for (LogFileState state : states) {
ArrayList fetched = new ArrayList<>();
Iterators.addAll(fetched, dao.listLogFilesByOwnerAndDateAndState(state.name(), owner.toString(), startDate, endDate));
accumulator = (accumulator == null) ? fetched.iterator() : Iterators.concat(accumulator, fetched.iterator());
}
final Iterator result = accumulator;
return new ResultIterator() {
@Override
public void close() {
// no op for this one.
}
@Override
public boolean hasNext() {
return result.hasNext();
}
@Override
public LogFile next() {
return result.next();
}
@Override
public void remove() {
throw new UnsupportedOperationException();
}
};
}
}
public static ResultIterator listLogFilesByDateAndState(JDBILogFileDAO dao, Set states, DateTime startDate, DateTime endDate) {
if (dao instanceof JDBILogFileDAOWithArraySupport) {
return ((JDBILogFileDAOWithArraySupport) dao).listLogFilesByDateAndState(states, startDate, endDate);
} else if (states.size() == 1) {
return dao.listLogFilesByDateAndState(states.iterator().next(), startDate, endDate);
} else {
//May be the case that it isn't possible to hold multiple database cursors open at once:
//java.sql.SQLException: invalid cursor state: identified cursor is not open
//Therefore drain each iterator fully. (Alternatively could implement smart iterator that doesn't issue
//next query until prior query exhausted...)
Iterator accumulator = Iterators.emptyIterator();
for (String state : states) {
ArrayList fetched = new ArrayList<>();
Iterators.addAll(fetched, dao.listLogFilesByDateAndState(state, startDate, endDate));
accumulator = (accumulator == null) ? fetched.iterator() : Iterators.concat(accumulator, fetched.iterator());
}
final Iterator result = accumulator;
return new ResultIterator() {
@Override
public void close() {
// no op for this one.
}
@Override
public boolean hasNext() {
return result.hasNext();
}
@Override
public LogFile next() {
return result.next();
}
@Override
public void remove() {
throw new UnsupportedOperationException();
}
};
}
}
public static ResultIterator listLogFilesByDate(JDBILogFileDAO dao, DateTime startDate, DateTime endDate) {
return dao.listLogFilesByDate(startDate, endDate);
}
public static int count(JDBILogFileDAO dao, Set states) {
if (dao instanceof JDBILogFileDAOWithArraySupport) {
return ((JDBILogFileDAOWithArraySupport)dao).count(states);
} else if (states.size() == 1) {
return dao.count(states.iterator().next());
} else {
int result = 0;
for (String state : states) {
result += dao.count(state);
}
return result;
}
}
public static int count(JDBILogFileDAO dao, Set states, DateTime startDate, DateTime endDate) {
if (dao instanceof JDBILogFileDAOWithArraySupport) {
return ((JDBILogFileDAOWithArraySupport)dao).count(states, startDate, endDate);
} else if (states.size() == 1) {
return dao.count(states.iterator().next(), startDate, endDate);
} else {
int result = 0;
for (String state : states) {
result += dao.count(state, startDate, endDate);
}
return result;
}
}
public static long totalSize(JDBILogFileDAO dao, Set states, DateTime startDate, DateTime endDate) {
if (dao instanceof JDBILogFileDAOWithArraySupport) {
return ((JDBILogFileDAOWithArraySupport)dao).totalSize(states, startDate, endDate);
} else if (states.size() == 1) {
return dao.totalSize(states.iterator().next(), startDate, endDate);
} else {
long result = 0;
for (String state : states) {
result += dao.totalSize(state, startDate, endDate);
}
return result;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy