Nicolas Cannasse 14 年之前
父节点
当前提交
40164c64bc
共有 1 个文件被更改,包括 221 次插入0 次删除
  1. 221 0
      std/php/_std/sys/db/Mysql.hx

+ 221 - 0
std/php/_std/sys/db/Mysql.hx

@@ -0,0 +1,221 @@
+/*
+ * Copyright (c) 2005, The haXe Project Contributors
+ * All rights reserved.
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ *   - Redistributions of source code must retain the above copyright
+ *     notice, this list of conditions and the following disclaimer.
+ *   - Redistributions in binary form must reproduce the above copyright
+ *     notice, this list of conditions and the following disclaimer in the
+ *     documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE HAXE PROJECT CONTRIBUTORS "AS IS" AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE HAXE PROJECT CONTRIBUTORS BE LIABLE FOR
+ * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
+ * DAMAGE.
+ */
+package sys.db;
+
+import sys.db.Connection;
+
+private class MysqlConnection implements Connection {
+
+	var c : Void;
+
+	public function new( c : Void) {
+		this.c = c;
+	}
+
+	public function close() {
+		untyped __call__("mysql_close", c);
+		untyped __call__("unset", c);
+	}
+
+	public function request( s : String ) : ResultSet {
+		var h = untyped __call__("mysql_query", s, c);
+		if(untyped __physeq__(h, false))
+			throw "Error while executing "+s+" ("+untyped __call__("mysql_error", c)+")";
+		return new MysqlResultSet(cast h, cast c);
+	}
+
+	public function escape( s : String ) {
+		return untyped __call__("mysql_real_escape_string", s, c);
+	}
+
+	public function quote( s : String ) {
+		return "'" + untyped __call__("mysql_real_escape_string", s, c) + "'";
+	}
+
+	public function addValue( s : StringBuf, v : Dynamic ) {
+		if( untyped __call__("is_int", v) || __call__("is_null", v))
+			s.add(v);
+		else if( untyped __call__("is_bool", v) )
+			s.add(if( v ) 1 else 0);
+		else
+			s.add(quote(Std.string(v)));
+	}
+
+	public function lastInsertId() {
+		return untyped __call__("mysql_insert_id", c);
+	}
+
+	public function dbName() {
+		return "MySQL";
+	}
+
+	public function startTransaction() {
+		request("START TRANSACTION");
+	}
+
+	public function commit() {
+		request("COMMIT");
+	}
+
+	public function rollback() {
+		request("ROLLBACK");
+	}
+}
+
+
+private class MysqlResultSet implements ResultSet {
+	public var length(getLength,null) : Int;
+	public var nfields(getNFields,null) : Int;
+	private var __r : Void;
+	private var __c : Void;
+	private var cache : Dynamic;
+
+	public function new(r, c) {
+		__r = r;
+		__c = c;
+	}
+
+	private function getLength() {
+		if(untyped __physeq__(__r, true))
+			return untyped __call__("mysql_affected_rows", __c);
+		else if (untyped __physeq__(__r, false))
+			return 0;
+		return untyped __call__("mysql_num_rows", __r);
+	}
+
+	private var _nfields : Int;
+	private function getNFields() {
+		if(_nfields == null)
+			_nfields = untyped __call__("mysql_num_fields", __r);
+		return _nfields;
+	}
+
+	private var _fieldsDesc : Array<Dynamic>;
+	private function getFieldsDescription() {
+		if(_fieldsDesc == null) {
+			_fieldsDesc = [];
+			for (i in 0...nfields) {
+				var item = {
+					name : untyped __call__("mysql_field_name", __r, i),
+					type : untyped __call__("mysql_field_type", __r, i)
+				};
+				_fieldsDesc.push(item);
+			}
+		}
+		return _fieldsDesc;
+	}
+
+	private function convert(v : String, type : String) : Dynamic {
+		if (v == null) return v;
+		switch(type) {
+			case "int", "year":
+				return untyped __call__("intval", v);
+			case "real":
+				return untyped __call__("floatval", v);
+			case "datetime", "date":
+				return Date.fromString(v);
+			default:
+				return v;
+		}
+	}
+
+	public function hasNext() {
+		if( cache == null )
+			cache = next();
+		return (cache != null);
+	}
+
+	private var cRow : ArrayAccess<String>;
+	private function fetchRow() : Bool {
+		cRow = untyped __call__("mysql_fetch_array", __r, __php__("MYSQL_NUM"));
+		return ! untyped __physeq__(cRow, false);
+	}
+
+	public function next() : Dynamic {
+		if( cache != null ) {
+			var t = cache;
+			cache = null;
+			return t;
+		}
+		if(!fetchRow()) return null;
+
+		var o : Dynamic = {};
+		var descriptions = getFieldsDescription();
+		for(i in 0...nfields)
+			Reflect.setField(o, descriptions[i].name, convert(cRow[i], descriptions[i].type));
+		return o;
+	}
+
+	public function results() : List<Dynamic> {
+		var l = new List();
+		while( hasNext() )
+			l.add(next());
+		return l;
+	}
+
+	public function getResult( n : Int ) : String {
+		if(cRow == null)
+			if(!fetchRow())
+				return null;
+		return cRow[n];
+	}
+
+	public function getIntResult( n : Int ) : Int {
+		return untyped __call__("intval", getResult(n));
+	}
+
+	public function getFloatResult( n : Int ) : Float {
+		return untyped __call__("floatval", getResult(n));
+	}
+
+	public function getFieldsNames() : Array<String> {
+		var fields = [];
+		for( i in 0...nfields )
+			fields.push(untyped __call__("mysql_field_name", __r, i));
+		return fields;
+	}
+
+}
+
+@:core_api class Mysql {
+
+	public static function connect( params : {
+		host : String,
+		port : Int,
+		user : String,
+		pass : String,
+		socket : String,
+		database : String
+	} ) : sys.db.Connection {
+		var c = untyped __call__("mysql_connect",
+			params.host + (params.port == null ? '' : ':'+params.port) + (params.socket == null ? '' : ':'+params.socket),
+			params.user,
+			params.pass);
+		if(!untyped __call__("mysql_select_db", params.database, c))
+			throw "Unable to connect to " + params.database;
+		return new MysqlConnection(c);
+	}
+
+}