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

common/src/main/java/com/moparisthebest/jdbc/codegen/JdbcMapper.java

@@ -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
 		 */

jdbcmapper/src/main/java/com/moparisthebest/jdbc/codegen/JdbcMapperProcessor.java

@@ -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())

jdbcmapper/src/main/java/com/moparisthebest/jdbc/codegen/ManualSQLParser.java

@@ -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()");
+	}
+}

jdbcmapper/src/test/java/com/moparisthebest/jdbc/codegen/PersonDAO.java

@@ -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;
 

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

@@ -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;