forked from LeenkxTeam/LNXSDK
Update Files
This commit is contained in:
318
Kha/Tools/linux_arm64/std/java/db/Jdbc.hx
Normal file
318
Kha/Tools/linux_arm64/std/java/db/Jdbc.hx
Normal file
@ -0,0 +1,318 @@
|
||||
/*
|
||||
* Copyright (C)2005-2019 Haxe Foundation
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a
|
||||
* copy of this software and associated documentation files (the "Software"),
|
||||
* to deal in the Software without restriction, including without limitation
|
||||
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
||||
* and/or sell copies of the Software, and to permit persons to whom the
|
||||
* Software is furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in
|
||||
* all copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||
* DEALINGS IN THE SOFTWARE.
|
||||
*/
|
||||
|
||||
package java.db;
|
||||
|
||||
import haxe.io.Bytes;
|
||||
import java.sql.Types;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
|
||||
@:native('haxe.java.db.Jdbc')
|
||||
class Jdbc {
|
||||
public static function create(cnx:java.sql.Connection):sys.db.Connection {
|
||||
return new JdbcConnection(cnx);
|
||||
}
|
||||
}
|
||||
|
||||
@:native('haxe.java.db.JdbcConnection')
|
||||
private class JdbcConnection implements sys.db.Connection {
|
||||
private static var ids = new AtomicInteger(0);
|
||||
|
||||
private var id:Int;
|
||||
|
||||
private var cnx:java.sql.Connection;
|
||||
private var _lastInsertId:Int;
|
||||
// escape handling
|
||||
private var escapeRegex:EReg;
|
||||
private var escapes:Array<Dynamic>;
|
||||
|
||||
public function new(cnx) {
|
||||
this.id = ids.getAndIncrement();
|
||||
this.cnx = cnx;
|
||||
this.escapes = [];
|
||||
this.escapeRegex = ~/@@HX_ESCAPE(\d+)_(\d+)@@/;
|
||||
}
|
||||
|
||||
public function close() {
|
||||
try
|
||||
this.cnx.close()
|
||||
catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
|
||||
public function escape(s:String):String {
|
||||
return "@@HX_ESCAPE" + id + "_" + escapes.push(s) + "@@";
|
||||
}
|
||||
|
||||
public function quote(s:String):String {
|
||||
return "@@HX_ESCAPE" + id + "_" + escapes.push(s) + "@@";
|
||||
}
|
||||
|
||||
public function addValue(s:StringBuf, v:Dynamic) {
|
||||
if (Std.isOfType(v, Date)) {
|
||||
v = Std.string(v);
|
||||
} else if (Std.isOfType(v, Bytes)) {
|
||||
var bt:Bytes = v;
|
||||
v = bt.getData();
|
||||
}
|
||||
s.add("@@HX_ESCAPE");
|
||||
s.add(id);
|
||||
s.add("_");
|
||||
s.add(escapes.push(v));
|
||||
s.add("@@");
|
||||
}
|
||||
|
||||
public function lastInsertId():Int {
|
||||
return _lastInsertId;
|
||||
}
|
||||
|
||||
public function dbName():String {
|
||||
try {
|
||||
var ret = cnx.getMetaData().getDriverName();
|
||||
var retc = ret.toLowerCase();
|
||||
if (retc.indexOf("mysql") != -1)
|
||||
return "MySQL";
|
||||
else if (retc.indexOf("sqlite") != -1)
|
||||
return "SQLite";
|
||||
return ret;
|
||||
} catch (e:Dynamic) {
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
public function startTransaction() {
|
||||
try {
|
||||
cnx.setAutoCommit(false);
|
||||
} catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
|
||||
public function commit() {
|
||||
try {
|
||||
cnx.commit();
|
||||
} catch (e:Dynamic) {
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
public function rollback() {
|
||||
try
|
||||
cnx.rollback()
|
||||
catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
|
||||
public function request(s:String):sys.db.ResultSet {
|
||||
var newst = new StringBuf();
|
||||
var sentArray = [];
|
||||
|
||||
// cycle through the request string, adding any @@HX_ESCAPE@@ reference to the sentArray
|
||||
var r = escapeRegex;
|
||||
var myid = id + "", escapes = escapes, elen = escapes.length;
|
||||
try {
|
||||
while (r.match(s)) {
|
||||
var id = r.matched(1);
|
||||
if (id != myid)
|
||||
throw "Request quotes are only valid for one single request; They can't be cached.";
|
||||
|
||||
newst.add(r.matchedLeft());
|
||||
var eid = Std.parseInt(r.matched(2));
|
||||
if (eid == null || eid > elen)
|
||||
throw "Invalid request quote ID " + eid;
|
||||
sentArray.push(escapes[eid - 1]);
|
||||
newst.add("?");
|
||||
s = r.matchedRight();
|
||||
}
|
||||
newst.add(s);
|
||||
var stmt = cnx.prepareStatement(newst.toString(),
|
||||
#if jvm java.sql.Statement.RETURN_GENERATED_KEYS #else java.sql.Statement.Statement_Statics.RETURN_GENERATED_KEYS #end);
|
||||
for (i in 0...sentArray.length) {
|
||||
stmt.setObject(i + 1, sentArray[i]);
|
||||
}
|
||||
|
||||
var ret = null, dbName = dbName();
|
||||
if (stmt.execute()) {
|
||||
// is a result set
|
||||
var rs = stmt.getResultSet();
|
||||
ret = new JdbcResultSet(rs, dbName, stmt.getMetaData());
|
||||
} else {
|
||||
// is an update
|
||||
var affected = stmt.getUpdateCount();
|
||||
if (affected == 1) {
|
||||
var autogen = stmt.getGeneratedKeys();
|
||||
if (autogen.next()) {
|
||||
this._lastInsertId = autogen.getInt(1);
|
||||
}
|
||||
}
|
||||
ret = new JdbcResultSet(null, dbName, null);
|
||||
}
|
||||
|
||||
if (escapes.length != 0)
|
||||
escapes = [];
|
||||
this.id = ids.getAndIncrement();
|
||||
return ret;
|
||||
} catch (e:Dynamic) {
|
||||
if (escapes.length != 0)
|
||||
escapes = [];
|
||||
this.id = ids.getAndIncrement();
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@:native('haxe.java.db.JdbcResultSet')
|
||||
private class JdbcResultSet implements sys.db.ResultSet {
|
||||
@:isVar public var length(get, null):Int;
|
||||
public var nfields(get, null):Int;
|
||||
|
||||
private var rs:java.sql.ResultSet;
|
||||
private var names:Array<String>;
|
||||
private var types:java.NativeArray<Int>;
|
||||
private var dbName:String;
|
||||
private var didNext:Bool;
|
||||
|
||||
public function new(rs, dbName, meta:java.sql.ResultSetMetaData) {
|
||||
this.dbName = dbName;
|
||||
this.rs = rs;
|
||||
if (meta != null) {
|
||||
try {
|
||||
var count = meta.getColumnCount();
|
||||
var names = [], types = new NativeArray(count);
|
||||
for (i in 0...count) {
|
||||
names.push(meta.getColumnName(i + 1));
|
||||
types[i] = meta.getColumnType(i + 1);
|
||||
}
|
||||
this.types = types;
|
||||
this.names = names;
|
||||
} catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
private function get_length():Int {
|
||||
if (length == 0) {
|
||||
try {
|
||||
var cur = rs.getRow();
|
||||
rs.last();
|
||||
this.length = rs.getRow();
|
||||
rs.absolute(cur);
|
||||
} catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
return length;
|
||||
}
|
||||
|
||||
private function get_nfields():Int {
|
||||
return names == null ? 0 : names.length;
|
||||
}
|
||||
|
||||
public function hasNext():Bool {
|
||||
try {
|
||||
didNext = true;
|
||||
return rs != null && rs.next();
|
||||
} catch (e:Dynamic) {
|
||||
return throw e;
|
||||
}
|
||||
}
|
||||
|
||||
public function next():Dynamic {
|
||||
try {
|
||||
if (rs == null)
|
||||
return null;
|
||||
if (didNext) {
|
||||
didNext = false;
|
||||
} else {
|
||||
if (!rs.next()) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
var ret = {}, names = names, types = types;
|
||||
for (i in 0...names.length) {
|
||||
var name = names[i], t = types[i], val:Dynamic = null;
|
||||
if (t == Types.FLOAT) {
|
||||
val = rs.getDouble(i + 1);
|
||||
} else if (t == Types.DATE || t == Types.TIME) {
|
||||
if (dbName == "SQLite") {
|
||||
var str = rs.getString(i + 1);
|
||||
if (str != null) {
|
||||
var d:Date = Date.fromString(str);
|
||||
val = d;
|
||||
}
|
||||
} else {
|
||||
var d:java.sql.Date = rs.getDate(i + 1);
|
||||
if (d != null)
|
||||
val = Date.fromTime(cast d.getTime());
|
||||
}
|
||||
} else if (t == Types.LONGVARBINARY || t == Types.VARBINARY || t == Types.BINARY || t == Types.BLOB) {
|
||||
var b = rs.getBytes(i + 1);
|
||||
if (b != null)
|
||||
val = Bytes.ofData(b);
|
||||
} else {
|
||||
val = rs.getObject(i + 1);
|
||||
}
|
||||
Reflect.setField(ret, name, val);
|
||||
}
|
||||
return ret;
|
||||
} catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
|
||||
public function results():List<Dynamic> {
|
||||
var l = new List();
|
||||
if (rs == null)
|
||||
return l;
|
||||
|
||||
try {
|
||||
while (hasNext())
|
||||
l.add(next());
|
||||
} catch (e:Dynamic)
|
||||
throw e;
|
||||
return l;
|
||||
}
|
||||
|
||||
public function getResult(n:Int):String {
|
||||
try {
|
||||
return rs.getString(n);
|
||||
} catch (e:Dynamic)
|
||||
throw e;
|
||||
}
|
||||
|
||||
public function getIntResult(n:Int):Int {
|
||||
try {
|
||||
return rs.getInt(n);
|
||||
} catch (e:Dynamic) {
|
||||
return throw e;
|
||||
};
|
||||
}
|
||||
|
||||
public function getFloatResult(n:Int):Float {
|
||||
try {
|
||||
return rs.getFloat(n);
|
||||
} catch (e:Dynamic) {
|
||||
return throw e;
|
||||
};
|
||||
}
|
||||
|
||||
public function getFieldsNames():Null<Array<String>> {
|
||||
return this.names;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user