moparisthebest/JdbcMapper
1
0
Fork 0
mirror of https://github.com/moparisthebest/JdbcMapper synced 2024-11-21 16:45:02 -05:00
Code Issues Releases Wiki Activity

Allow to manually specify columnNames and isSelect for unparseable SQL queries in jdbcmapper

Browse Source
This commit is contained in:
Travis Burtrum 2017-08-04 13:55:05 -04:00
parent c1b8b2e705
commit 6f23696952
5 changed files with 65 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
common/src/main/java/com/moparisthebest/jdbc/codegen/JdbcMapper.java
View File

@ -58,6 +58,16 @@ public interface JdbcMapper extends Closeable {
*/
String value();
/**
* This avoids parsing SQL for select statements by specifying column names directly, implies SELECT statement, ie isSelect = true
*/
String[] columnNames() default {};
/**
* This avoids parsing SQL for non-select statements, set to false if executeUpdate should be ran instead of selection
*/
boolean isSelect() default true;
/**
* This defaults to the value of the class-level @JdbcMapper.Mapper.cachePreparedStatements annotation, but can be configured on a per-method level here
*/

2
jdbcmapper/src/main/java/com/moparisthebest/jdbc/codegen/JdbcMapperProcessor.java
View File

@ -322,7 +322,7 @@ public class JdbcMapperProcessor extends AbstractProcessor {
}
}
final SQLParser parsedSQl = parser.parse(sqlStatement);
final SQLParser parsedSQl = ManualSQLParser.getSQLParser(sql, parser, sqlStatement);
// now implementation
w.write("\t\tPreparedStatement ps = null;\n");
if (parsedSQl.isSelect())

32
jdbcmapper/src/main/java/com/moparisthebest/jdbc/codegen/ManualSQLParser.java Normal file
View File

@ -0,0 +1,32 @@
package com.moparisthebest.jdbc.codegen;
/**
* Created by mopar on 5/25/17.
*/
class ManualSQLParser extends AbstractSQLParser {
private static final SQLParser notSelect = new ManualSQLParser(null, false);
private ManualSQLParser(final String[] columnNames, final boolean isSelect) {
super(columnNames, isSelect);
}
public static SQLParser getSQLParser(final JdbcMapper.SQL sql, final SQLParser parser, final String sqlStatement) {
if(sql.columnNames().length != 0) {
final String[] newColumnNames = new String[sql.columnNames().length + 1];
int count = 0;
for (final String columnName : sql.columnNames())
newColumnNames[++count] = columnName.toUpperCase();
return new ManualSQLParser(newColumnNames, true);
} else if(!sql.isSelect()) {
return notSelect;
} else {
return parser.parse(sqlStatement);
}
}
@Override
public SQLParser parse(final String sql) {
throw new UnsupportedOperationException("use getDefinedColumns()");
}
}

11
jdbcmapper/src/test/java/com/moparisthebest/jdbc/codegen/PersonDAO.java
View File

@ -57,9 +57,20 @@ public interface PersonDAO extends JdbcMapper {
@JdbcMapper.SQL(value = "SELECT first_name, last_name FROM person WHERE last_name = {lastName}", cachePreparedStatement = JdbcMapper.OptionalBool.TRUE)
ResultSet getPeopleResultSetCached(String lastName) throws SQLException;
@JdbcMapper.SQL(value = "SELECT first_name FROM person WHERE person_no = {personNo}", columnNames = {"firstName"})
String getFirstNameColumnNames(long personNo) throws SQLException;
@JdbcMapper.SQL("SELECT first_name FROM person WHERE person_no = {personNo}")
String getFirstName(long personNo) throws SQLException;
@JdbcMapper.SQL(value = "SELECT first_name FROM person WHERE person_no = {personNo}", isSelect = false)
int getFirstNameUpdate(long personNo) throws SQLException;
@JdbcMapper.SQL(value = "SELECT person_no, first_name, last_name, birth_date FROM person WHERE person_no = {personNo}",
columnNames = {"personNo", "firstName", "lastName", "birthDate"})
FieldPerson getPersonColumnNames(long personNo) throws SQLException;
@JdbcMapper.SQL(value = "SELECT person_no, first_name, last_name, birth_date FROM person WHERE person_no = {personNo}")
FieldPerson getPerson(long personNo) throws SQLException;

11
presto-sqlparser/src/test/java/com/moparisthebest/jdbc/codegen/PrestoPersonDAO.java
View File

@ -57,9 +57,20 @@ public interface PrestoPersonDAO extends PersonDAO {
@JdbcMapper.SQL(value = "SELECT first_name, last_name FROM person WHERE last_name = {lastName}", cachePreparedStatement = JdbcMapper.OptionalBool.TRUE)
ResultSet getPeopleResultSetCached(String lastName) throws SQLException;
@JdbcMapper.SQL(value = "SELECT first_name FROM person WHERE person_no = {personNo}", columnNames = {"firstName"})
String getFirstNameColumnNames(long personNo) throws SQLException;
@JdbcMapper.SQL("SELECT first_name FROM person WHERE person_no = {personNo}")
String getFirstName(long personNo) throws SQLException;
@JdbcMapper.SQL(value = "SELECT first_name FROM person WHERE person_no = {personNo}", isSelect = false)
int getFirstNameUpdate(long personNo) throws SQLException;
@JdbcMapper.SQL(value = "SELECT person_no, first_name, last_name, birth_date FROM person WHERE person_no = {personNo}",
columnNames = {"personNo", "firstName", "lastName", "birthDate"})
FieldPerson getPersonColumnNames(long personNo) throws SQLException;
@JdbcMapper.SQL(value = "SELECT person_no, first_name, last_name, birth_date FROM person WHERE person_no = {personNo}")
FieldPerson getPerson(long personNo) throws SQLException;