org.duckdb.DuckDBNative Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of duckdb_jdbc Show documentation
A JDBC-Compliant driver for the DuckDB data management system
There is a newer version: 1.1.3
package org.duckdb;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.nio.ByteBuffer;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;

public class DuckDBNative {
	static {
		try {
			String os_name = "";
			String os_arch = "";
			String os_name_detect = System.getProperty("os.name").toLowerCase().trim();
			String os_arch_detect = System.getProperty("os.arch").toLowerCase().trim();
			if (os_arch_detect.equals("x86_64") || os_arch_detect.equals("amd64")) {
				os_arch = "amd64";
			}
            if (os_arch_detect.equals("aarch64") || os_arch_detect.equals("arm64")) {
                os_arch = "arm64";
            }
			// TODO 32 bit gunk

			if (os_name_detect.startsWith("windows")) {
				os_name = "windows";
			} else if (os_name_detect.startsWith("mac")) {
				os_name = "osx";
			} else if (os_name_detect.startsWith("linux")) {
				os_name = "linux";
			}
			String lib_res_name = "/libduckdb_java.so" + "_" + os_name + "_" + os_arch;

			Path lib_file = Files.createTempFile("libduckdb_java", ".so");
			URL lib_res = DuckDBNative.class.getResource(lib_res_name);
			if (lib_res == null) {
				throw new IOException(lib_res_name + " not found");
			}
			Files.copy(lib_res.openStream(), lib_file, StandardCopyOption.REPLACE_EXISTING);
			new File(lib_file.toString()).deleteOnExit();
			System.load(lib_file.toAbsolutePath().toString());
		} catch (IOException e) {
			e.printStackTrace();
		}

	}
	// We use zero-length ByteBuffer-s as a hacky but cheap way to pass C++ pointers
	// back and forth

	/*
	 * NB: if you change anything below, run `javah` on this class to re-generate
	 * the C header. CMake does this as well
	 */

	// results db_ref database reference object
	protected static native ByteBuffer duckdb_jdbc_startup(byte[] path, boolean read_only);

	protected static native void duckdb_jdbc_shutdown(ByteBuffer db_ref);

	// returns conn_ref connection reference object
	protected static native ByteBuffer duckdb_jdbc_connect(ByteBuffer db_ref);

	protected static native void duckdb_jdbc_set_auto_commit(ByteBuffer conn_ref, boolean auto_commit);

	protected static native boolean duckdb_jdbc_get_auto_commit(ByteBuffer conn_ref);

	protected static native void duckdb_jdbc_disconnect(ByteBuffer conn_ref);

	// returns stmt_ref result reference object
	protected static native ByteBuffer duckdb_jdbc_prepare(ByteBuffer conn_ref, byte[] query);

	protected static native String duckdb_jdbc_prepare_type(ByteBuffer stmt_ref);

	protected static native void duckdb_jdbc_release(ByteBuffer stmt_ref);

	protected static native DuckDBResultSetMetaData duckdb_jdbc_meta(ByteBuffer stmt_ref);

	
	// returns res_ref result reference object
	protected static native ByteBuffer duckdb_jdbc_execute(ByteBuffer stmt_ref, Object[] params);


	protected static native void duckdb_jdbc_free_result(ByteBuffer res_ref);

	protected static native DuckDBVector[] duckdb_jdbc_fetch(ByteBuffer res_ref);
	
	protected static native int duckdb_jdbc_fetch_size();

	protected static native ByteBuffer duckdb_jdbc_create_appender(ByteBuffer conn_ref, byte[] schema_name, byte[] table_name);

	protected static native void duckdb_jdbc_appender_begin_row(ByteBuffer appender_ref);

	protected static native void duckdb_jdbc_appender_end_row(ByteBuffer appender_ref);

	protected static native void duckdb_jdbc_appender_flush(ByteBuffer appender_ref);

	protected static native void duckdb_jdbc_appender_close(ByteBuffer appender_ref);

	protected static native void duckdb_jdbc_appender_append_boolean(ByteBuffer appender_ref, boolean value);

	protected static native void duckdb_jdbc_appender_append_byte(ByteBuffer appender_ref, byte value);

	protected static native void duckdb_jdbc_appender_append_short(ByteBuffer appender_ref, short value);

	protected static native void duckdb_jdbc_appender_append_int(ByteBuffer appender_ref, int value);

	protected static native void duckdb_jdbc_appender_append_long(ByteBuffer appender_ref, long value);

	protected static native void duckdb_jdbc_appender_append_float(ByteBuffer appender_ref, float value);

	protected static native void duckdb_jdbc_appender_append_double(ByteBuffer appender_ref, double value);

	protected static native void duckdb_jdbc_appender_append_string(ByteBuffer appender_ref, byte[] value);

}