Manager.hx 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730
  1. /*
  2. * Copyright (C)2005-2012 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( v != vc && (!isBinary(f.t) || hasBinaryChanged(v,vc)) ) {
  217. switch( f.t ) {
  218. case DData:
  219. v = doUpdateCache(x, name, v);
  220. if( !hasBinaryChanged(v,vc) )
  221. continue;
  222. default:
  223. }
  224. if( mod )
  225. s.add(", ");
  226. else
  227. mod = true;
  228. s.add(quoteField(name));
  229. s.add(" = ");
  230. getCnx().addValue(s,v);
  231. Reflect.setField(cache,name,v);
  232. }
  233. }
  234. if( !mod )
  235. return null;
  236. s.add(" WHERE ");
  237. addKeys(s,x);
  238. return s.toString();
  239. }
  240. function doDelete( x : T ) {
  241. var s = new StringBuf();
  242. s.add("DELETE FROM ");
  243. s.add(table_name);
  244. s.add(" WHERE ");
  245. addKeys(s,x);
  246. unsafeExecute(s.toString());
  247. removeFromCache(x);
  248. }
  249. function doLock( i : T ) {
  250. if( untyped i._lock )
  251. return;
  252. var s = new StringBuf();
  253. s.add("SELECT * FROM ");
  254. s.add(table_name);
  255. s.add(" WHERE ");
  256. addKeys(s, i);
  257. // will force sync
  258. if( unsafeObject(s.toString(),true) != i )
  259. throw "Could not lock object (was deleted ?); try restarting transaction";
  260. }
  261. function objectToString( it : T ) : String {
  262. var s = new StringBuf();
  263. s.add(table_name);
  264. if( table_keys.length == 1 ) {
  265. s.add("#");
  266. s.add(Reflect.field(it,table_keys[0]));
  267. } else {
  268. s.add("(");
  269. var first = true;
  270. for( f in table_keys ) {
  271. if( first )
  272. first = false;
  273. else
  274. s.add(",");
  275. s.add(quoteField(f));
  276. s.add(":");
  277. s.add(Reflect.field(it,f));
  278. }
  279. s.add(")");
  280. }
  281. return s.toString();
  282. }
  283. function doSerialize( field : String, v : Dynamic ) : haxe.io.Bytes {
  284. var s = new haxe.Serializer();
  285. s.useEnumIndex = true;
  286. s.serialize(v);
  287. var str = s.toString();
  288. #if neko
  289. return neko.Lib.bytesReference(str);
  290. #else
  291. return haxe.io.Bytes.ofString(str);
  292. #end
  293. }
  294. function doUnserialize( field : String, b : haxe.io.Bytes ) : Dynamic {
  295. if( b == null )
  296. return null;
  297. var str;
  298. #if neko
  299. str = neko.Lib.stringReference(b);
  300. #else
  301. str = b.toString();
  302. #end
  303. if( str == "" )
  304. return null;
  305. return haxe.Unserializer.run(str);
  306. }
  307. /* ---------------------------- INTERNAL API -------------------------- */
  308. function normalizeCache(x:CacheType<T>)
  309. {
  310. for (f in Reflect.fields(x) )
  311. {
  312. var val:Dynamic = Reflect.field(x,f), info = table_infos.hfields.get(f);
  313. if (info != null)
  314. {
  315. if (val != null) switch (info.t) {
  316. case DDate, DDateTime if (!Std.is(val,Date)):
  317. if (Std.is(val,Float))
  318. {
  319. val = Date.fromTime(val);
  320. } else {
  321. var v = val + "";
  322. var index = v.indexOf('.');
  323. if (index >= 0)
  324. v = v.substr(0,index);
  325. val = Date.fromString(v);
  326. }
  327. case DSmallBinary, DLongBinary, DBinary, DBytes(_), DData if (Std.is(val, String)):
  328. val = haxe.io.Bytes.ofString(val);
  329. case DString(_) | DTinyText | DSmallText | DText if(!Std.is(val,String)):
  330. val = val + "";
  331. case DBool if (!Std.is(val,Bool)):
  332. if (Std.is(val,Int))
  333. val = val != 0;
  334. else if (Std.is(val, String)) switch (val.toLowerCase()) {
  335. case "1", "true": val = true;
  336. case "0", "false": val = false;
  337. }
  338. case DFloat if (Std.is(val,String)):
  339. val = Std.parseFloat(val);
  340. case _:
  341. }
  342. Reflect.setField(x, f, val);
  343. }
  344. }
  345. }
  346. function cacheObject( x : T, lock : Bool ) {
  347. #if neko
  348. var o = untyped __dollar__new(x);
  349. untyped __dollar__objsetproto(o, class_proto.prototype);
  350. #else
  351. var o : T = Type.createEmptyInstance(cast class_proto);
  352. untyped o._manager = this;
  353. #end
  354. normalizeCache(x);
  355. for (f in Reflect.fields(x) )
  356. {
  357. var val:Dynamic = Reflect.field(x,f), info = table_infos.hfields.get(f);
  358. if (info != null)
  359. {
  360. var fieldName = getFieldName(info);
  361. Reflect.setField(o, fieldName, val);
  362. }
  363. }
  364. Reflect.setField(o,cache_field,x);
  365. addToCache(o);
  366. untyped o._lock = lock;
  367. return o;
  368. }
  369. function make( x : T ) {
  370. }
  371. function unmake( x : T ) {
  372. }
  373. function quoteField(f : String) {
  374. return KEYWORDS.exists(f.toLowerCase()) ? "`"+f+"`" : f;
  375. }
  376. function addKeys( s : StringBuf, x : {} ) {
  377. var first = true;
  378. for( k in table_keys ) {
  379. if( first )
  380. first = false;
  381. else
  382. s.add(" AND ");
  383. s.add(quoteField(k));
  384. s.add(" = ");
  385. var f = Reflect.field(x,k);
  386. if( f == null )
  387. throw ("Missing key "+k);
  388. getCnx().addValue(s,f);
  389. }
  390. }
  391. function unsafeExecute( sql : String ) {
  392. return getCnx().request(sql);
  393. }
  394. public function unsafeObject( sql : String, lock : Bool ) : T {
  395. if( lock != false ) {
  396. lock = true;
  397. sql += getLockMode();
  398. }
  399. var r = unsafeExecute(sql).next();
  400. if( r == null )
  401. return null;
  402. normalizeCache(r);
  403. var c = getFromCache(r,lock);
  404. if( c != null )
  405. return c;
  406. r = cacheObject(r,lock);
  407. make(r);
  408. return r;
  409. }
  410. public function unsafeObjects( sql : String, lock : Bool ) : List<T> {
  411. if( lock != false ) {
  412. lock = true;
  413. sql += getLockMode();
  414. }
  415. var l = unsafeExecute(sql).results();
  416. var l2 = new List<T>();
  417. for( x in l ) {
  418. normalizeCache(x);
  419. var c = getFromCache(x,lock);
  420. if( c != null )
  421. l2.add(c);
  422. else {
  423. x = cacheObject(x,lock);
  424. make(x);
  425. l2.add(x);
  426. }
  427. }
  428. return l2;
  429. }
  430. public function unsafeCount( sql : String ) {
  431. return unsafeExecute(sql).getIntResult(0);
  432. }
  433. public function unsafeDelete( sql : String ) {
  434. unsafeExecute(sql);
  435. }
  436. public function unsafeGet( id : Dynamic, ?lock : Bool ) : T {
  437. if( lock == null ) lock = true;
  438. if( table_keys.length != 1 )
  439. throw "Invalid number of keys";
  440. if( id == null )
  441. return null;
  442. var x : Dynamic = getFromCacheKey(Std.string(id) + table_name);
  443. if( x != null && (!lock || x._lock) )
  444. return x;
  445. var s = new StringBuf();
  446. s.add("SELECT * FROM ");
  447. s.add(table_name);
  448. s.add(" WHERE ");
  449. s.add(quoteField(table_keys[0]));
  450. s.add(" = ");
  451. getCnx().addValue(s,id);
  452. return unsafeObject(s.toString(), lock);
  453. }
  454. public function unsafeGetWithKeys( keys : { }, ?lock : Bool ) : T {
  455. if( lock == null ) lock = true;
  456. var x : Dynamic = getFromCacheKey(makeCacheKey(cast keys));
  457. if( x != null && (!lock || x._lock) )
  458. return x;
  459. var s = new StringBuf();
  460. s.add("SELECT * FROM ");
  461. s.add(table_name);
  462. s.add(" WHERE ");
  463. addKeys(s,keys);
  464. return unsafeObject(s.toString(),lock);
  465. }
  466. public function unsafeGetId( o : T ) : Dynamic {
  467. return o == null ? null : Reflect.field(o, table_keys[0]);
  468. }
  469. public static function nullCompare( a : String, b : String, eq : Bool ) {
  470. // we can't use a null-safe operator here
  471. if( cnx.dbName() != "MySQL" )
  472. return a + (eq ? " = " : " != ") + b;
  473. var sql = a+" <=> "+b;
  474. if( !eq ) sql = "NOT("+sql+")";
  475. return sql;
  476. }
  477. function addCondition(s : StringBuf,x) {
  478. var first = true;
  479. if( x != null )
  480. for( f in Reflect.fields(x) ) {
  481. if( first )
  482. first = false;
  483. else
  484. s.add(" AND ");
  485. s.add(quoteField(f));
  486. var d = Reflect.field(x,f);
  487. if( d == null )
  488. s.add(" IS NULL");
  489. else {
  490. s.add(" = ");
  491. getCnx().addValue(s,d);
  492. }
  493. }
  494. if( first )
  495. s.add("TRUE");
  496. }
  497. /* --------------------------- MISC API ------------------------------ */
  498. public function dbClass() : Class<Dynamic> {
  499. return cast class_proto;
  500. }
  501. public function dbInfos() {
  502. return table_infos;
  503. }
  504. function getCnx() {
  505. return cnx;
  506. }
  507. function getLockMode() {
  508. return lockMode;
  509. }
  510. /**
  511. Remove the cached value for the given Object field : this will ensure
  512. that the value is updated when calling .update(). This is necessary if
  513. you are modifying binary data in-place since the cache will be modified
  514. as well.
  515. **/
  516. public function forceUpdate( o : T, field : String ) {
  517. // set a reference that will ensure != and .compare() != 0
  518. Reflect.setField(Reflect.field(o,cache_field),field,null);
  519. }
  520. /* --------------------------- INIT / CLEANUP ------------------------- */
  521. public static function initialize() {
  522. var l = init_list;
  523. init_list = new List();
  524. for( m in l )
  525. for( r in m.table_infos.relations )
  526. m.initRelation(r);
  527. }
  528. public static function cleanup() {
  529. object_cache = new haxe.ds.StringMap();
  530. }
  531. function initRelation( r : RecordInfos.RecordRelation ) {
  532. // setup getter/setter
  533. var spod : Dynamic = Type.resolveClass(r.type);
  534. if( spod == null ) throw "Missing spod type " + r.type;
  535. var manager : Manager<Dynamic> = spod.manager;
  536. var hprop = "__"+r.prop;
  537. var hkey = r.key;
  538. var lock = r.lock;
  539. if( manager == null || manager.table_keys == null ) throw ("Invalid manager for relation "+table_name+":"+r.prop);
  540. if( manager.table_keys.length != 1 ) throw ("Relation " + r.prop + "(" + r.key + ") on a multiple key table");
  541. #if neko
  542. Reflect.setField(class_proto.prototype,"get_"+r.prop,function() {
  543. var othis = untyped __this__;
  544. var f = Reflect.field(othis,hprop);
  545. if( f != null )
  546. return f;
  547. var id = Reflect.field(othis, hkey);
  548. if( id == null )
  549. return null;
  550. f = manager.unsafeGet(id,lock);
  551. // it's highly possible that in that case the object has been inserted
  552. // after we started our transaction : in that case, let's lock it, since
  553. // it's still better than returning 'null' while it exists
  554. if( f == null && id != null && !lock )
  555. f = manager.unsafeGet(id,true);
  556. Reflect.setField(othis,hprop,f);
  557. return f;
  558. });
  559. Reflect.setField(class_proto.prototype,"set_"+r.prop,function(f) {
  560. var othis = untyped __this__;
  561. Reflect.setField(othis,hprop,f);
  562. Reflect.setField(othis,hkey,Reflect.field(f,manager.table_keys[0]));
  563. return f;
  564. });
  565. #end
  566. }
  567. #if !neko
  568. function __get( x : Dynamic, prop : String, key : String, lock ) {
  569. var v = Reflect.field(x,prop);
  570. if( v != null )
  571. return v;
  572. var y = unsafeGet(Reflect.field(x, key), lock);
  573. Reflect.setField(x,prop,v);
  574. return y;
  575. }
  576. function __set( x : Dynamic, prop : String, key : String, v : T ) {
  577. Reflect.setField(x,prop,v);
  578. if( v == null )
  579. Reflect.setField(x,key,null);
  580. else
  581. Reflect.setField(x,key,Reflect.field(v,table_keys[0]));
  582. }
  583. #end
  584. /* ---------------------------- OBJECT CACHE -------------------------- */
  585. function makeCacheKey( x : T ) : String {
  586. if( table_keys.length == 1 ) {
  587. var k = Reflect.field(x,table_keys[0]);
  588. if( k == null )
  589. throw("Missing key "+table_keys[0]);
  590. return Std.string(k)+table_name;
  591. }
  592. var s = new StringBuf();
  593. for( k in table_keys ) {
  594. var v = Reflect.field(x,k);
  595. if( k == null )
  596. throw("Missing key "+k);
  597. s.add(v);
  598. s.add("#");
  599. }
  600. s.add(table_name);
  601. return s.toString();
  602. }
  603. function addToCache( x : CacheType<T> ) {
  604. object_cache.set(makeCacheKey(x),x);
  605. }
  606. function removeFromCache( x : CacheType<T> ) {
  607. object_cache.remove(makeCacheKey(x));
  608. }
  609. function getFromCacheKey( key : String ) : T {
  610. return cast object_cache.get(key);
  611. }
  612. function getFromCache( x : CacheType<T>, lock : Bool ) : T {
  613. var c : Dynamic = object_cache.get(makeCacheKey(x));
  614. if( c != null && lock && !c._lock ) {
  615. // synchronize the fields since our result is up-to-date !
  616. for( f in Reflect.fields(c) )
  617. Reflect.deleteField(c,f);
  618. for (f in table_infos.fields)
  619. {
  620. var name = f.name,
  621. fieldName = getFieldName(f);
  622. Reflect.setField(c,fieldName,Reflect.field(x,name));
  623. }
  624. // mark as locked
  625. c._lock = true;
  626. // restore our manager
  627. #if !neko
  628. untyped c._manager = this;
  629. #end
  630. // use the new object as our cache of fields
  631. Reflect.setField(c,cache_field,x);
  632. // remake object
  633. make(c);
  634. }
  635. return c;
  636. }
  637. /* ---------------------------- QUOTES -------------------------- */
  638. public static function quoteAny( v : Dynamic ) {
  639. var s = new StringBuf();
  640. cnx.addValue(s, v);
  641. return s.toString();
  642. }
  643. public static function quoteList( v : String, it : Iterable<Dynamic> ) {
  644. var b = new StringBuf();
  645. var first = true;
  646. if( it != null )
  647. for( v in it ) {
  648. if( first ) first = false else b.addChar(','.code);
  649. cnx.addValue(b, v);
  650. }
  651. if( first )
  652. return "FALSE";
  653. return v + " IN (" + b.toString() + ")";
  654. }
  655. }
  656. private typedef CacheType<T> = Dynamic;