|
|
|
/*
|
|
|
|
This file is part of Airsonic.
|
|
|
|
|
|
|
|
Airsonic is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
Airsonic is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with Airsonic. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
Copyright 2016 (C) Airsonic Authors
|
|
|
|
Based upon Subsonic, Copyright 2009 (C) Sindre Mehus
|
|
|
|
*/
|
|
|
|
package org.airsonic.player.dao;
|
|
|
|
|
|
|
|
import org.apache.commons.lang.StringUtils;
|
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
import org.springframework.jdbc.core.JdbcTemplate;
|
|
|
|
import org.springframework.jdbc.core.RowMapper;
|
|
|
|
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
|
|
|
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
import java.util.Date;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
|
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Abstract superclass for all DAO's.
|
|
|
|
*
|
|
|
|
* @author Sindre Mehus
|
|
|
|
*/
|
|
|
|
public class AbstractDao {
|
|
|
|
private static final Logger LOG = LoggerFactory.getLogger(AbstractDao.class);
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
private DaoHelper daoHelper;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Returns a JDBC template for performing database operations.
|
|
|
|
* @return A JDBC template.
|
|
|
|
*/
|
|
|
|
public JdbcTemplate getJdbcTemplate() {
|
|
|
|
return daoHelper.getJdbcTemplate();
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Similar to {@link #getJdbcTemplate()}, but with named parameters.
|
|
|
|
*/
|
|
|
|
public NamedParameterJdbcTemplate getNamedParameterJdbcTemplate() {
|
|
|
|
return daoHelper.getNamedParameterJdbcTemplate();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected String questionMarks(String columns) {
|
|
|
|
int numberOfColumns = StringUtils.countMatches(columns, ",") + 1;
|
|
|
|
return StringUtils.repeat("?", ", ", numberOfColumns);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected String prefix(String columns, String prefix) {
|
|
|
|
List<String> l = Arrays.asList(columns.split(", "));
|
|
|
|
l.replaceAll(s -> prefix + "." + s);
|
|
|
|
return String.join(", ", l);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected int update(String sql, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
LOG.trace("Executing query: [{}]", sql);
|
|
|
|
int result = getJdbcTemplate().update(sql, args);
|
|
|
|
LOG.trace("Updated {} rows", result);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void log(String sql, long startTimeNano) {
|
|
|
|
long millis = (System.nanoTime() - startTimeNano) / 1000000L;
|
|
|
|
|
|
|
|
// Log queries that take more than 2 seconds.
|
|
|
|
if (millis > TimeUnit.SECONDS.toMillis(2L)) {
|
|
|
|
LOG.debug(millis + " ms: " + sql);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
protected <T> List<T> query(String sql, RowMapper rowMapper, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<T> result = getJdbcTemplate().query(sql, args, rowMapper);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected <T> List<T> namedQuery(String sql, RowMapper rowMapper, Map<String, Object> args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<T> result = getNamedParameterJdbcTemplate().query(sql, args, rowMapper);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected <T> List<T> namedQueryWithLimit(String sql, RowMapper<T> rowMapper, Map<String, Object> args, int limit) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
JdbcTemplate jdbcTemplate = new JdbcTemplate(daoHelper.getDataSource());
|
|
|
|
jdbcTemplate.setMaxRows(limit);
|
|
|
|
NamedParameterJdbcTemplate namedTemplate = new NamedParameterJdbcTemplate(jdbcTemplate);
|
|
|
|
List<T> result = namedTemplate.query(sql, args, rowMapper);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected List<String> queryForStrings(String sql, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<String> result = getJdbcTemplate().queryForList(sql, args, String.class);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected List<Integer> queryForInts(String sql, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<Integer> result = getJdbcTemplate().queryForList(sql, args, Integer.class);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected List<String> namedQueryForStrings(String sql, Map<String, Object> args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<String> result = getNamedParameterJdbcTemplate().queryForList(sql, args, String.class);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected Integer queryForInt(String sql, Integer defaultValue, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<Integer> list = getJdbcTemplate().queryForList(sql, args, Integer.class);
|
|
|
|
Integer result = list.isEmpty() ? defaultValue : list.get(0) == null ? defaultValue : list.get(0);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected Integer namedQueryForInt(String sql, Integer defaultValue, Map<String, Object> args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<Integer> list = getNamedParameterJdbcTemplate().queryForList(sql, args, Integer.class);
|
|
|
|
Integer result = list.isEmpty() ? defaultValue : list.get(0) == null ? defaultValue : list.get(0);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected Date queryForDate(String sql, Date defaultValue, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<Date> list = getJdbcTemplate().queryForList(sql, args, Date.class);
|
|
|
|
Date result = list.isEmpty() ? defaultValue : list.get(0) == null ? defaultValue : list.get(0);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected Long queryForLong(String sql, Long defaultValue, Object... args) {
|
|
|
|
long t = System.nanoTime();
|
|
|
|
List<Long> list = getJdbcTemplate().queryForList(sql, args, Long.class);
|
|
|
|
Long result = list.isEmpty() ? defaultValue : list.get(0) == null ? defaultValue : list.get(0);
|
|
|
|
log(sql, t);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected <T> T queryOne(String sql, RowMapper rowMapper, Object... args) {
|
|
|
|
List<T> list = query(sql, rowMapper, args);
|
|
|
|
return list.isEmpty() ? null : list.get(0);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected <T> T namedQueryOne(String sql, RowMapper rowMapper, Map<String, Object> args) {
|
|
|
|
List<T> list = namedQuery(sql, rowMapper, args);
|
|
|
|
return list.isEmpty() ? null : list.get(0);
|
|
|
|
}
|
|
|
|
|
|
|
|
public void setDaoHelper(DaoHelper daoHelper) {
|
|
|
|
this.daoHelper = daoHelper;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void checkpoint() {
|
|
|
|
daoHelper.checkpoint();
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|