2
0

Manager.hx 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743
  1. /*
  2. * Copyright (C)2005-2015 Haxe Foundation
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a
  5. * copy of this software and associated documentation files (the "Software"),
  6. * to deal in the Software without restriction, including without limitation
  7. * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  8. * and/or sell copies of the Software, and to permit persons to whom the
  9. * Software is furnished to do so, subject to the following conditions:
  10. *
  11. * The above copyright notice and this permission notice shall be included in
  12. * all copies or substantial portions of the Software.
  13. *
  14. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  19. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  20. * DEALINGS IN THE SOFTWARE.
  21. */
  22. package sys.db;
  23. import Reflect;
  24. import sys.db.Connection;
  25. import sys.db.RecordInfos;
  26. /**
  27. Record Manager : the persistent object database manager. See the tutorial on
  28. Haxe website to learn how to use Record.
  29. **/
  30. #if !macro @:build(sys.db.RecordMacros.addRtti()) #end
  31. class Manager<T : Object> {
  32. /* ----------------------------- STATICS ------------------------------ */
  33. public static var cnx(default, set) : Connection;
  34. public static var lockMode : String;
  35. private static inline var cache_field = "__cache__";
  36. private static var object_cache : haxe.ds.StringMap<Object> = new haxe.ds.StringMap();
  37. private static var init_list : List<Manager<Dynamic>> = new List();
  38. private static var KEYWORDS = {
  39. var h = new haxe.ds.StringMap();
  40. for( k in "ADD|ALL|ALTER|ANALYZE|AND|AS|ASC|ASENSITIVE|BEFORE|BETWEEN|BIGINT|BINARY|BLOB|BOTH|BY|CALL|CASCADE|CASE|CHANGE|CHAR|CHARACTER|CHECK|COLLATE|COLUMN|CONDITION|CONSTRAINT|CONTINUE|CONVERT|CREATE|CROSS|CURRENT_DATE|CURRENT_TIME|CURRENT_TIMESTAMP|CURRENT_USER|CURSOR|DATABASE|DATABASES|DAY_HOUR|DAY_MICROSECOND|DAY_MINUTE|DAY_SECOND|DEC|DECIMAL|DECLARE|DEFAULT|DELAYED|DELETE|DESC|DESCRIBE|DETERMINISTIC|DISTINCT|DISTINCTROW|DIV|DOUBLE|DROP|DUAL|EACH|ELSE|ELSEIF|ENCLOSED|ESCAPED|EXISTS|EXIT|EXPLAIN|FALSE|FETCH|FLOAT|FLOAT4|FLOAT8|FOR|FORCE|FOREIGN|FROM|FULLTEXT|GRANT|GROUP|HAVING|HIGH_PRIORITY|HOUR_MICROSECOND|HOUR_MINUTE|HOUR_SECOND|IF|IGNORE|IN|INDEX|INFILE|INNER|INOUT|INSENSITIVE|INSERT|INT|INT1|INT2|INT3|INT4|INT8|INTEGER|INTERVAL|INTO|IS|ITERATE|JOIN|KEY|KEYS|KILL|LEADING|LEAVE|LEFT|LIKE|LIMIT|LINES|LOAD|LOCALTIME|LOCALTIMESTAMP|LOCK|LONG|LONGBLOB|LONGTEXT|LOOP|LOW_PRIORITY|MATCH|MEDIUMBLOB|MEDIUMINT|MEDIUMTEXT|MIDDLEINT|MINUTE_MICROSECOND|MINUTE_SECOND|MOD|MODIFIES|NATURAL|NOT|NO_WRITE_TO_BINLOG|NULL|NUMERIC|ON|OPTIMIZE|OPTION|OPTIONALLY|OR|ORDER|OUT|OUTER|OUTFILE|PRECISION|PRIMARY|PROCEDURE|PURGE|READ|READS|REAL|REFERENCES|REGEXP|RELEASE|RENAME|REPEAT|REPLACE|REQUIRE|RESTRICT|RETURN|REVOKE|RIGHT|RLIKE|SCHEMA|SCHEMAS|SECOND_MICROSECOND|SELECT|SENSITIVE|SEPARATOR|SET|SHOW|SMALLINT|SONAME|SPATIAL|SPECIFIC|SQL|SQLEXCEPTION|SQLSTATE|SQLWARNING|SQL_BIG_RESULT|SQL_CALC_FOUND_ROWS|SQL_SMALL_RESULT|SSL|STARTING|STRAIGHT_JOIN|TABLE|TERMINATED|THEN|TINYBLOB|TINYINT|TINYTEXT|TO|TRAILING|TRIGGER|TRUE|UNDO|UNION|UNIQUE|UNLOCK|UNSIGNED|UPDATE|USAGE|USE|USING|UTC_DATE|UTC_TIME|UTC_TIMESTAMP|VALUES|VARBINARY|VARCHAR|VARCHARACTER|VARYING|WHEN|WHERE|WHILE|WITH|WRITE|XOR|YEAR_MONTH|ZEROFILL|ASENSITIVE|CALL|CONDITION|CONNECTION|CONTINUE|CURSOR|DECLARE|DETERMINISTIC|EACH|ELSEIF|EXIT|FETCH|GOTO|INOUT|INSENSITIVE|ITERATE|LABEL|LEAVE|LOOP|MODIFIES|OUT|READS|RELEASE|REPEAT|RETURN|SCHEMA|SCHEMAS|SENSITIVE|SPECIFIC|SQL|SQLEXCEPTION|SQLSTATE|SQLWARNING|TRIGGER|UNDO|UPGRADE|WHILE".split("|") )
  41. h.set(k.toLowerCase(),true);
  42. h;
  43. }
  44. private static function set_cnx( c : Connection ) {
  45. cnx = c;
  46. lockMode = (c != null && c.dbName() == "MySQL") ? " FOR UPDATE" : "";
  47. return c;
  48. }
  49. /* ---------------------------- BASIC API ----------------------------- */
  50. var table_infos : RecordInfos;
  51. var table_name : String;
  52. var table_keys : Array<String>;
  53. var class_proto : { prototype : Dynamic };
  54. public function new( classval : Class<T> ) {
  55. var m : Array<Dynamic> = haxe.rtti.Meta.getType(classval).rtti;
  56. if( m == null ) throw "Missing @rtti for class " + Type.getClassName(classval);
  57. table_infos = haxe.Unserializer.run(m[0]);
  58. table_name = quoteField(table_infos.name);
  59. table_keys = table_infos.key;
  60. // set the manager and ready for further init
  61. class_proto = cast classval;
  62. #if neko
  63. class_proto.prototype._manager = this;
  64. init_list.add(this);
  65. #end
  66. }
  67. public function all( ?lock: Bool ) : List<T> {
  68. return unsafeObjects("SELECT * FROM " + table_name,lock);
  69. }
  70. public macro function get(ethis,id,?lock:haxe.macro.Expr.ExprOf<Bool>) : #if macro haxe.macro.Expr #else haxe.macro.Expr.ExprOf<T> #end {
  71. return RecordMacros.macroGet(ethis,id,lock);
  72. }
  73. public macro function select(ethis, cond, ?options, ?lock:haxe.macro.Expr.ExprOf<Bool>) : #if macro haxe.macro.Expr #else haxe.macro.Expr.ExprOf<T> #end {
  74. return RecordMacros.macroSearch(ethis, cond, options, lock, true);
  75. }
  76. public macro function search(ethis, cond, ?options, ?lock:haxe.macro.Expr.ExprOf<Bool>) : #if macro haxe.macro.Expr #else haxe.macro.Expr.ExprOf<List<T>> #end {
  77. return RecordMacros.macroSearch(ethis, cond, options, lock);
  78. }
  79. public macro function count(ethis, cond) : #if macro haxe.macro.Expr #else haxe.macro.Expr.ExprOf<Int> #end {
  80. return RecordMacros.macroCount(ethis, cond);
  81. }
  82. public macro function delete(ethis, cond, ?options) : #if macro haxe.macro.Expr #else haxe.macro.Expr.ExprOf<Void> #end {
  83. return RecordMacros.macroDelete(ethis, cond, options);
  84. }
  85. public function dynamicSearch( x : {}, ?lock : Bool ) : List<T> {
  86. var s = new StringBuf();
  87. s.add("SELECT * FROM ");
  88. s.add(table_name);
  89. s.add(" WHERE ");
  90. addCondition(s,x);
  91. return unsafeObjects(s.toString(),lock);
  92. }
  93. function quote( s : String ) : String {
  94. return getCnx().quote( s );
  95. }
  96. /* -------------------------- RECORDOBJECT API -------------------------- */
  97. function doUpdateCache( x : T, name : String, v : Dynamic ) {
  98. var cache : { v : Dynamic } = Reflect.field(x, "cache_" + name);
  99. // if the cache has not been fetched (for instance if the field was set by reflection)
  100. // then we directly use the new value
  101. if( cache == null )
  102. return v;
  103. var v = doSerialize(name, cache.v);
  104. // don't set it since the value might change again later
  105. // Reflect.setField(x, name, v);
  106. return v;
  107. }
  108. static function getFieldName(field:RecordField):String
  109. {
  110. return switch (field.t) {
  111. case DData | DEnum(_):
  112. "data_" + field.name;
  113. case _:
  114. field.name;
  115. }
  116. }
  117. function doInsert( x : T ) {
  118. unmake(x);
  119. var s = new StringBuf();
  120. var fields = new List();
  121. var values = new List();
  122. var cache = Reflect.field(x,cache_field);
  123. if (cache == null)
  124. {
  125. Reflect.setField(x,cache_field,cache = {});
  126. }
  127. for( f in table_infos.fields ) {
  128. var name = f.name,
  129. fieldName = getFieldName(f);
  130. var v:Dynamic = Reflect.field(x,fieldName);
  131. if( v != null ) {
  132. fields.add(quoteField(name));
  133. switch( f.t ) {
  134. case DData: v = doUpdateCache(x, name, v);
  135. default:
  136. }
  137. values.add(v);
  138. } else if( !f.isNull ) {
  139. // if the field is not defined, give it a default value on insert
  140. switch( f.t ) {
  141. case DUInt, DTinyInt, DInt, DSingle, DFloat, DFlags(_), DBigInt, DTinyUInt, DSmallInt, DSmallUInt, DMediumInt, DMediumUInt, DEnum(_):
  142. Reflect.setField(x, fieldName, 0);
  143. case DBool:
  144. Reflect.setField(x, fieldName, false);
  145. case DTinyText, DText, DString(_), DSmallText, DSerialized:
  146. Reflect.setField(x, fieldName, "");
  147. case DSmallBinary, DNekoSerialized, DLongBinary, DBytes(_), DBinary:
  148. Reflect.setField(x, fieldName, haxe.io.Bytes.alloc(0));
  149. case DDate, DDateTime, DTimeStamp:
  150. // default date might depend on database
  151. case DId, DUId, DBigId, DNull, DInterval, DEncoded, DData:
  152. // no default value for these
  153. }
  154. }
  155. Reflect.setField(cache, name, v);
  156. }
  157. s.add("INSERT INTO ");
  158. s.add(table_name);
  159. if (fields.length > 0 || cnx.dbName() != "SQLite")
  160. {
  161. s.add(" (");
  162. s.add(fields.join(","));
  163. s.add(") VALUES (");
  164. var first = true;
  165. for( v in values ) {
  166. if( first )
  167. first = false;
  168. else
  169. s.add(", ");
  170. getCnx().addValue(s,v);
  171. }
  172. s.add(")");
  173. } else {
  174. s.add(" DEFAULT VALUES");
  175. }
  176. unsafeExecute(s.toString());
  177. untyped x._lock = true;
  178. // table with one key not defined : suppose autoincrement
  179. if( table_keys.length == 1 && Reflect.field(x,table_keys[0]) == null )
  180. Reflect.setField(x,table_keys[0],getCnx().lastInsertId());
  181. addToCache(x);
  182. }
  183. inline function isBinary( t : RecordInfos.RecordType ) {
  184. return switch( t ) {
  185. case DSmallBinary, DNekoSerialized, DLongBinary, DBytes(_), DBinary: true;
  186. //case DData: true // -- disabled for implementation purposes
  187. default: false;
  188. };
  189. }
  190. inline function hasBinaryChanged( a : haxe.io.Bytes, b : haxe.io.Bytes ) {
  191. return a != b && (a == null || b == null || a.compare(b) != 0);
  192. }
  193. function doUpdate( x : T ) {
  194. if( untyped !x._lock )
  195. throw "Cannot update a not locked object";
  196. var upd = getUpdateStatement(x);
  197. if (upd == null) return;
  198. unsafeExecute(upd);
  199. }
  200. function getUpdateStatement( x : T ):Null<String>
  201. {
  202. unmake(x);
  203. var s = new StringBuf();
  204. s.add("UPDATE ");
  205. s.add(table_name);
  206. s.add(" SET ");
  207. var cache = Reflect.field(x,cache_field);
  208. var mod = false;
  209. for( f in table_infos.fields ) {
  210. if (table_keys.indexOf(f.name) >= 0)
  211. continue;
  212. var name = f.name,
  213. fieldName = getFieldName(f);
  214. var v : Dynamic = Reflect.field(x,fieldName);
  215. var vc : Dynamic = Reflect.field(cache,name);
  216. if( cache == null || v != vc ) {
  217. switch( f.t ) {
  218. case DSmallBinary, DNekoSerialized, DLongBinary, DBytes(_), DBinary:
  219. if ( !hasBinaryChanged(v,vc) )
  220. continue;
  221. case DData:
  222. v = doUpdateCache(x, name, v);
  223. if( !hasBinaryChanged(v,vc) )
  224. continue;
  225. default:
  226. }
  227. if( mod )
  228. s.add(", ");
  229. else
  230. mod = true;
  231. s.add(quoteField(name));
  232. s.add(" = ");
  233. getCnx().addValue(s,v);
  234. if ( cache != null )
  235. Reflect.setField(cache,name,v);
  236. }
  237. }
  238. if( !mod )
  239. return null;
  240. s.add(" WHERE ");
  241. addKeys(s,x);
  242. return s.toString();
  243. }
  244. function doDelete( x : T ) {
  245. var s = new StringBuf();
  246. s.add("DELETE FROM ");
  247. s.add(table_name);
  248. s.add(" WHERE ");
  249. addKeys(s,x);
  250. unsafeExecute(s.toString());
  251. removeFromCache(x);
  252. }
  253. function doLock( i : T ) {
  254. if( untyped i._lock )
  255. return;
  256. var s = new StringBuf();
  257. s.add("SELECT * FROM ");
  258. s.add(table_name);
  259. s.add(" WHERE ");
  260. addKeys(s, i);
  261. // will force sync
  262. if( unsafeObject(s.toString(),true) != i )
  263. throw "Could not lock object (was deleted ?); try restarting transaction";
  264. }
  265. function objectToString( it : T ) : String {
  266. var s = new StringBuf();
  267. s.add(table_name);
  268. if( table_keys.length == 1 ) {
  269. s.add("#");
  270. s.add(Reflect.field(it,table_keys[0]));
  271. } else {
  272. s.add("(");
  273. var first = true;
  274. for( f in table_keys ) {
  275. if( first )
  276. first = false;
  277. else
  278. s.add(",");
  279. s.add(quoteField(f));
  280. s.add(":");
  281. s.add(Reflect.field(it,f));
  282. }
  283. s.add(")");
  284. }
  285. return s.toString();
  286. }
  287. function doSerialize( field : String, v : Dynamic ) : haxe.io.Bytes {
  288. var s = new haxe.Serializer();
  289. s.useEnumIndex = true;
  290. s.serialize(v);
  291. var str = s.toString();
  292. #if neko
  293. return neko.Lib.bytesReference(str);
  294. #else
  295. return haxe.io.Bytes.ofString(str);
  296. #end
  297. }
  298. function doUnserialize( field : String, b : haxe.io.Bytes ) : Dynamic {
  299. if( b == null )
  300. return null;
  301. var str;
  302. #if neko
  303. str = neko.Lib.stringReference(b);
  304. #else
  305. str = b.toString();
  306. #end
  307. if( str == "" )
  308. return null;
  309. return haxe.Unserializer.run(str);
  310. }
  311. /* ---------------------------- INTERNAL API -------------------------- */
  312. function normalizeCache(x:CacheType<T>)
  313. {
  314. for (f in Reflect.fields(x) )
  315. {
  316. var val:Dynamic = Reflect.field(x,f), info = table_infos.hfields.get(f);
  317. if (info != null)
  318. {
  319. if (val != null) switch (info.t) {
  320. case DDate, DDateTime if (!Std.is(val,Date)):
  321. if (Std.is(val,Float))
  322. {
  323. val = Date.fromTime(val);
  324. } else {
  325. var v = val + "";
  326. var index = v.indexOf('.');
  327. if (index >= 0)
  328. v = v.substr(0,index);
  329. val = Date.fromString(v);
  330. }
  331. case DSmallBinary, DLongBinary, DBinary, DBytes(_), DData if (Std.is(val, String)):
  332. val = haxe.io.Bytes.ofString(val);
  333. case DString(_) | DTinyText | DSmallText | DText if(!Std.is(val,String)):
  334. val = val + "";
  335. #if (cs && erase_generics)
  336. // on C#, SQLite Ints are returned as Int64
  337. case DInt if (!Std.is(val,Int)):
  338. val = cast(val,Int);
  339. #end
  340. case DBool if (!Std.is(val,Bool)):
  341. if (Std.is(val,Int))
  342. val = val != 0;
  343. else if (Std.is(val, String)) switch (val.toLowerCase()) {
  344. case "1", "true": val = true;
  345. case "0", "false": val = false;
  346. }
  347. case DFloat if (Std.is(val,String)):
  348. val = Std.parseFloat(val);
  349. case _:
  350. }
  351. Reflect.setField(x, f, val);
  352. }
  353. }
  354. }
  355. function cacheObject( x : T, lock : Bool ) {
  356. #if neko
  357. var o = untyped __dollar__new(x);
  358. untyped __dollar__objsetproto(o, class_proto.prototype);
  359. #else
  360. var o : T = Type.createEmptyInstance(cast class_proto);
  361. untyped o._manager = this;
  362. #end
  363. normalizeCache(x);
  364. for (f in Reflect.fields(x) )
  365. {
  366. var val:Dynamic = Reflect.field(x,f), info = table_infos.hfields.get(f);
  367. if (info != null)
  368. {
  369. var fieldName = getFieldName(info);
  370. Reflect.setField(o, fieldName, val);
  371. }
  372. }
  373. Reflect.setField(o,cache_field,x);
  374. addToCache(o);
  375. untyped o._lock = lock;
  376. return o;
  377. }
  378. function make( x : T ) {
  379. }
  380. function unmake( x : T ) {
  381. }
  382. function quoteField(f : String) {
  383. return KEYWORDS.exists(f.toLowerCase()) ? "`"+f+"`" : f;
  384. }
  385. function addKeys( s : StringBuf, x : {} ) {
  386. var first = true;
  387. for( k in table_keys ) {
  388. if( first )
  389. first = false;
  390. else
  391. s.add(" AND ");
  392. s.add(quoteField(k));
  393. s.add(" = ");
  394. var f = Reflect.field(x,k);
  395. if( f == null )
  396. throw ("Missing key "+k);
  397. getCnx().addValue(s,f);
  398. }
  399. }
  400. function unsafeExecute( sql : String ) {
  401. return getCnx().request(sql);
  402. }
  403. public function unsafeObject( sql : String, lock : Bool ) : T {
  404. if( lock != false ) {
  405. lock = true;
  406. sql += getLockMode();
  407. }
  408. var r = unsafeExecute(sql);
  409. var r = r.hasNext() ? r.next() : null;
  410. if( r == null )
  411. return null;
  412. normalizeCache(r);
  413. var c = getFromCache(r,lock);
  414. if( c != null )
  415. return c;
  416. r = cacheObject(r,lock);
  417. make(r);
  418. return r;
  419. }
  420. public function unsafeObjects( sql : String, lock : Bool ) : List<T> {
  421. if( lock != false ) {
  422. lock = true;
  423. sql += getLockMode();
  424. }
  425. var l = unsafeExecute(sql).results();
  426. var l2 = new List<T>();
  427. for( x in l ) {
  428. normalizeCache(x);
  429. var c = getFromCache(x,lock);
  430. if( c != null )
  431. l2.add(c);
  432. else {
  433. x = cacheObject(x,lock);
  434. make(x);
  435. l2.add(x);
  436. }
  437. }
  438. return l2;
  439. }
  440. public function unsafeCount( sql : String ) {
  441. return unsafeExecute(sql).getIntResult(0);
  442. }
  443. public function unsafeDelete( sql : String ) {
  444. unsafeExecute(sql);
  445. }
  446. public function unsafeGet( id : Dynamic, ?lock : Bool ) : T {
  447. if( lock == null ) lock = true;
  448. if( table_keys.length != 1 )
  449. throw "Invalid number of keys";
  450. if( id == null )
  451. return null;
  452. var x : Dynamic = getFromCacheKey(Std.string(id) + table_name);
  453. if( x != null && (!lock || x._lock) )
  454. return x;
  455. var s = new StringBuf();
  456. s.add("SELECT * FROM ");
  457. s.add(table_name);
  458. s.add(" WHERE ");
  459. s.add(quoteField(table_keys[0]));
  460. s.add(" = ");
  461. getCnx().addValue(s,id);
  462. return unsafeObject(s.toString(), lock);
  463. }
  464. public function unsafeGetWithKeys( keys : { }, ?lock : Bool ) : T {
  465. if( lock == null ) lock = true;
  466. var x : Dynamic = getFromCacheKey(makeCacheKey(cast keys));
  467. if( x != null && (!lock || x._lock) )
  468. return x;
  469. var s = new StringBuf();
  470. s.add("SELECT * FROM ");
  471. s.add(table_name);
  472. s.add(" WHERE ");
  473. addKeys(s,keys);
  474. return unsafeObject(s.toString(),lock);
  475. }
  476. public function unsafeGetId( o : T ) : Dynamic {
  477. return o == null ? null : Reflect.field(o, table_keys[0]);
  478. }
  479. public static function nullCompare( a : String, b : String, eq : Bool ) {
  480. // we can't use a null-safe operator here
  481. if( cnx.dbName() != "MySQL" )
  482. return a + (eq ? " = " : " != ") + b;
  483. var sql = a+" <=> "+b;
  484. if( !eq ) sql = "NOT("+sql+")";
  485. return sql;
  486. }
  487. function addCondition(s : StringBuf,x) {
  488. var first = true;
  489. if( x != null )
  490. for( f in Reflect.fields(x) ) {
  491. if( first )
  492. first = false;
  493. else
  494. s.add(" AND ");
  495. s.add(quoteField(f));
  496. var d = Reflect.field(x,f);
  497. if( d == null )
  498. s.add(" IS NULL");
  499. else {
  500. s.add(" = ");
  501. getCnx().addValue(s,d);
  502. }
  503. }
  504. if( first )
  505. s.add("TRUE");
  506. }
  507. /* --------------------------- MISC API ------------------------------ */
  508. public function dbClass() : Class<Dynamic> {
  509. return cast class_proto;
  510. }
  511. public function dbInfos() {
  512. return table_infos;
  513. }
  514. function getCnx() {
  515. return cnx;
  516. }
  517. function getLockMode() {
  518. return lockMode;
  519. }
  520. /**
  521. Remove the cached value for the given Object field : this will ensure
  522. that the value is updated when calling .update(). This is necessary if
  523. you are modifying binary data in-place since the cache will be modified
  524. as well.
  525. **/
  526. public function forceUpdate( o : T, field : String ) {
  527. // set a reference that will ensure != and .compare() != 0
  528. Reflect.setField(Reflect.field(o,cache_field),field,null);
  529. }
  530. /* --------------------------- INIT / CLEANUP ------------------------- */
  531. public static function initialize() {
  532. var l = init_list;
  533. init_list = new List();
  534. for( m in l )
  535. for( r in m.table_infos.relations )
  536. m.initRelation(r);
  537. }
  538. public static function cleanup() {
  539. object_cache = new haxe.ds.StringMap();
  540. }
  541. function initRelation( r : RecordInfos.RecordRelation ) {
  542. // setup getter/setter
  543. var spod : Dynamic = Type.resolveClass(r.type);
  544. if( spod == null ) throw "Missing spod type " + r.type;
  545. var manager : Manager<Dynamic> = spod.manager;
  546. var hprop = "__"+r.prop;
  547. var hkey = r.key;
  548. var lock = r.lock;
  549. if( manager == null || manager.table_keys == null ) throw ("Invalid manager for relation "+table_name+":"+r.prop);
  550. if( manager.table_keys.length != 1 ) throw ("Relation " + r.prop + "(" + r.key + ") on a multiple key table");
  551. #if neko
  552. Reflect.setField(class_proto.prototype,"get_"+r.prop,function() {
  553. var othis = untyped __this__;
  554. var f = Reflect.field(othis,hprop);
  555. if( f != null )
  556. return f;
  557. var id = Reflect.field(othis, hkey);
  558. if( id == null )
  559. return null;
  560. f = manager.unsafeGet(id,lock);
  561. // it's highly possible that in that case the object has been inserted
  562. // after we started our transaction : in that case, let's lock it, since
  563. // it's still better than returning 'null' while it exists
  564. if( f == null && id != null && !lock )
  565. f = manager.unsafeGet(id,true);
  566. Reflect.setField(othis,hprop,f);
  567. return f;
  568. });
  569. Reflect.setField(class_proto.prototype,"set_"+r.prop,function(f) {
  570. var othis = untyped __this__;
  571. Reflect.setField(othis,hprop,f);
  572. Reflect.setField(othis,hkey,Reflect.field(f,manager.table_keys[0]));
  573. return f;
  574. });
  575. #end
  576. }
  577. #if !neko
  578. function __get( x : Dynamic, prop : String, key : String, lock ) {
  579. var v = Reflect.field(x,prop);
  580. if( v != null )
  581. return v;
  582. var y = unsafeGet(Reflect.field(x, key), lock);
  583. Reflect.setField(x,prop,v);
  584. return y;
  585. }
  586. function __set( x : Dynamic, prop : String, key : String, v : T ) {
  587. Reflect.setField(x,prop,v);
  588. if( v == null )
  589. Reflect.setField(x,key,null);
  590. else
  591. Reflect.setField(x,key,Reflect.field(v,table_keys[0]));
  592. return v;
  593. }
  594. #end
  595. /* ---------------------------- OBJECT CACHE -------------------------- */
  596. function makeCacheKey( x : T ) : String {
  597. if( table_keys.length == 1 ) {
  598. var k = Reflect.field(x,table_keys[0]);
  599. if( k == null )
  600. throw("Missing key "+table_keys[0]);
  601. return Std.string(k)+table_name;
  602. }
  603. var s = new StringBuf();
  604. for( k in table_keys ) {
  605. var v = Reflect.field(x,k);
  606. if( k == null )
  607. throw("Missing key "+k);
  608. s.add(v);
  609. s.add("#");
  610. }
  611. s.add(table_name);
  612. return s.toString();
  613. }
  614. function addToCache( x : CacheType<T> ) {
  615. object_cache.set(makeCacheKey(x),x);
  616. }
  617. function removeFromCache( x : CacheType<T> ) {
  618. object_cache.remove(makeCacheKey(x));
  619. }
  620. function getFromCacheKey( key : String ) : T {
  621. return cast object_cache.get(key);
  622. }
  623. function getFromCache( x : CacheType<T>, lock : Bool ) : T {
  624. var c : Dynamic = object_cache.get(makeCacheKey(x));
  625. if( c != null && lock && !c._lock ) {
  626. // synchronize the fields since our result is up-to-date !
  627. for( f in Reflect.fields(c) )
  628. Reflect.deleteField(c,f);
  629. for (f in table_infos.fields)
  630. {
  631. var name = f.name,
  632. fieldName = getFieldName(f);
  633. Reflect.setField(c,fieldName,Reflect.field(x,name));
  634. }
  635. // mark as locked
  636. c._lock = true;
  637. // restore our manager
  638. #if !neko
  639. untyped c._manager = this;
  640. #end
  641. // use the new object as our cache of fields
  642. Reflect.setField(c,cache_field,x);
  643. // remake object
  644. make(c);
  645. }
  646. return c;
  647. }
  648. /* ---------------------------- QUOTES -------------------------- */
  649. public static function quoteAny( v : Dynamic ) {
  650. var s = new StringBuf();
  651. cnx.addValue(s, v);
  652. return s.toString();
  653. }
  654. public static function quoteList( v : String, it : Iterable<Dynamic> ) {
  655. var b = new StringBuf();
  656. var first = true;
  657. if( it != null )
  658. for( v in it ) {
  659. if( first ) first = false else b.addChar(','.code);
  660. cnx.addValue(b, v);
  661. }
  662. if( first )
  663. return "FALSE";
  664. return v + " IN (" + b.toString() + ")";
  665. }
  666. // We need Bytes.toString to not be DCE'd. See #1937
  667. @:keep static function __depends() { return haxe.io.Bytes.alloc(0).toString(); }
  668. }
  669. private typedef CacheType<T> = Dynamic;