Mysql.hx 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234
  1. /*
  2. * Copyright (c) 2005, The haXe Project Contributors
  3. * All rights reserved.
  4. * Redistribution and use in source and binary forms, with or without
  5. * modification, are permitted provided that the following conditions are met:
  6. *
  7. * - Redistributions of source code must retain the above copyright
  8. * notice, this list of conditions and the following disclaimer.
  9. * - Redistributions in binary form must reproduce the above copyright
  10. * notice, this list of conditions and the following disclaimer in the
  11. * documentation and/or other materials provided with the distribution.
  12. *
  13. * THIS SOFTWARE IS PROVIDED BY THE HAXE PROJECT CONTRIBUTORS "AS IS" AND ANY
  14. * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  15. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  16. * DISCLAIMED. IN NO EVENT SHALL THE HAXE PROJECT CONTRIBUTORS BE LIABLE FOR
  17. * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  18. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  19. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  20. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  21. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  22. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
  23. * DAMAGE.
  24. */
  25. package neko.db;
  26. #if !spod_macro
  27. import neko.db.Connection;
  28. private class D {
  29. static function load(fun,args) : Dynamic {
  30. return neko.Lib.load(lib,fun,args);
  31. }
  32. static var lib = try { neko.Lib.load("mysql5","connect",1); "mysql5"; } catch( e : Dynamic ) "mysql";
  33. public static var connect = load("connect",1);
  34. public static var select_db = load("select_db",2);
  35. public static var request = load("request",2);
  36. public static var close = load("close",1);
  37. public static var escape = load("escape",2);
  38. public static var result_get_length = load("result_get_length",1);
  39. public static var result_get_nfields = load("result_get_nfields",1);
  40. public static var result_next = load("result_next",1);
  41. public static var result_get = load("result_get",2);
  42. public static var result_get_int = load("result_get_int",2);
  43. public static var result_get_float = load("result_get_float",2);
  44. public static var result_set_conv_date = load("result_set_conv_date",2);
  45. public static var result_fields_names = neko.Lib.loadLazy(lib,"result_get_fields_names",1);
  46. }
  47. private class MysqlResultSet implements ResultSet {
  48. public var length(getLength,null) : Int;
  49. public var nfields(getNFields,null) : Int;
  50. private var __r : Void;
  51. private var cache : Dynamic;
  52. public function new(r) {
  53. __r = r;
  54. }
  55. private function getLength() {
  56. return D.result_get_length(__r);
  57. }
  58. private function getNFields() {
  59. return D.result_get_nfields(__r);
  60. }
  61. public function hasNext() {
  62. if( cache == null )
  63. cache = next();
  64. return (cache != null);
  65. }
  66. public function next() : Dynamic {
  67. var c = cache;
  68. if( c != null ) {
  69. cache = null;
  70. return c;
  71. }
  72. c = D.result_next(__r);
  73. if( c == null )
  74. return null;
  75. untyped {
  76. var f = __dollar__objfields(c);
  77. var i = 0;
  78. var l = __dollar__asize(f);
  79. while( i < l ) {
  80. var v = __dollar__objget(c,f[i]);
  81. if( __dollar__typeof(v) == __dollar__tstring )
  82. __dollar__objset(c,f[i],new String(v));
  83. i = i + 1;
  84. }
  85. }
  86. return c;
  87. }
  88. public function results() : List<Dynamic> {
  89. var l = new List();
  90. while( hasNext() )
  91. l.add(next());
  92. return l;
  93. }
  94. public function getResult( n : Int ) {
  95. return new String(D.result_get(__r,n));
  96. }
  97. public function getIntResult( n : Int ) : Int {
  98. return D.result_get_int(__r,n);
  99. }
  100. public function getFloatResult( n : Int ) : Float {
  101. return D.result_get_float(__r,n);
  102. }
  103. public function getFieldsNames() : Array<String> {
  104. var a = D.result_fields_names(__r);
  105. untyped {
  106. var i = 0;
  107. var l = __dollar__asize(a);
  108. while( i < l ) {
  109. a[i] = new String(a[i]);
  110. i += 1;
  111. }
  112. a = Array.new1(cast a,l);
  113. }
  114. return a;
  115. }
  116. }
  117. private class MysqlConnection implements Connection {
  118. private var __c : Void;
  119. public function new(c) {
  120. __c = c;
  121. }
  122. public function request( s : String ) : ResultSet {
  123. try {
  124. var r = D.request(this.__c,untyped s.__s);
  125. D.result_set_conv_date(r,function(d) { return untyped Date.new1(d); });
  126. return new MysqlResultSet(r);
  127. } catch( e : Dynamic ) {
  128. untyped if( __dollar__typeof(e) == __dollar__tobject && __dollar__typeof(e.msg) == __dollar__tstring )
  129. e = e.msg;
  130. untyped __dollar__rethrow(e);
  131. return null;
  132. }
  133. }
  134. public function close() {
  135. D.close(__c);
  136. }
  137. public function escape( s : String ) {
  138. return new String(D.escape(__c,untyped s.__s));
  139. }
  140. public function quote( s : String ) {
  141. return "'"+escape(s)+"'";
  142. }
  143. public function addValue( s : StringBuf, v : Dynamic ) {
  144. var t = untyped __dollar__typeof(v);
  145. if( untyped (t == __dollar__tint || t == __dollar__tnull) )
  146. s.add(v);
  147. else if( untyped t == __dollar__tbool )
  148. s.addChar(if( v ) "1".code else "0".code);
  149. else {
  150. s.addChar("'".code);
  151. s.add(escape(Std.string(v)));
  152. s.addChar("'".code);
  153. }
  154. }
  155. public function lastInsertId() {
  156. return request("SELECT LAST_INSERT_ID()").getIntResult(0);
  157. }
  158. public function dbName() {
  159. return "MySQL";
  160. }
  161. public function startTransaction() {
  162. request("START TRANSACTION");
  163. }
  164. public function commit() {
  165. request("COMMIT");
  166. }
  167. public function rollback() {
  168. request("ROLLBACK");
  169. }
  170. private static var __use_date = Date;
  171. }
  172. class Mysql {
  173. public static function connect( params : {
  174. host : String,
  175. port : Int,
  176. user : String,
  177. pass : String,
  178. socket : String,
  179. database : String
  180. } ) : neko.db.Connection {
  181. var o = untyped {
  182. host : params.host.__s,
  183. port : params.port,
  184. user : params.user.__s,
  185. pass : params.pass.__s,
  186. socket : if( params.socket == null ) null else params.socket.__s
  187. };
  188. var c = D.connect(o);
  189. try {
  190. D.select_db(c,untyped params.database.__s);
  191. } catch( e : Dynamic ) {
  192. D.close(c);
  193. neko.Lib.rethrow(e);
  194. }
  195. return new MysqlConnection(c);
  196. }
  197. }
  198. #end