mirror of https://github.com/apache/nifi.git
NIFI-4253: Add Oracle 12+ database adapter
Signed-off-by: Pierre Villard <pierre.villard.fr@gmail.com> This closes #2048.
This commit is contained in:
parent
2502b79bae
commit
2d33582147
|
@ -0,0 +1,72 @@
|
||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
||||||
|
* (the "License"); you may not use this file except in compliance with
|
||||||
|
* the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.apache.nifi.processors.standard.db.impl;
|
||||||
|
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.apache.nifi.processors.standard.db.DatabaseAdapter;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A database adapter that generates MS SQL Compatible SQL.
|
||||||
|
*/
|
||||||
|
public class Oracle12DatabaseAdapter implements DatabaseAdapter {
|
||||||
|
@Override
|
||||||
|
public String getName() {
|
||||||
|
return "Oracle 12+";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getDescription() {
|
||||||
|
return "Generates Oracle compliant SQL for version 12 or greater";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String getSelectStatement(String tableName, String columnNames, String whereClause, String orderByClause, Long limit, Long offset) {
|
||||||
|
if (StringUtils.isEmpty(tableName)) {
|
||||||
|
throw new IllegalArgumentException("Table name cannot be null or empty");
|
||||||
|
}
|
||||||
|
final StringBuilder query = new StringBuilder("SELECT ");
|
||||||
|
|
||||||
|
if (StringUtils.isEmpty(columnNames) || columnNames.trim().equals("*")) {
|
||||||
|
query.append("*");
|
||||||
|
} else {
|
||||||
|
query.append(columnNames);
|
||||||
|
}
|
||||||
|
query.append(" FROM ");
|
||||||
|
query.append(tableName);
|
||||||
|
|
||||||
|
if (!StringUtils.isEmpty(whereClause)) {
|
||||||
|
query.append(" WHERE ");
|
||||||
|
query.append(whereClause);
|
||||||
|
}
|
||||||
|
if (!StringUtils.isEmpty(orderByClause)) {
|
||||||
|
query.append(" ORDER BY ");
|
||||||
|
query.append(orderByClause);
|
||||||
|
}
|
||||||
|
if (offset != null && offset > 0) {
|
||||||
|
query.append(" OFFSET ");
|
||||||
|
query.append(offset);
|
||||||
|
query.append(" ROWS");
|
||||||
|
}
|
||||||
|
if (limit != null) {
|
||||||
|
query.append(" FETCH NEXT ");
|
||||||
|
query.append(limit);
|
||||||
|
query.append(" ROWS ONLY");
|
||||||
|
}
|
||||||
|
|
||||||
|
return query.toString();
|
||||||
|
}
|
||||||
|
}
|
|
@ -14,5 +14,6 @@
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
org.apache.nifi.processors.standard.db.impl.GenericDatabaseAdapter
|
org.apache.nifi.processors.standard.db.impl.GenericDatabaseAdapter
|
||||||
org.apache.nifi.processors.standard.db.impl.OracleDatabaseAdapter
|
org.apache.nifi.processors.standard.db.impl.OracleDatabaseAdapter
|
||||||
|
org.apache.nifi.processors.standard.db.impl.Oracle12DatabaseAdapter
|
||||||
org.apache.nifi.processors.standard.db.impl.MSSQLDatabaseAdapter
|
org.apache.nifi.processors.standard.db.impl.MSSQLDatabaseAdapter
|
||||||
org.apache.nifi.processors.standard.db.impl.MSSQL2008DatabaseAdapter
|
org.apache.nifi.processors.standard.db.impl.MSSQL2008DatabaseAdapter
|
|
@ -0,0 +1,70 @@
|
||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
||||||
|
* (the "License"); you may not use this file except in compliance with
|
||||||
|
* the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.apache.nifi.processors.standard.db.impl;
|
||||||
|
|
||||||
|
import org.apache.nifi.processors.standard.db.DatabaseAdapter;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
public class TestOracle12DatabaseAdapter {
|
||||||
|
|
||||||
|
private final DatabaseAdapter db = new Oracle12DatabaseAdapter();
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGeneration() throws Exception {
|
||||||
|
String sql1 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","","",null,null);
|
||||||
|
String expected1 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename";
|
||||||
|
Assert.assertEquals(sql1,expected1);
|
||||||
|
|
||||||
|
String sql2 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","that=\'some\"\' value\'","",null,null);
|
||||||
|
String expected2 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename WHERE that=\'some\"\' value\'";
|
||||||
|
Assert.assertEquals(sql2,expected2);
|
||||||
|
|
||||||
|
String sql3 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","that=\'some\"\' value\'","might DESC",null,null);
|
||||||
|
String expected3 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename WHERE that=\'some\"\' value\' ORDER BY might DESC";
|
||||||
|
Assert.assertEquals(sql3,expected3);
|
||||||
|
|
||||||
|
String sql4 = db.getSelectStatement("database.tablename", "","that=\'some\"\' value\'","might DESC",null,null);
|
||||||
|
String expected4 = "SELECT * FROM database.tablename WHERE that=\'some\"\' value\' ORDER BY might DESC";
|
||||||
|
Assert.assertEquals(sql4,expected4);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test(expected = IllegalArgumentException.class)
|
||||||
|
public void testNoTableName() throws Exception {
|
||||||
|
db.getSelectStatement("", "some(set),of(columns),that,might,contain,methods,a.*","","",null,null);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testPagingQuery() throws Exception {
|
||||||
|
String sql1 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","","contain",100L,0L);
|
||||||
|
String expected1 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename ORDER BY contain FETCH NEXT 100 ROWS ONLY";
|
||||||
|
Assert.assertEquals(sql1,expected1);
|
||||||
|
|
||||||
|
String sql2 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","","contain",10000L,123456L);
|
||||||
|
String expected2 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename ORDER BY contain OFFSET 123456 ROWS FETCH NEXT 10000 ROWS ONLY";
|
||||||
|
Assert.assertEquals(sql2,expected2);
|
||||||
|
|
||||||
|
String sql3 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","methods='strange'","contain",10000L,123456L);
|
||||||
|
String expected3 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename WHERE methods='strange' ORDER BY contain OFFSET 123456 ROWS FETCH NEXT 10000 ROWS ONLY";
|
||||||
|
Assert.assertEquals(sql3,expected3);
|
||||||
|
|
||||||
|
String sql4 = db.getSelectStatement("database.tablename", "some(set),of(columns),that,might,contain,methods,a.*","","",100L,null);
|
||||||
|
String expected4 = "SELECT some(set),of(columns),that,might,contain,methods,a.* FROM database.tablename FETCH NEXT 100 ROWS ONLY";
|
||||||
|
Assert.assertEquals(sql4,expected4);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue