Hey all,
I am working on a UDF that can be used with prepared statements ad a
technique to go from Hive->SQL. The usage would be something like
from src SELECT
dboutput("jdbc:mysql://localhost:20","user","pass","INSERT INTO A
(col1,col2) VALUES (?,?)", key,value)
--Sniplet--
public class GenericUDFDBOutput extends GenericUDF {
private static Log LOG =
LogFactory.getLog(GenericUDFDBOutput.class.getName());
ObjectInspector[] argumentOIs;
GenericUDFUtils.ReturnObjectInspectorResolver returnOIResolver;
Connection connection = null;
private String url;
private String user;
private String pass;
private IntWritable result = new IntWritable(-1);
public ObjectInspector initialize(ObjectInspector[] arguments)
throws UDFArgumentTypeException {
this.argumentOIs = arguments;
public Object evaluate(DeferredObject[] arguments) throws HiveException {
System.err.println("we are here");
url = arguments[0].get().toString()+";create=true";
url = "jdbc:derby:;databaseName=test_dboutput_db;create=true";
user = arguments[1].get().toString();
pass = arguments[2].get().toString();
try {
try {
Class.forName("org.apache.derby.jdbc.EmbeddedDriver").newInstance();
} catch (InstantiationException e) {
System.out.println(e);
} catch (IllegalAccessException e) {
System.out.println(e);
} catch (ClassNotFoundException e) {
System.out.println(e);
}
connection = DriverManager.getConnection(url, user, pass);
PreparedStatement ps = connection.prepareStatement(arguments[3].get()
.toString());
for (int i = 4; i < arguments.length; ++i) {
PrimitiveObjectInspector poi = ((PrimitiveObjectInspector)
arguments[i]);
ps.setObject(i - 3, poi.getPrimitiveJavaObject(arguments[i]));
}
ps.execute();
ps.close();
--end snip--
In my test case I am attempting to use the derby embedded driver since
I figured it would probably already be in the classpath.
org.apache.derby.jdbc.EmbeddedDriver. It is not. I took several stabs
and adding it to the ant files but I get the feeling that the
TestCliDriver is not inheriting those paths. Any suggestions?
Thanks
