tripal_chado.query.api.inc 73 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027
  1. <?php
  2. /**
  3. * @file
  4. * Provides an API for querying of chado including inserting, updating, deleting and
  5. * selecting from chado.
  6. */
  7. /**
  8. * @defgroup tripal_chado_query_api Chado Query
  9. * @ingroup tripal_chado_api
  10. * @{
  11. * Provides an API for querying of chado including inserting, updating, deleting and
  12. * selecting from specific chado tables. There is also a generic function, chado_query(),
  13. * to execute and SQL statement on chado. It is ideal to use these functions to interact
  14. * with chado in order to keep your module compatible with both local & external chado
  15. * databases. Furthermore, it ensures connection to the chado database is taken care
  16. * of for you.
  17. *
  18. * Generic Queries to a specifc chado table:
  19. *
  20. * chado_select_record( [table name], [columns to select], [specify record to select], [options*] )
  21. * This function allows you to select various columns from the specified chado table. Although
  22. * you can only select from a single table, you can specify the record to select using values
  23. * from related tables through use of a nested array. For example, the following code shows
  24. * you how to select the name and uniquename of a feature based on it's type and source
  25. * organism.
  26. * @code
  27. * $values = array(
  28. * 'organism_id' => array(
  29. * 'genus' => 'Citrus',
  30. * 'species' => 'sinensis',
  31. * ),
  32. * 'type_id' => array (
  33. * 'cv_id' => array (
  34. * 'name' => 'sequence',
  35. * ),
  36. * 'name' => 'gene',
  37. * 'is_obsolete' => 0
  38. * ),
  39. * );
  40. * $result = chado_select_record(
  41. * 'feature', // table to select from
  42. * array('name', 'uniquename'), // columns to select
  43. * $values // record to select (see variable defn. above)
  44. * );
  45. * @endcode
  46. *
  47. * chado_insert_record( [table name], [values to insert], [options*] )
  48. * This function allows you to insert a single record into a specific table. The values to
  49. * insert are specified using an associative array where the keys are the column names to
  50. * insert into and they point to the value to be inserted into that column. If the column
  51. * is a foreign key, the key will point to an array specifying the record in the foreign
  52. * table and then the primary key of that record will be inserted in the column. For example,
  53. * the following code will insert a feature and for the type_id, the cvterm.cvterm_id of
  54. * the cvterm record will be inserted and for the organism_id, the organism.organism_id
  55. * of the organism_record will be inserted.
  56. * @code
  57. * $values = array(
  58. * 'organism_id' => array(
  59. * 'genus' => 'Citrus',
  60. * 'species' => 'sinensis',
  61. * ),
  62. * 'name' => 'orange1.1g000034m.g',
  63. * 'uniquename' => 'orange1.1g000034m.g',
  64. * 'type_id' => array (
  65. * 'cv_id' => array (
  66. * 'name' => 'sequence',
  67. * ),
  68. * 'name' => 'gene',
  69. * 'is_obsolete' => 0
  70. * ),
  71. * );
  72. * $result = chado_insert_record(
  73. * 'feature', // table to insert into
  74. * $values // values to insert
  75. * );
  76. * @endcode
  77. *
  78. * chado_update_record( [table name], [specify record to update], [values to change], [options*] )
  79. * This function allows you to update records in a specific chado table. The record(s)
  80. * you wish to update are specified the same as in the select function above and
  81. * the values to be update are specified the same as the values to be inserted were. For
  82. * example, the following code species that a feature with a given uniquename, organism_id,
  83. * and type_id (the unique constraint for the feature table) will be updated with a new name,
  84. * and the type changed from a gene to an mRNA.
  85. * @code
  86. * $umatch = array(
  87. * 'organism_id' => array(
  88. * 'genus' => 'Citrus',
  89. * 'species' => 'sinensis',
  90. * ),
  91. * 'uniquename' => 'orange1.1g000034m.g7',
  92. * 'type_id' => array (
  93. * 'cv_id' => array (
  94. * 'name' => 'sequence',
  95. * ),
  96. * 'name' => 'gene',
  97. * 'is_obsolete' => 0
  98. * ),
  99. * );
  100. * $uvalues = array(
  101. * 'name' => 'orange1.1g000034m.g',
  102. * 'type_id' => array (
  103. * 'cv_id' => array (
  104. * 'name' => 'sequence',
  105. * ),
  106. * 'name' => 'mRNA',
  107. * 'is_obsolete' => 0
  108. * ),
  109. * );
  110. * $result = chado_update_record('feature',$umatch,$uvalues);
  111. * @endcode
  112. *
  113. * chado_delete_record( [table name], [specify records to delete], [options*] )
  114. * This function allows you to delete records from a specific chado table. The record(s)
  115. * to delete are specified the same as the record to select/update was above. For example,
  116. * the following code will delete all genes from the organism Citrus sinensis.
  117. * @code
  118. * $values = array(
  119. * 'organism_id' => array(
  120. * 'genus' => 'Citrus',
  121. * 'species' => 'sinensis',
  122. * ),
  123. * 'type_id' => array (
  124. * 'cv_id' => array (
  125. * 'name' => 'sequence',
  126. * ),
  127. * 'name' => 'gene',
  128. * 'is_obsolete' => 0
  129. * ),
  130. * );
  131. * $result = chado_select_record(
  132. * 'feature', // table to select from
  133. * $values // records to delete (see variable defn. above)
  134. * );
  135. * @endcode
  136. *
  137. * Generic Queries for any SQL:
  138. *
  139. * Often it is necessary to select from more then one table in chado or to execute
  140. * other complex queries that cannot be handled efficiently by the above functions. It is
  141. * for this reason that the chado_query( [sql string], [arguments to sub-in to the sql] )
  142. * function was created. This function allows you to execute any SQL directly on the
  143. * chado database and should be used with care. If any user input will be used in the query
  144. * make sure to put a placeholder in your SQL string and then define the value in the
  145. * arguments array. This will make sure that the user input is santized and safe through
  146. * type-checking and escaping. The following code shows an example of how to use user input
  147. * resulting from a form and would be called withing the form submit function.
  148. * @code
  149. * $sql = "SELECT F.name, CVT.name as type_name, ORG.common_name
  150. * FROM feature F
  151. * LEFT JOIN cvterm CVT ON F.type_id = CVT.cvterm_id
  152. * LEFT JOIN organism ORG ON F.organism_id = ORG.organism_id
  153. * WHERE
  154. * F.uniquename = :feature_uniquename";
  155. * $args = array( ':feature_uniquename' => $form_state['values']['uniquename'] );
  156. * $result = chado_query( $sql, $args );
  157. * foreach ($result as $r) { [Do something with the records here] }
  158. * @endcode
  159. *
  160. * If you are going to need more then a couple fields, you might want to use the
  161. * Chado Variables API (specifically chado_generate_var()) to select all
  162. * of the common fields needed including following foreign keys.
  163. *
  164. * Loading of Variables from chado data:
  165. *
  166. * These functions, chado_generate_var() and chado_expand_var(), generate objects
  167. * containing the full details of a record(s) in chado. These should be used in all
  168. * theme templates.
  169. *
  170. * This differs from the objects returned by chado_select_record in so far as all foreign key
  171. * relationships have been followed meaning you have more complete details. Thus this function
  172. * should be used whenever you need a full variable and chado_select_record should be used if
  173. * you only case about a few columns.
  174. *
  175. * The initial variable is generated by the
  176. * chado_generate_var([table], [filter criteria], [optional options])
  177. * function. An example of how to use this function is:
  178. * @code
  179. $values = array(
  180. 'name' => 'Medtr4g030710'
  181. );
  182. $features = chado_generate_var('feature', $values);
  183. * @endcode
  184. * This will return an object if there is only one feature with the name Medtr4g030710 or it will
  185. * return an array of feature objects if more than one feature has that name.
  186. *
  187. * Some tables and fields are excluded by default. To have those tables & fields added to
  188. * your variable you can use the
  189. * chado_expand_var([chado variable], [type], [what to expand], [optional options])
  190. * function. An example of how to use this function is:
  191. * @code
  192. // Get a chado object to be expanded
  193. $values = array(
  194. 'name' => 'Medtr4g030710'
  195. );
  196. $features = chado_generate_var('feature', $values);
  197. // Expand the organism node
  198. $feature = chado_expand_var($feature, 'node', 'organism');
  199. // Expand the feature.residues field
  200. $feature = chado_expand_var($feature, 'field', 'feature.residues');
  201. // Expand the feature properties (featureprop table)
  202. $feature = chado_expand_var($feature, 'table', 'featureprop');
  203. * @endcode
  204. */
  205. /**
  206. * Get max rank for a given set of criteria
  207. * This function was developed with the many property tables in chado in mind but will
  208. * work for any table with a rank
  209. *
  210. * @params tablename: the name of the chado table you want to select the max rank from
  211. * this table must contain a rank column of type integer
  212. * @params where_options: array(
  213. * <column_name> => array(
  214. * 'type' => <type of column: INT/STRING>,
  215. * 'value' => <the value you want to filter on>,
  216. * 'exact' => <if TRUE use =; if FALSE use ~>,
  217. * )
  218. * )
  219. * where options should include the id and type for that table to correctly
  220. * group a set of records together where the only difference are the value and rank
  221. *
  222. * @return the maximum rank
  223. *
  224. * @ingroup tripal_chado_query_api
  225. */
  226. function chado_get_table_max_rank($tablename, $where_options) {
  227. $where_clauses = array();
  228. $where_args = array();
  229. //generate the where clause from supplied options
  230. // the key is the column name
  231. $i = 0;
  232. $sql = "
  233. SELECT max(rank) as max_rank, count(rank) as count
  234. FROM {".$tablename."}
  235. WHERE
  236. ";
  237. foreach ($where_options as $key => $value) {
  238. $where_clauses[] = "$key = :$key";
  239. $where_args[":$key"] = $value;
  240. }
  241. $sql .= implode($where_clauses, ' AND ');
  242. $result = chado_query($sql, $where_args)->fetchObject();
  243. if ($result->count > 0) {
  244. return $result->max_rank;
  245. }
  246. else {
  247. return -1;
  248. }
  249. }
  250. /**
  251. * Alter Chado connection settings.
  252. *
  253. * This hook is useful for multi-chado instances. Tripal core functions
  254. * call the chado_set_active() function (e.g. chado_query) but there is no
  255. * opportunity elsewhere to set the active database. This is useful in two
  256. * cases: 1) Users are managed at the database level as in the case of
  257. * SouthGreen Bioinformatics Platform tools (e.g. Banana Genone Hub).
  258. * This allows custom modules to change the database connections on a per-user
  259. * basis, and each user permissions is managed at the database level. Users
  260. * are managed at the database level to provid the same access restrictions
  261. * across various tools that use Chado (e,g, Artemis) 2) When there are
  262. * simply two Chado instances housed in different Chado databases and the
  263. * module needs to control which one is being used at any given time.
  264. *
  265. * @param $settings
  266. * An array containing
  267. *
  268. * @see chado_set_active()
  269. *
  270. * @ingroup tripal_chado_query_api
  271. */
  272. function hook_chado_connection_alter(&$settings) {
  273. // This example shows how we could make sure no table of the 'public' schema
  274. // would be allowed in the coming queries: to do so, the caller will call
  275. // "chado_set_active('chado_only');" and the hook will remove 'public' from
  276. // the search path.
  277. if ('chado_only' == $settings['dbname']) {
  278. $settings['new_active_db'] = 'chado';
  279. // We don't include 'public' in search path.
  280. $settings['new_search_path'] = 'chado';
  281. }
  282. }
  283. /**
  284. * Set the Tripal Database
  285. *
  286. * The chado_set_active function is used to prevent namespace collisions
  287. * when Chado and Drupal are installed in the same database but in different
  288. * schemas. It is also used when using Drupal functions such as
  289. * db_table_exists().
  290. *
  291. * The connection settings can be altered through the hook
  292. * hook_chado_connection_alter.
  293. *
  294. * Current active connection name is stored in the global variable
  295. * $GLOBALS['chado_active_db'].
  296. *
  297. * @see hook_chado_connection_alter()
  298. *
  299. * @ingroup tripal_chado_query_api
  300. */
  301. function chado_set_active($dbname = 'default') {
  302. // Check if the chado_active_db has been set yet.
  303. if (!array_key_exists('chado_active_db', $GLOBALS)) {
  304. $GLOBALS['chado_active_db'] = 'default';
  305. }
  306. $previous_db = $active_db = $GLOBALS['chado_active_db'];
  307. $search_path = tripal_get_schema_name('drupal');
  308. // Change only if 'chado' has been specified.
  309. // Notice that we leave the active_db set as chado but use the possibly user-altered
  310. // schema name for the actual search path. This is to keep outward facing mentions of
  311. // chado as "chado" while still allowing the user to alter the schema name used.
  312. if ($dbname == 'chado') {
  313. $active_db = 'chado';
  314. $search_path = tripal_get_schema_name('chado') . ',' . tripal_get_schema_name('drupal');
  315. }
  316. else {
  317. $active_db = $dbname;
  318. }
  319. $settings = array(
  320. 'dbname' => $dbname,
  321. 'new_active_db' => &$active_db,
  322. 'new_search_path' => &$search_path,
  323. );
  324. // Will call all modules implementing hook_chado_search_path_alter
  325. // note: hooks can alter $active_db and $search_path.
  326. drupal_alter('chado_connection', $settings);
  327. // set chado_active_db to remember active db
  328. $GLOBALS['chado_active_db'] = $active_db;
  329. // set PostgreSQL search_path
  330. db_query('SET search_path TO ' . $search_path);
  331. return $previous_db;
  332. }
  333. /**
  334. * Provides a generic routine for inserting into any Chado table
  335. *
  336. * Use this function to insert a record into any Chado table. The first
  337. * argument specifies the table for inserting and the second is an array
  338. * of values to be inserted. The array is mutli-dimensional such that
  339. * foreign key lookup values can be specified.
  340. *
  341. * @param $table
  342. * The name of the chado table for inserting
  343. * @param $values
  344. * An associative array containing the values for inserting.
  345. * @param $options
  346. * An array of options such as:
  347. * - skip_validation: TRUE or FALSE. If TRUE will skip all the validation steps and
  348. * just try to insert as is. This is much faster but results in unhandled
  349. * non user-friendly errors if the insert fails.
  350. * - return_record: by default, the function will return the record but with
  351. * the primary keys added after insertion. To simply return TRUE on success
  352. * set this option to FALSE
  353. *
  354. * @return
  355. * On success this function returns the inserted record with the new primary keys
  356. * added to the returned array. On failure, it returns FALSE.
  357. *
  358. * Example usage:
  359. * @code
  360. * $values = array(
  361. * 'organism_id' => array(
  362. * 'genus' => 'Citrus',
  363. * 'species' => 'sinensis',
  364. * ),
  365. * 'name' => 'orange1.1g000034m.g',
  366. * 'uniquename' => 'orange1.1g000034m.g',
  367. * 'type_id' => array (
  368. * 'cv_id' => array (
  369. * 'name' => 'sequence',
  370. * ),
  371. * 'name' => 'gene',
  372. * 'is_obsolete' => 0
  373. * ),
  374. * );
  375. * $result = chado_insert_record('feature',$values);
  376. * @endcode
  377. * The above code inserts a record into the feature table. The $values array is
  378. * nested such that the organism is selected by way of the organism_id foreign
  379. * key constraint by specifying the genus and species. The cvterm is also
  380. * specified using its foreign key and the cv_id for the cvterm is nested as
  381. * well.
  382. *
  383. * @ingroup tripal_chado_query_api
  384. */
  385. function chado_insert_record($table, $values, $options = array()) {
  386. $print_errors = (isset($options['print_errors'])) ? $options['print_errors'] : FALSE;
  387. if (!is_array($values)) {
  388. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  389. 'Cannot pass non array as values for inserting.', array(),
  390. array('print' => $print_errors)
  391. );
  392. return FALSE;
  393. }
  394. if (count($values)==0) {
  395. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  396. 'Cannot pass an empty array as values for inserting.',
  397. array(),array('print' => $print_errors)
  398. );
  399. return FALSE;
  400. }
  401. // set defaults for options. If we don't set defaults then
  402. // we get memory leaks when we try to access the elements
  403. if (!is_array($options)) {
  404. $options = array();
  405. }
  406. if (!array_key_exists('skip_validation', $options)) {
  407. $options['skip_validation'] = FALSE;
  408. }
  409. if (!array_key_exists('return_record', $options)) {
  410. $options['return_record'] = TRUE;
  411. }
  412. $insert_values = array();
  413. if (array_key_exists('skip_validation', $options)) {
  414. $validate = !$options['skip_validation'];
  415. }
  416. else {
  417. $validate = TRUE;
  418. }
  419. // get the table description
  420. $table_desc = chado_get_schema($table);
  421. if (!$table_desc) {
  422. tripal_report_error('tripal_chado', TRIPAL_WARNING,
  423. 'chado_insert_record; There is no table description for !table_name',
  424. array('!table_name' => $table), array('print' => $print_errors)
  425. );
  426. return;
  427. }
  428. // iterate through the values array and create a new 'insert_values' array
  429. // that has all the values needed for insert with all foreign relationsihps
  430. // resolved.
  431. foreach ($values as $field => $value) {
  432. // make sure the field is in the table description. If not then return an error
  433. // message
  434. if (!array_key_exists($field, $table_desc['fields'])) {
  435. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  436. "chado_insert_record; The field '%field' does not exist " .
  437. "for the table '%table'. Cannot perform insert. Values: %array",
  438. array('%field' => $field, '%table' => $table, '%array' => print_r($values, 1)),
  439. array('print' => $print_errors)
  440. );
  441. return FALSE;
  442. }
  443. if (is_array($value)) {
  444. // select the value from the foreign key relationship for this value
  445. $results = chado_schema_get_foreign_key($table_desc, $field, $value);
  446. if (sizeof($results) > 1) {
  447. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  448. 'chado_insert_record: Too many records match the criteria supplied for !foreign_key foreign key constraint (!criteria)',
  449. array('!foreign_key' => $field, '!criteria' => print_r($value, TRUE)),
  450. array('print' => $print_errors)
  451. );
  452. return FALSE;
  453. }
  454. elseif (sizeof($results) < 1) {
  455. tripal_report_error('tripal_chado', TRIPAL_DEBUG,
  456. 'chado_insert_record: no record matches criteria supplied for !foreign_key foreign key constraint (!criteria)',
  457. array('!foreign_key' => $field, '!criteria' => print_r($value, TRUE)),
  458. array('print' => $print_errors)
  459. );
  460. return FALSE;
  461. }
  462. else {
  463. $insert_values[$field] = $results[0];
  464. }
  465. }
  466. else {
  467. $insert_values[$field] = $value;
  468. }
  469. }
  470. if ($validate) {
  471. // check for violation of any unique constraints
  472. $ukeys = array();
  473. if (array_key_exists('unique keys', $table_desc)) {
  474. $ukeys = $table_desc['unique keys'];
  475. }
  476. $ukselect_cols = array();
  477. $ukselect_vals = array();
  478. if ($ukeys) {
  479. foreach ($ukeys as $name => $fields) {
  480. foreach ($fields as $index => $field) {
  481. // build the arrays for performing a select that will check the contraint
  482. $ukselect_cols[] = $field;
  483. if (!array_key_exists($field, $insert_values)) {
  484. if (array_key_exists('default', $table_desc['fields'][$field])) {
  485. $ukselect_vals[$field] = $table_desc['fields'][$field]['default'];
  486. }
  487. }
  488. else {
  489. $ukselect_vals[$field] = $insert_values[$field];
  490. }
  491. }
  492. // now check the constraint
  493. if (chado_select_record($table, $ukselect_cols, $ukselect_vals)) {
  494. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  495. "chado_insert_record; Cannot insert duplicate record into $table table: !values",
  496. array('!values' => print_r($values, TRUE)), array('print' => $print_errors)
  497. );
  498. return FALSE;
  499. }
  500. }
  501. }
  502. // If trying to insert a field that is the primary key, make sure it also
  503. // is unique.
  504. if (array_key_exists('primary key', $table_desc)) {
  505. $pkey = $table_desc['primary key'][0];
  506. if (array_key_exists($pkey, $insert_values)) {
  507. $coptions = array();
  508. if (chado_select_record($table, array($pkey), array($pkey => $insert_values[$pkey]), $coptions)) {
  509. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  510. 'chado_insert_record; Cannot insert duplicate primary key into !table table: !values',
  511. array('!table' => $table, '!values' => print_r($values, TRUE)),
  512. array('print' => $print_errors)
  513. );
  514. return FALSE;
  515. }
  516. }
  517. }
  518. // Make sure required fields have a value.
  519. if (!is_array($table_desc['fields'])) {
  520. $table_desc['fields'] = array();
  521. tripal_report_error('tripal_chado', TRIPAL_WARNING,
  522. "chado_insert_record; %table missing fields: \n %schema",
  523. array('%table' => $table, '%schema' => print_r($table_desc, 1)),
  524. array('print' => $print_errors)
  525. );
  526. }
  527. foreach ($table_desc['fields'] as $field => $def) {
  528. // A field is considered missing if it cannot be NULL and there is no
  529. // default value for it or it is of type 'serial'.
  530. if (array_key_exists('NOT NULL', $def) and
  531. !array_key_exists($field, $insert_values) and
  532. !array_key_exists('default', $def) and
  533. strcmp($def['type'], serial) != 0) {
  534. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  535. "chado_insert_record; Field %table.%field cannot be NULL: %values",
  536. array('%table' => $table, '%field' => $field, '%values' => print_r($values, 1)),
  537. array('print' => $print_errors)
  538. );
  539. return FALSE;
  540. }
  541. }
  542. }
  543. // End of validation.
  544. // Now build the insert SQL statement
  545. $ifields = array(); // contains the names of the fields
  546. $itypes = array(); // contains placeholders for the sql query
  547. $ivalues = array(); // contains the values of the fields
  548. foreach ($insert_values as $field => $value) {
  549. $ifields[] = $field;
  550. if (strcmp($value, '__NULL__')==0) {
  551. $itypes[] = "NULL";
  552. }
  553. else {
  554. $itypes[] = ":$field";
  555. $ivalues[":$field"] = $value;
  556. }
  557. }
  558. // create the SQL
  559. $sql = 'INSERT INTO {' . $table . '} (' . implode(", ", $ifields) . ") VALUES (" . implode(", ", $itypes) . ")";
  560. $result = chado_query($sql, $ivalues);
  561. // if we have a result then add primary keys to return array
  562. if ($options['return_record'] == TRUE and $result) {
  563. if (array_key_exists('primary key', $table_desc) and is_array($table_desc['primary key'])) {
  564. foreach ($table_desc['primary key'] as $field) {
  565. $sql = "SELECT CURRVAL('{" . $table . "_" . $field . "_seq}')";
  566. $results = chado_query($sql);
  567. $value = $results->fetchField();
  568. if (!$value) {
  569. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  570. "chado_insert_record; not able to retrieve primary key after insert: %sql",
  571. array('%sql' => $sql),
  572. array('print' => $print_errors)
  573. );
  574. return FALSE;
  575. }
  576. $values[$field] = $value;
  577. }
  578. }
  579. return $values;
  580. }
  581. elseif ($options['return_record'] == FALSE and $result) {
  582. return TRUE;
  583. }
  584. else {
  585. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  586. 'chado_insert_record; Cannot insert record into "%table": %values',
  587. array('%table' => $table, '%values' => print_r($values, 1)),
  588. array('print' => $print_errors)
  589. );
  590. return FALSE;
  591. }
  592. return FALSE;
  593. }
  594. /**
  595. * Provides a generic routine for updating into any Chado table
  596. *
  597. * Use this function to update a record in any Chado table. The first
  598. * argument specifies the table for inserting, the second is an array
  599. * of values to matched for locating the record for updating, and the third
  600. * argument give the values to update. The arrays are mutli-dimensional such
  601. * that foreign key lookup values can be specified.
  602. *
  603. * @param $table
  604. * The name of the chado table for inserting
  605. * @param $match
  606. * An associative array containing the values for locating a record to update.
  607. * @param $values
  608. * An associative array containing the values for updating.
  609. * @param $options
  610. * An array of options such as:
  611. * - return_record: by default, the function will return the TRUE if the record
  612. * was succesfully updated. However, set this option to TRUE to return the
  613. * record that was updated. The returned record will have the fields provided
  614. * but the primary key (if available for the table) will be added to the record.
  615. * @return
  616. * On success this function returns TRUE. On failure, it returns FALSE.
  617. *
  618. * Example usage:
  619. * @code
  620. $umatch = array(
  621. 'organism_id' => array(
  622. 'genus' => 'Citrus',
  623. 'species' => 'sinensis',
  624. ),
  625. 'uniquename' => 'orange1.1g000034m.g7',
  626. 'type_id' => array (
  627. 'cv_id' => array (
  628. 'name' => 'sequence',
  629. ),
  630. 'name' => 'gene',
  631. 'is_obsolete' => 0
  632. ),
  633. );
  634. $uvalues = array(
  635. 'name' => 'orange1.1g000034m.g',
  636. 'type_id' => array (
  637. 'cv_id' => array (
  638. 'name' => 'sequence',
  639. ),
  640. 'name' => 'mRNA',
  641. 'is_obsolete' => 0
  642. ),
  643. );
  644. * $result = chado_update_record('feature',$umatch,$uvalues);
  645. * @endcode
  646. * The above code species that a feature with a given uniquename, organism_id,
  647. * and type_id (the unique constraint for the feature table) will be updated.
  648. * The organism_id is specified as a nested array that uses the organism_id
  649. * foreign key constraint to lookup the specified values to find the exact
  650. * organism_id. The same nested struture is also used for specifying the
  651. * values to update. The function will find the record that matches the
  652. * columns specified and update the record with the avlues in the $uvalues array.
  653. *
  654. * @TODO: Support Complex filtering as is done in chado_select_record();
  655. *
  656. * @ingroup tripal_chado_query_api
  657. */
  658. function chado_update_record($table, $match, $values, $options = NULL) {
  659. $print_errors = (isset($options['print_errors'])) ? $options['print_errors'] : FALSE;
  660. if (!is_array($values)) {
  661. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  662. 'Cannot pass non array as values for updating.',
  663. array(), array('print' => $print_errors)
  664. );
  665. return FALSE;
  666. }
  667. if (count($values)==0) {
  668. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  669. 'Cannot pass an empty array as values for updating.',
  670. array(), array('print' => $print_errors)
  671. );
  672. return FALSE;
  673. }
  674. if (!is_array($match)) {
  675. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  676. 'Cannot pass non array as values for matching.',
  677. array(), array('print' => $print_errors)
  678. );
  679. return FALSE;
  680. }
  681. if (count($match)==0) {
  682. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  683. 'Cannot pass an empty array as values for matching.',
  684. array(), array('print' => $print_errors)
  685. );
  686. return FALSE;
  687. }
  688. // set defaults for options. If we don't set defaults then
  689. // we get memory leaks when we try to access the elements
  690. if (!is_array($options)) {
  691. $options = array();
  692. }
  693. if (!array_key_exists('return_record', $options)) {
  694. $options['return_record'] = FALSE;
  695. }
  696. $update_values = array(); // contains the values to be updated
  697. $update_matches = array(); // contains the values for the where clause
  698. // get the table description
  699. $table_desc = chado_get_schema($table);
  700. if (!$table_desc) {
  701. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  702. 'The table name, %table, does not exist.',
  703. array('%table', $table), array('print' => $print_errors)
  704. );
  705. return FALSE;
  706. }
  707. // if the user wants us to return the record then we need to get the
  708. // unique primary key if one exists. That way we can add it to the
  709. // values that get returned at the end of the function
  710. $pkeys = array();
  711. if ($options['return_record'] == TRUE) {
  712. if (array_key_exists('primary key', $table_desc) and is_array($table_desc['primary key'])) {
  713. $columns = array();
  714. $stmt_suffix = '';
  715. foreach ($table_desc['primary key'] as $field) {
  716. $columns[] = $field;
  717. $stmt_suffix .= substr($field, 0, 2);
  718. }
  719. $options2 = array();
  720. $results = chado_select_record($table, $columns, $match, $options2);
  721. if (count($results) > 0) {
  722. foreach ($results as $index => $pkey) {
  723. $pkeys[] = $pkey;
  724. }
  725. }
  726. }
  727. }
  728. // get the values needed for matching in the SQL statement
  729. foreach ($match as $field => $value) {
  730. if (is_array($value)) {
  731. $results = chado_schema_get_foreign_key($table_desc, $field, $value);
  732. if (sizeof($results) > 1) {
  733. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  734. 'chado_update_record: When trying to find record to update, too many records match the criteria supplied for !foreign_key foreign key constraint (!criteria)',
  735. array('!foreign_key' => $field, '!criteria' => print_r($value, TRUE)),
  736. array('print' => $print_errors)
  737. );
  738. return FALSE;
  739. }
  740. elseif (sizeof($results) < 1) {
  741. tripal_report_error('tripal_chado', TRIPAL_DEBUG,
  742. 'chado_update_record: When trying to find record to update, no record matches criteria supplied for !foreign_key foreign key constraint (!criteria)',
  743. array('!foreign_key' => $field, '!criteria' => print_r($value, TRUE)),
  744. array('print' => $print_errors)
  745. );
  746. return FALSE;
  747. }
  748. else {
  749. $update_matches[$field] = $results[0];
  750. }
  751. }
  752. else {
  753. $update_matches[$field] = $value;
  754. }
  755. }
  756. // get the values used for updating
  757. foreach ($values as $field => &$value) {
  758. if (is_array($value)) {
  759. $foreign_options = array();
  760. // select the value from the foreign key relationship for this value
  761. $results = chado_schema_get_foreign_key($table_desc, $field, $value, $foreign_options);
  762. if (sizeof($results) > 1) {
  763. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  764. 'chado_update_record: When trying to find update values, too many records match the criteria supplied for !foreign_key foreign key constraint (!criteria)',
  765. array('!foreign_key' => $field, '!criteria' => print_r($value, TRUE)),
  766. array('print' => $print_errors)
  767. );
  768. return FALSE;
  769. }
  770. elseif (sizeof($results) < 1) {
  771. tripal_report_error('tripal_chado', TRIPAL_DEBUG,
  772. 'chado_update_record: When trying to find update values, no record matches criteria supplied for !foreign_key foreign key constraint (!criteria)',
  773. array('!foreign_key' => $field, '!criteria' => print_r($value,TRUE)),
  774. array('print' => $print_errors)
  775. );
  776. return FALSE;
  777. }
  778. else {
  779. $update_values[$field] = $results[0];
  780. }
  781. }
  782. else {
  783. $update_values[$field] = $value;
  784. }
  785. }
  786. // now build the SQL statement
  787. $sql = 'UPDATE {' . $table . '} SET ';
  788. $args = array(); // arguments passed to chado_query
  789. foreach ($update_values as $field => $value) {
  790. if (strcmp($value, '__NULL__') == 0) {
  791. $sql .= " $field = NULL, ";
  792. }
  793. else {
  794. $sql .= " $field = :$field, ";
  795. $args[":$field"] = $value;
  796. }
  797. }
  798. $sql = drupal_substr($sql, 0, -2); // get rid of the trailing comma & space
  799. $sql .= " WHERE ";
  800. foreach ($update_matches as $field => $value) {
  801. if (strcmp($value, '__NULL__')==0) {
  802. $sql .= " $field = NULL AND ";
  803. }
  804. else {
  805. $sql .= " $field = :$field AND ";
  806. $args[":$field"] = $value;
  807. }
  808. }
  809. $sql = drupal_substr($sql, 0, -4); // get rid of the trailing 'AND'
  810. $result = chado_query($sql, $args);
  811. // if we have a result then add primary keys to return array
  812. if ($options['return_record'] == TRUE and $result) {
  813. // only if we have a single result do we want to add the primary keys to the values
  814. // array. If the update matched many records we can't add the pkeys
  815. if (count($pkeys) == 1) {
  816. foreach ($pkeys as $index => $pkey) {
  817. foreach ($pkey as $field => $fvalue) {
  818. $values[$field] = $fvalue;
  819. }
  820. }
  821. }
  822. return $values;
  823. }
  824. elseif ($options['return_record'] == FALSE and $result) {
  825. return TRUE;
  826. }
  827. else {
  828. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  829. "chado_update_record: Cannot update record in %table table. \nMatch: %match \nValues: %values",
  830. array('%table' => table, '%match' => print_r($match,TRUE), '%values' => print_r($values, 1)),
  831. array('print' => $print_errors)
  832. );
  833. return FALSE;
  834. }
  835. return FALSE;
  836. }
  837. /**
  838. * Provides a generic function for deleting a record(s) from any chado table
  839. *
  840. * Use this function to delete a record(s) in any Chado table. The first
  841. * argument specifies the table to delete from and the second is an array
  842. * of values to match for locating the record(s) to be deleted. The arrays
  843. * are mutli-dimensional such that foreign key lookup values can be specified.
  844. *
  845. * @param $table
  846. * The name of the chado table for inserting
  847. * @param $match
  848. * An associative array containing the values for locating a record to update.
  849. * @param $options
  850. * Currently there are no options
  851. * @return
  852. * On success this function returns TRUE. On failure, it returns FALSE.
  853. *
  854. * Example usage:
  855. * @code
  856. $umatch = array(
  857. 'organism_id' => array(
  858. 'genus' => 'Citrus',
  859. 'species' => 'sinensis',
  860. ),
  861. 'uniquename' => 'orange1.1g000034m.g7',
  862. 'type_id' => array (
  863. 'cv_id' => array (
  864. 'name' => 'sequence',
  865. ),
  866. 'name' => 'gene',
  867. 'is_obsolete' => 0
  868. ),
  869. );
  870. $uvalues = array(
  871. 'name' => 'orange1.1g000034m.g',
  872. 'type_id' => array (
  873. 'cv_id' => array (
  874. 'name' => 'sequence',
  875. ),
  876. 'name' => 'mRNA',
  877. 'is_obsolete' => 0
  878. ),
  879. );
  880. * $result = chado_update_record('feature', $umatch, $uvalues);
  881. * @endcode
  882. * The above code species that a feature with a given uniquename, organism_id,
  883. * and type_id (the unique constraint for the feature table) will be deleted.
  884. * The organism_id is specified as a nested array that uses the organism_id
  885. * foreign key constraint to lookup the specified values to find the exact
  886. * organism_id. The same nested struture is also used for specifying the
  887. * values to update. The function will find all records that match the
  888. * columns specified and delete them.
  889. *
  890. * @TODO: Support Complex filtering as is done in chado_select_record();
  891. *
  892. * @ingroup tripal_chado_query_api
  893. */
  894. function chado_delete_record($table, $match, $options = NULL) {
  895. $print_errors = (isset($options['print_errors'])) ? $options['print_errors'] : FALSE;
  896. if (!is_array($match)) {
  897. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  898. 'Cannot pass non array as values for matching.', array());
  899. return FALSE;
  900. }
  901. if (count($match)==0) {
  902. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  903. 'Cannot pass an empty array as values for matching.', array());
  904. return FALSE;
  905. }
  906. // set defaults for options. If we don't set defaults then
  907. // we get memory leaks when we try to access the elements
  908. if (!is_array($options)) {
  909. $options = array();
  910. }
  911. $delete_matches = array(); // contains the values for the where clause
  912. // get the table description
  913. $table_desc = chado_get_schema($table);
  914. $fields = $table_desc['fields'];
  915. if (empty($table_desc)) {
  916. tripal_report_error('tripal_chado', TRIPAL_WARNING,
  917. 'chado_delete_record; There is no table description for !table_name',
  918. array('!table_name' => $table), array('print' => $print_errors)
  919. );
  920. }
  921. // get the values needed for matching in the SQL statement
  922. foreach ($match as $field => $value) {
  923. if (is_array($value)) {
  924. // if the user has specified an array of values to delete rather than
  925. // FK relationships the keep those in our match
  926. if (array_values($value) === $value) {
  927. $delete_matches[$field] = $value;
  928. }
  929. else {
  930. $results = chado_schema_get_foreign_key($table_desc, $field, $value);
  931. if (sizeof($results) > 1) {
  932. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  933. 'chado_delete_record: When trying to find record to delete, too many records match the criteria supplied for !foreign_key foreign key constraint (!criteria)',
  934. array('!foreign_key' => $field, '!criteria' => print_r($value, TRUE)));
  935. return FALSE;
  936. }
  937. elseif (sizeof($results) < 1) {
  938. //tripal_report_error('tripal_chado', TRIPAL_ERROR, 'chado_delete_record: When trying to find record to delete, no record matches criteria supplied for !foreign_key foreign key constraint (!criteria)', array('!foreign_key' => $field, '!criteria' => print_r($value,TRUE)));
  939. }
  940. else {
  941. $delete_matches[$field] = $results[0];
  942. }
  943. }
  944. }
  945. else {
  946. $delete_matches[$field] = $value;
  947. }
  948. }
  949. // now build the SQL statement
  950. $sql = 'DELETE FROM {' . $table . '} WHERE ';
  951. $args = array();
  952. foreach ($delete_matches as $field => $value) {
  953. // if we have an array values then this is an "IN" clasue.
  954. if (count($value) > 1) {
  955. $sql .= "$field IN (";
  956. $index = 0;
  957. foreach ($value as $v) {
  958. $sql .= ":$field" . $index . ", ";
  959. $args[":$field" . $index] = $v;
  960. $index++;
  961. }
  962. $sql = drupal_substr($sql, 0, -2); // get rid of trailing ', '
  963. $sql .= ") AND ";
  964. }
  965. else {
  966. if (strcmp($value, '__NULL__') == 0) {
  967. $sql .= " $field = NULL AND ";
  968. }
  969. else {
  970. $sql .= " $field = :$field AND ";
  971. $args[":$field"] = $value;
  972. }
  973. }
  974. }
  975. $sql = drupal_substr($sql, 0, -4); // get rid of the trailing 'AND'
  976. // finally perform the delete. If successful, return the updated record
  977. $result = chado_query($sql, $args);
  978. if ($result) {
  979. return TRUE;
  980. }
  981. else {
  982. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  983. "Cannot delete record in $table table. Match:" . print_r($match, 1) . ". Values: " . print_r($values, 1), array());
  984. return FALSE;
  985. }
  986. return FALSE;
  987. }
  988. /**
  989. * Provides a generic routine for selecting data from a Chado table
  990. *
  991. * Use this function to perform a simple select from any Chado table.
  992. *
  993. * @param $table
  994. * The name of the chado table for inserting
  995. * @param $columns
  996. * An array of column names
  997. * @param $values
  998. * An associative array containing the values for filtering the results. In the
  999. * case where multiple values for the same time are to be selected an additional
  1000. * entry for the field should appear for each value. If you need to filter
  1001. * results using more complex methods see the 'Complex Filtering' section below.
  1002. * @param $options
  1003. * An associative array of additional options where the key is the option
  1004. * and the value is the value of that option.
  1005. *
  1006. * Additional Options Include:
  1007. * - has_record
  1008. * Set this argument to 'TRUE' to have this function return a numeric
  1009. * value for the number of records rather than the array of records. this
  1010. * can be useful in 'if' statements to check the presence of particula records.
  1011. * - return_sql
  1012. * Set this to 'TRUE' to have this function return an array where the first
  1013. * element is the sql that would have been run and the second is an array of
  1014. * arguments.
  1015. * - case_insensitive_columns
  1016. * An array of columns to do a case insensitive search on.
  1017. * - regex_columns
  1018. * An array of columns where the value passed in should be treated as a regular expression
  1019. * - order_by
  1020. * An associative array containing the column names of the table as keys
  1021. * and the type of sort (i.e. ASC, DESC) as the values. The results in the
  1022. * query will be sorted by the key values in the direction listed by the value
  1023. * - is_duplicate: TRUE or FALSE. Checks the values submited to see if
  1024. * they violate any of the unique constraints. If not, the record
  1025. * is returned, if so, FALSE is returned.
  1026. * - pager: Use this option if it is desired to return only a subset of results
  1027. * so that they may be shown with in a Drupal-style pager. This should be
  1028. * an array with two keys: 'limit' and 'element'. The value of 'limit'
  1029. * should specify the number of records to return and 'element' is a
  1030. * unique integer to differentiate between pagers when more than one
  1031. * appear on a page. The 'element' should start with zero and increment by
  1032. * one for each pager.
  1033. * -limit: Specifies the number of records to return.
  1034. * -offset: Indicates the number of records to skip before returning records.
  1035. *
  1036. * @return
  1037. * An array of results, FALSE if the query was not executed
  1038. * correctly, an empty array if no records were matched, or the number of records
  1039. * in the dataset if $has_record is set.
  1040. * If the option 'is_duplicate' is provided and the record is a duplicate it
  1041. * will return the duplicated record. If the 'has_record' option is provided
  1042. * a value of TRUE will be returned if a record exists and FALSE will bee
  1043. * returned if there are not records.
  1044. *
  1045. * Example usage:
  1046. * @code
  1047. * $columns = array('feature_id', 'name');
  1048. * $values = array(
  1049. * 'organism_id' => array(
  1050. * 'genus' => 'Citrus',
  1051. * 'species' => array('sinensis', 'clementina'),
  1052. * ),
  1053. * 'uniquename' => 'orange1.1g000034m.g',
  1054. * 'type_id' => array (
  1055. * 'cv_id' => array (
  1056. * 'name' => 'sequence',
  1057. * ),
  1058. * 'name' => 'gene',
  1059. * 'is_obsolete' => 0
  1060. * ),
  1061. * );
  1062. * $options = array(
  1063. * 'order_by' => array(
  1064. * 'name' => 'ASC'
  1065. * ),
  1066. * );
  1067. * $result = chado_select_record('feature',$columns,$values,$options);
  1068. * @endcode
  1069. * The above code selects a record from the feature table using the three fields
  1070. * that uniquely identify a feature. The $columns array simply lists the columns
  1071. * to select. The $values array is nested such that the organism is identified by
  1072. * way of the organism_id foreign key constraint by specifying the genus and
  1073. * species. The cvterm is also specified using its foreign key and the cv_id
  1074. * for the cvterm is nested as well. In the example above, two different species
  1075. * are allowed to match
  1076. *
  1077. * Complex Filtering:
  1078. * All of the documentation above supports filtering based on 'is equal to'
  1079. * or 'is NULL'. If your criteria doesn't fall into one of these two categories
  1080. * then you need to provide an array with additional details such as the operator
  1081. * as well as the value. An example follows and will be discussed in detail.
  1082. * @code
  1083. $columns = array('feature_id', 'fmin', 'fmax');
  1084. // Regular criteria specifying the parent feature to retrieve locations from.
  1085. $values = array(
  1086. 'srcfeature_id' => array(
  1087. 'uniquename' => 'MtChr01'
  1088. 'type_id' => array(
  1089. 'name' => 'pseudomolecule'
  1090. ),
  1091. ),
  1092. );
  1093. // Complex filtering to specify the range to return locations from.
  1094. $values['fmin'][] = array(
  1095. 'op' => '>',
  1096. 'data' => 15
  1097. );
  1098. $values['fmin'][] = array(
  1099. 'op' => '<',
  1100. 'data' => 100
  1101. );
  1102. $results = chado_select_record('featureloc', $columns, $values);
  1103. * @endcode
  1104. * The above code example will return all of the name, start and end of all
  1105. * the features that start within MtChr1:15-100bp. Note that complex filtering
  1106. * can be used in conjunction with basic filtering and that multiple criteria,
  1107. * even for the same field can be entered.
  1108. *
  1109. * @ingroup tripal_chado_query_api
  1110. */
  1111. function chado_select_record($table, $columns, $values, $options = NULL) {
  1112. // Set defaults for options. If we don't set defaults then
  1113. // we get memory leaks when we try to access the elements.
  1114. if (!is_array($options)) {
  1115. $options = array();
  1116. }
  1117. if (!array_key_exists('case_insensitive_columns', $options)) {
  1118. $options['case_insensitive_columns'] = array();
  1119. }
  1120. if (!array_key_exists('regex_columns', $options)) {
  1121. $options['regex_columns'] = array();
  1122. }
  1123. if (!array_key_exists('order_by', $options)) {
  1124. $options['order_by'] = array();
  1125. }
  1126. if (!array_key_exists('return_sql', $options)) {
  1127. $options['return_sql'] = FALSE;
  1128. }
  1129. if (!array_key_exists('has_record', $options)) {
  1130. $options['has_record'] = FALSE;
  1131. }
  1132. if (!array_key_exists('is_duplicate', $options)) {
  1133. $options['is_duplicate'] = FALSE;
  1134. }
  1135. $pager = array();
  1136. if (array_key_exists('pager', $options)) {
  1137. $pager = $options['pager'];
  1138. }
  1139. $print_errors = FALSE;
  1140. if (isset($options['print_errors'])) {
  1141. $print_errors = $options['print_errors'];
  1142. }
  1143. // Check that our columns and values arguments are proper arrays.
  1144. if (!is_array($columns)) {
  1145. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1146. 'chado_select_record; the $columns argument must be an array. Columns:%columns',
  1147. array('%columns' => print_r($columns, TRUE)),
  1148. array('print' => $print_errors)
  1149. );
  1150. return FALSE;
  1151. }
  1152. if (!is_array($values)) {
  1153. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1154. 'chado_select_record; the $values argument must be an array. Values:%values',
  1155. array('%values' => print_r($values, TRUE)),
  1156. array('print' => $print_errors)
  1157. );
  1158. return FALSE;
  1159. }
  1160. // Get the table description.
  1161. $table_desc = chado_get_schema($table);
  1162. if (!is_array($table_desc)) {
  1163. tripal_report_error('tripal_chado', TRIPAL_WARNING,
  1164. 'chado_insert_record; There is no table description for !table_name',
  1165. array('!table_name' => $table), array('print' => $print_errors)
  1166. );
  1167. return FALSE;
  1168. }
  1169. $where = array();
  1170. $args = array();
  1171. if ($options['is_duplicate'] and array_key_exists('unique keys', $table_desc)) {
  1172. $ukeys = $table_desc['unique keys'];
  1173. $has_results = 0;
  1174. // iterate through the unique constraints and reset the values and columns
  1175. // arrays to only include these fields
  1176. foreach ($ukeys as $cname => $fields) {
  1177. if ($has_results) {
  1178. continue;
  1179. }
  1180. $new_values = array();
  1181. $new_columns = array();
  1182. $new_options = array();
  1183. $has_pkey = 0;
  1184. // include the primary key in the results returned
  1185. if (array_key_exists('primary key', $table_desc)) {
  1186. $has_pkey = 1;
  1187. $pkeys = $table_desc['primary key'];
  1188. foreach ($pkeys as $index => $key) {
  1189. array_push($new_columns, $key);
  1190. }
  1191. }
  1192. // recreate the $values and $columns arrays
  1193. foreach ($fields as $field) {
  1194. if (array_key_exists($field, $values)) {
  1195. $new_values[$field] = $values[$field];
  1196. // if there is no primary key then use the unique contraint fields
  1197. if (!$has_pkey) {
  1198. array_push($new_columns, $field);
  1199. }
  1200. }
  1201. // if the field doesn't exist in the values array then
  1202. // substitute any default values
  1203. elseif (array_key_exists('default', $table_desc['fields'][$field])) {
  1204. $new_values[$field] = $table_desc['fields'][$field]['default'];
  1205. if (!$has_pkey) {
  1206. array_push($new_columns, $field);
  1207. }
  1208. }
  1209. // if there is no value (default or otherwise) check if this field is
  1210. // allowed to be null
  1211. elseif (!$table_desc['fields'][$field]['not null']) {
  1212. $new_values[$field] = NULL;
  1213. if (!$has_pkey) {
  1214. array_push($new_columns, $field);
  1215. }
  1216. }
  1217. // if the array key doesn't exist in the values given by the caller
  1218. // and there is no default value then we cannot check if the record
  1219. // is a duplicate so return FALSE
  1220. else {
  1221. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1222. 'chado_select_record: There is no value for %field thus we cannot ' .
  1223. 'check if this record for table, %table, is unique. %values',
  1224. array('%field' => $field, '%table' => $table, '%values' => print_r($values, TRUE)),
  1225. array('print' => $print_errors));
  1226. return FALSE;
  1227. }
  1228. }
  1229. $results = chado_select_record($table, $new_columns, $new_values, $new_options);
  1230. // if we have a duplicate record then return the results
  1231. if (count($results) > 0) {
  1232. $has_results = 1;
  1233. }
  1234. unset($new_columns);
  1235. unset($new_values);
  1236. unset($new_options);
  1237. }
  1238. if ($options['has_record'] and $has_results) {
  1239. return TRUE;
  1240. }
  1241. else {
  1242. return $results;
  1243. }
  1244. }
  1245. // Process the values array into where clauses and retrieve foreign keys. The
  1246. // $where array should always be an integer-indexed array with each value
  1247. // being an array with a 'field', 'op', and 'data' keys with all foreign keys
  1248. // followed.
  1249. foreach ($values as $field => $value) {
  1250. // Require the field be in the table description.
  1251. if (!array_key_exists($field, $table_desc['fields'])) {
  1252. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1253. 'chado_select_record: The field "%field" does not exist in the table "%table". Cannot perform query. Values: %array. Fields: %fields',
  1254. array(
  1255. '%field' => $field,
  1256. '%table' => $table,
  1257. '%array' => print_r($values, 1),
  1258. '%fields' => print_r($table_desc['fields'], 1)),
  1259. array('print' => $print_errors)
  1260. );
  1261. return array();
  1262. }
  1263. // CASE 1: We have an array for a value.
  1264. if (is_array($value)) {
  1265. // CASE 1a: If there is only one element in the array, treat it the same
  1266. // as a non-array value.
  1267. if (count($value) == 1 AND is_int(key($value))) {
  1268. $value = array_pop($value);
  1269. $op = '=';
  1270. chado_select_record_check_value_type($op, $value, $table_desc['fields'][$field]['type']);
  1271. $where[] = array(
  1272. 'field' => $field,
  1273. 'op' => $op,
  1274. 'data' => $value
  1275. );
  1276. }
  1277. // CASE 1b: If there is a 'data' key in the array then we have the new
  1278. // complex filtering format with a single criteria.
  1279. elseif (isset($value['data']) AND isset($value['op'])) {
  1280. $value['field'] = $field;
  1281. $where[] = $value;
  1282. }
  1283. // CASE 1c: If we have an integer indexed array and the first element is
  1284. // not an array then we have a simple array of values to be used for an IN clause.
  1285. elseif (is_int(key($value)) AND !is_array(current($value))) {
  1286. $where[] = array(
  1287. 'field' => $field,
  1288. 'op' => 'IN',
  1289. 'data' => $value
  1290. );
  1291. }
  1292. // We have a multi-dimensional array: 2 cases...
  1293. else {
  1294. // CASE 1d: If there is a multi-dimensional array with each sub-array
  1295. // containing a data key then we have the new complex filtering format
  1296. // with multiple criteria.
  1297. if (isset($value[0]['data']) AND isset($value[0]['op'])) {
  1298. foreach ($value as $subvalue) {
  1299. $subvalue['field'] = $field;
  1300. $where[] = $subvalue;
  1301. }
  1302. }
  1303. // CASE 1e: We have a multi-dimensional array that doesn't fit any of the
  1304. // above cases then we have a foreign key definition to follow.
  1305. else {
  1306. // Select the value from the foreign key relationship for this value.
  1307. $foreign_options = array(
  1308. 'regex_columns' => $options['regex_columns'],
  1309. );
  1310. $results = chado_schema_get_foreign_key($table_desc, $field, $value, $foreign_options);
  1311. // Ensure that looking up the foreign key didn't fail in an error.
  1312. if ($results === FALSE OR $results === NULL) {
  1313. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1314. 'chado_select_record: could not follow the foreign key definition
  1315. for %field where the definition supplied was %value',
  1316. array('%field' => $field, '%value' => print_r($value, TRUE))
  1317. );
  1318. return array();
  1319. }
  1320. // Ensure that there were results returned.
  1321. elseif (count($results)==0) {
  1322. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1323. 'chado_select_record: the foreign key definition for \'%field\' on table \'%table\' '.
  1324. 'returned no results where the definition supplied was %value',
  1325. array('%field' => $field, '%table' => $table, '%value' => print_r($value, TRUE))
  1326. );
  1327. return array();
  1328. }
  1329. // If there was only a single resutlt then add it using an op of =.
  1330. elseif (count($results) == 1) {
  1331. $results = array_pop($results);
  1332. $op = '=';
  1333. chado_select_record_check_value_type($op, $results, $table_desc['fields'][$field]['type']);
  1334. $where[] = array(
  1335. 'field' => $field,
  1336. 'op' => $op,
  1337. 'data' => $results
  1338. );
  1339. }
  1340. // Otherwise multiple results were returned so we want to form an
  1341. // IN (x, y, z) expression.
  1342. else {
  1343. $where[] = array(
  1344. 'field' => $field,
  1345. 'op' => 'IN',
  1346. 'data' => $results
  1347. );
  1348. }
  1349. }
  1350. }
  1351. }
  1352. // CASE 2: We have a single value.
  1353. else {
  1354. $op = '=';
  1355. chado_select_record_check_value_type($op, $value, $table_desc['fields'][$field]['type']);
  1356. $where[] = array(
  1357. 'field' => $field,
  1358. 'op' => $op,
  1359. 'data' => $value
  1360. );
  1361. }
  1362. // Support Deprecated method for regex conditions.
  1363. $current_key = key($where);
  1364. if (in_array($field, $options['regex_columns'])) {
  1365. $where[$current_key]['op'] = '~*';
  1366. }
  1367. }
  1368. // Now build the SQL.
  1369. if (empty($where)) {
  1370. // Sometimes want to select everything.
  1371. $sql = "SELECT " . implode(', ', $columns) . " ";
  1372. $sql .= 'FROM {' . $table . '} ';
  1373. }
  1374. else {
  1375. $sql = "SELECT " . implode(', ', $columns) . " ";
  1376. $sql .= 'FROM {' . $table . '} ';
  1377. // If $values is empty then we want all results so no where clause.
  1378. if (!empty($values)) {
  1379. $sql .= "WHERE ";
  1380. }
  1381. foreach ($where as $clause_num => $value_def) {
  1382. switch ($value_def['op']) {
  1383. // Deal with 'field IN (x, y, z)' where clauses.
  1384. case 'IN':
  1385. $sql .= $value_def['field'] . " IN (";
  1386. $index = 0;
  1387. foreach ($value_def['data'] as $v) {
  1388. $placeholder = ':' . $value_def['field'] . $clause_num .'_' . $index;
  1389. $sql .= $placeholder . ', ';
  1390. $args[$placeholder] = $v;
  1391. $index++;
  1392. }
  1393. $sql = drupal_substr($sql, 0, -2); // remove trailing ', '
  1394. $sql .= ") AND ";
  1395. break;
  1396. // Deal with IS NULL.
  1397. case 'IS NULL':
  1398. $sql .= $value_def['field'] . ' IS NULL AND ';
  1399. break;
  1400. // Default is [field] [op] [data].
  1401. default:
  1402. $placeholder = ':'. $value_def['field'] . $clause_num;
  1403. // Support case insensitive columns.
  1404. if (in_array($value_def['field'], $options['case_insensitive_columns'])) {
  1405. $sql .= 'lower(' . $value_def['field'] .') '. $value_def['op'] .' lower('. $placeholder . ') AND ';
  1406. }
  1407. else {
  1408. $sql .= $value_def['field'] .' '. $value_def['op'] .' '. $placeholder . ' AND ';
  1409. }
  1410. $args[$placeholder] = $value_def['data'];
  1411. }
  1412. } // end foreach item in where clause.
  1413. $sql = drupal_substr($sql, 0, -4); // get rid of the trailing 'AND '
  1414. } // end if (empty($where)){ } else {
  1415. // Add any ordering of the results to the SQL statement.
  1416. if (count($options['order_by']) > 0) {
  1417. $sql .= " ORDER BY ";
  1418. foreach ($options['order_by'] as $field => $dir) {
  1419. $sql .= "$field $dir, ";
  1420. }
  1421. $sql = drupal_substr($sql, 0, -2); // get rid of the trailing ', '
  1422. }
  1423. // Limit the records returned
  1424. if (array_key_exists('limit', $options) and is_numeric($options['limit'])) {
  1425. $sql .= " LIMIT " . $options['limit'];
  1426. if (array_key_exists('offset', $options) and is_numeric($options['offset'])) {
  1427. $sql .= " OFFSET " . $options['offset'];
  1428. }
  1429. }
  1430. // if the caller has requested the SQL rather than the results then do so.
  1431. if ($options['return_sql'] == TRUE) {
  1432. return array('sql' => $sql, 'args' => $args);
  1433. }
  1434. if (array_key_exists('limit', $pager)) {
  1435. $total_records = 0;
  1436. $resource = chado_pager_query($sql, $args, $pager['limit'], $pager['element'], NULL, $total_records);
  1437. }
  1438. else {
  1439. $resource = chado_query($sql, $args);
  1440. }
  1441. // Format results into an array.
  1442. $results = array();
  1443. foreach ($resource as $r) {
  1444. $results[] = $r;
  1445. }
  1446. if ($options['has_record']) {
  1447. return count($results);
  1448. }
  1449. return $results;
  1450. }
  1451. /**
  1452. * Helper Function: check that the value is the correct type.
  1453. *
  1454. * This function is used by chado_select_record() when building the $where
  1455. * clause array to ensure that any single values are the correct type based
  1456. * on the table definition. Furthermore, it ensures that NULL's are caught
  1457. * changing the operator to 'IS NULL'.
  1458. * @code
  1459. $op = '=';
  1460. chado_select_record_check_value_type($op, $value, $table_desc['fields'][$field]['type']);
  1461. $where[] = array(
  1462. 'field' => $field,
  1463. 'op' => $op,
  1464. 'data' => $value
  1465. );
  1466. * @endcode
  1467. *
  1468. * @param $op
  1469. * The operator being used. This is mostly passed in to allow it to be changed
  1470. * if a NULL value is detected.
  1471. * @param $value
  1472. * The value to be checked and adjusted.
  1473. * @param $type
  1474. * The type from the table definition that's used to determine the type of
  1475. * value.
  1476. */
  1477. function chado_select_record_check_value_type(&$op, &$value, $type) {
  1478. if ($value === NULL) {
  1479. $op = 'IS NULL';
  1480. }
  1481. elseif ($type == 'int') {
  1482. $value = (int) $value;
  1483. }
  1484. }
  1485. /**
  1486. * A substitute for db_query() when querying from Chado.
  1487. *
  1488. * This function is needed to avoid switching databases when making query to
  1489. * the chado database.
  1490. *
  1491. * Will use a chado persistent connection if it already exists.
  1492. *
  1493. * @param $sql
  1494. * The sql statement to execute
  1495. *
  1496. * @param $args
  1497. * The array of arguments, with the same structure as passed to
  1498. * the db_query() function of Drupal.
  1499. *
  1500. * @return
  1501. * DatabaseStatementInterface A prepared statement object, already executed.
  1502. *
  1503. * Example usage:
  1504. * @code
  1505. * $sql = "SELECT F.name, CVT.name as type_name, ORG.common_name
  1506. * FROM {feature} F
  1507. * LEFT JOIN {cvterm} CVT ON F.type_id = CVT.cvterm_id
  1508. * LEFT JOIN {organism} ORG ON F.organism_id = ORG.organism_id
  1509. * WHERE
  1510. * F.uniquename = :feature_uniquename";
  1511. * $args = array( ':feature_uniquename' => $form_state['values']['uniquename'] );
  1512. * $result = chado_query($sql, $args);
  1513. * foreach ($result as $r) { [Do something with the records here] }
  1514. * @endcode
  1515. *
  1516. * @ingroup tripal_chado_query_api
  1517. */
  1518. function chado_query($sql, $args = array()) {
  1519. $results = NULL;
  1520. $is_local = isset($GLOBALS["chado_is_local"]) && $GLOBALS["chado_is_local"];
  1521. // Args should be an array
  1522. if (!is_array($args)) {
  1523. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1524. 'chado_query; Need to pass an array to chado_query, "%value" passed instead. Query: %query',
  1525. array('%value' => $args, '%query' => $sql)
  1526. );
  1527. return FALSE;
  1528. }
  1529. // if Chado is local to the database then prefix the Chado table
  1530. // names with 'chado'.
  1531. if ($is_local) {
  1532. // Remove carriage returns from the SQL.
  1533. $sql = preg_replace('/\n/', '', $sql);
  1534. // Prefix the tables with their correct schema.
  1535. // Chado tables should be enclosed in curly brackets (ie: {feature} )
  1536. // and Drupal tables should be enclosed in square brackets
  1537. // (ie: [tripal_jobs] ).
  1538. $chado_schema_name = tripal_get_schema_name('chado');
  1539. $drupal_schema_name = tripal_get_schema_name('drupal');
  1540. $sql = preg_replace('/\{(.*?)\}/', $chado_schema_name.'.$1', $sql);
  1541. $sql = preg_replace('/\[(\w+)\]/', $drupal_schema_name.'.$1', $sql);
  1542. // Add an alter hook to allow module developers to change the query right before it's
  1543. // executed. Since all queriying of chado by Tripal eventually goes through this
  1544. // function, we only need to provide an alter hook at this point in order to ensure
  1545. // developers have complete control over the query being executed. For example,
  1546. // a module developer might want to remove schema prefixing from queries and rely
  1547. // on the search path. This alter hook would allow them to do that by implementing
  1548. // mymodule_chado_query_alter($sql, $args) and using a regular expression to remove
  1549. // table prefixing from the query.
  1550. // @see hook_chado_query_alter().
  1551. drupal_alter('chado_query', $sql, $args);
  1552. // The featureloc table has some indexes that use function that call other
  1553. // functions and those calls do not reference a schema, therefore, any tables with featureloc
  1554. // must automaticaly have the chado schema set as active to find
  1555. if (preg_match('/'.$chado_schema_name . '.featureloc/i', $sql) or preg_match('/' . $chado_schema_name . '.feature/i', $sql)) {
  1556. $previous_db = chado_set_active('chado') ;
  1557. try {
  1558. $results = db_query($sql, $args);
  1559. chado_set_active($previous_db);
  1560. }
  1561. catch (Exception $e) {
  1562. chado_set_active($previous_db);
  1563. throw $e;
  1564. }
  1565. }
  1566. // for all other tables we should have everything in scope so just run the query
  1567. else {
  1568. $results = db_query($sql, $args);
  1569. }
  1570. }
  1571. // Check for any cross schema joins (ie: both drupal and chado tables
  1572. // represented and if present don't execute the query but instead warn the
  1573. // administrator.
  1574. else if (preg_match('/\[(\w*?)\]/', $sql)) {
  1575. tripal_report_error('chado_query', TRIPAL_ERROR,
  1576. 'The following query does not support external chado databases. Please file an issue with the Drupal.org Tripal Project. Query: @query',
  1577. array('@query' => $sql)
  1578. );
  1579. return FALSE;
  1580. }
  1581. // if Chado is not local to the Drupal database then we have to
  1582. // switch to another database
  1583. else {
  1584. $previous_db = chado_set_active('chado');
  1585. $results = db_query($sql, $args);
  1586. chado_set_active($previous_db);
  1587. }
  1588. return $results;
  1589. }
  1590. /**
  1591. * This hook provides a way for module developers to alter any/all queries on the chado
  1592. * schema by Tripal.
  1593. *
  1594. * Example: a module developer might want to remove schema prefixing from queries and rely
  1595. * on the search path. This alter hook would allow them to do that by implementing
  1596. * mymodule_chado_query_alter($sql, $args) and using a regular expression to remove
  1597. * table prefixing from the query.
  1598. *
  1599. * @param $sql
  1600. * A string describing the SQL query to be executed by Tripal. All parameters should be
  1601. * indicated by :tokens with values being in the $args array and all tables should
  1602. * be prefixed with the schema name described in tripal_get_schema_name().
  1603. * @param $args
  1604. * An array of arguments where the key is the token used in $sql (for example, :value)
  1605. * and the value is the value you would like substituted in.
  1606. *
  1607. * @ingroup tripal_chado_query_api
  1608. */
  1609. function hook_chado_query_alter(&$sql, &$args) {
  1610. // The following code is an example of how this alter function might be used.
  1611. // Say you would like only a portion of node => feature connections available
  1612. // for a period of time or under a specific condition. To "hide" the other connections
  1613. // you might create a temporary view of the chado_feature table that only includes
  1614. // the connections you would like to be available. In order to ensure this view
  1615. // is used rather than the original chado_feature table you could alter all Tripal
  1616. // queries referring to chado_feature to instead refer to your view.
  1617. if (preg_match('/(\w+)\.chado_feature/', $sql, $matches)) {
  1618. $sql = str_replace(
  1619. $matches[1] . '.chado_feature',
  1620. 'chado_feature_view',
  1621. $sql
  1622. );
  1623. }
  1624. }
  1625. /**
  1626. * Use this function instead of pager_query() when selecting a
  1627. * subset of records from a Chado table.
  1628. *
  1629. * @param $query
  1630. * The SQL statement to execute, this is followed by a variable number of args
  1631. * used as substitution values in the SQL statement.
  1632. * @param $args
  1633. * The array of arguments for the query. They keys are the placeholders
  1634. * @param $limit
  1635. * The number of query results to display per page.
  1636. * @param $element
  1637. * An numeric identifier used to distinguish between multiple pagers on one page.
  1638. * @param $count_query
  1639. * An SQL query used to count matching records.
  1640. *
  1641. * @returns
  1642. * A database query result resource or FALSE if the query was not
  1643. * executed correctly
  1644. *
  1645. * @ingroup tripal_chado_query_api
  1646. */
  1647. function chado_pager_query($query, $args, $limit, $element, $count_query = '') {
  1648. // get the page and offset for the pager
  1649. $page_arg = isset($_GET['page']) ? $_GET['page'] : '0';
  1650. $pages = explode(',', $page_arg);
  1651. $page = 0;
  1652. if (count($pages) >= $element) {
  1653. $page = key_exists($element, $pages) ? $pages[$element] : 0;
  1654. }
  1655. $offset = $limit * $page;
  1656. $q = $_GET['q'];
  1657. // Construct a count query if none was given.
  1658. if (!isset($count_query)) {
  1659. $count_query = preg_replace(array('/SELECT.*?FROM /As', '/ORDER BY .*/'),
  1660. array('SELECT COUNT(*) FROM ', ''), $query);
  1661. }
  1662. // We calculate the total of pages as ceil(items / limit).
  1663. $results = chado_query($count_query, $args);
  1664. if (!$results) {
  1665. tripal_report_error('tripal_chado', TRIPAL_ERROR,
  1666. "chado_pager_query(): Query failed: %cq", array('%cq' => $count_query));
  1667. return;
  1668. }
  1669. $total_records = $results->fetchField();
  1670. // set a session variable for storing the total number of records
  1671. $GLOBALS['chado_pager'][$q][$element]['total_records'] = $total_records;
  1672. pager_default_initialize($total_records, $limit, $element);
  1673. $query .= ' LIMIT ' . (int) $limit . ' OFFSET ' . (int) $offset;
  1674. $results = chado_query($query, $args);
  1675. return $results;
  1676. }
  1677. /**
  1678. * A function to retrieve the total number of records for a pager that
  1679. * was generated using the chado_pager_query() function
  1680. *
  1681. * @param $element
  1682. * The $element argument that was passed to the chado_pager_query function
  1683. *
  1684. * @ingroup tripal_chado_query_api
  1685. */
  1686. function chado_pager_get_count($element) {
  1687. $q = $_GET['q'];
  1688. if (array_key_exists($q, $GLOBALS['chado_pager']) and
  1689. array_key_exists($element, $GLOBALS['chado_pager'][$q])) {
  1690. return $GLOBALS['chado_pager'][$q][$element]['total_records'];
  1691. }
  1692. else {
  1693. return 0;
  1694. }
  1695. }
  1696. /**
  1697. * Gets the value of a foreign key relationship
  1698. *
  1699. * This function is used by chado_select_record, chado_insert_record,
  1700. * and chado_update_record to iterate through the associate array of
  1701. * values that gets passed to each of those routines. The values array
  1702. * is nested where foreign key contraints are used to specify a value that. See
  1703. * documentation for any of those functions for further information.
  1704. *
  1705. * @param $table_desc
  1706. * A table description for the table with the foreign key relationship to be identified generated by
  1707. * hook_chado_<table name>_schema()
  1708. * @param $field
  1709. * The field in the table that is the foreign key.
  1710. * @param $values
  1711. * An associative array containing the values
  1712. * @param $options
  1713. * An associative array of additional options where the key is the option
  1714. * and the value is the value of that option. These options are passed on to chado_select_record.
  1715. *
  1716. * Additional Options Include:
  1717. * - case_insensitive_columns
  1718. * An array of columns to do a case insensitive search on.
  1719. * - regex_columns
  1720. * An array of columns where the value passed in should be treated as a regular expression
  1721. *
  1722. * @return
  1723. * A string containg the results of the foreign key lookup, or FALSE if failed.
  1724. *
  1725. * Example usage:
  1726. * @code
  1727. *
  1728. * $values = array(
  1729. * 'genus' => 'Citrus',
  1730. * 'species' => 'sinensis',
  1731. * );
  1732. * $value = chado_schema_get_foreign_key('feature', 'organism_id',$values);
  1733. *
  1734. * @endcode
  1735. * The above code selects a record from the feature table using the three fields
  1736. * that uniquely identify a feature. The $columns array simply lists the columns
  1737. * to select. The $values array is nested such that the organism is identified by
  1738. * way of the organism_id foreign key constraint by specifying the genus and
  1739. * species. The cvterm is also specified using its foreign key and the cv_id
  1740. * for the cvterm is nested as well.
  1741. *
  1742. */
  1743. function chado_schema_get_foreign_key($table_desc, $field, $values, $options = NULL) {
  1744. // set defaults for options. If we don't set defaults then
  1745. // we get memory leaks when we try to access the elements
  1746. if (!is_array($options)) {
  1747. $options = array();
  1748. }
  1749. if (!array_key_exists('case_insensitive_columns', $options)) {
  1750. $options['case_insensitive_columns'] = array();
  1751. }
  1752. if (!array_key_exists('regex_columns', $options)) {
  1753. $options['regex_columns'] = array();
  1754. }
  1755. // get the list of foreign keys for this table description and
  1756. // iterate through those until we find the one we're looking for
  1757. $fkeys = '';
  1758. if (array_key_exists('foreign keys', $table_desc)) {
  1759. $fkeys = $table_desc['foreign keys'];
  1760. }
  1761. if ($fkeys) {
  1762. foreach ($fkeys as $name => $def) {
  1763. if (is_array($def['table'])) {
  1764. //foreign key was described 2X
  1765. $message = "The foreign key " . $name . " was defined twice. Please check modules "
  1766. . "to determine if hook_chado_schema_<version>_" . $table_desc['table'] . "() was "
  1767. . "implemented and defined this foreign key when it wasn't supposed to. Modules "
  1768. . "this hook was implemented in: " . implode(', ',
  1769. module_implements("chado_" . $table_desc['table'] . "_schema")) . ".";
  1770. tripal_report_error('tripal_chado', $message);
  1771. drupal_set_message(check_plain($message), 'error');
  1772. continue;
  1773. }
  1774. $table = $def['table'];
  1775. $columns = $def['columns'];
  1776. // iterate through the columns of the foreign key relationship
  1777. foreach ($columns as $left => $right) {
  1778. // does the left column in the relationship match our field?
  1779. if (strcmp($field, $left) == 0) {
  1780. // the column name of the foreign key matches the field we want
  1781. // so this is the right relationship. Now we want to select
  1782. $select_cols = array($right);
  1783. $result = chado_select_record($table, $select_cols, $values, $options);
  1784. $fields = array();
  1785. if ($result and count($result) > 0) {
  1786. foreach ($result as $obj) {
  1787. $fields[] = $obj->$right;
  1788. }
  1789. return $fields;
  1790. }
  1791. }
  1792. }
  1793. }
  1794. }
  1795. else {
  1796. // @todo: what do we do if we get to this point and we have a fk
  1797. // relationship expected but we don't have any definition for one in the
  1798. // table schema??
  1799. $version = $GLOBALS["chado_version"];
  1800. $message = t("There is no foreign key relationship defined for " . $field . " .
  1801. To define a foreign key relationship, determine the table this foreign
  1802. key referrs to (<foreign table>) and then implement
  1803. hook_chado_chado_schema_v<version>_<foreign table>(). See
  1804. tripal_chado_chado_v1_2_schema_feature for an example. Chado version: $version");
  1805. tripal_report_error('tripal_chado', TRIPAL_ERROR, $message);
  1806. drupal_set_message(check_plain($message), 'error');
  1807. }
  1808. return array();
  1809. }
  1810. /**
  1811. * Retrieve the name of the PostgreSQL schema housing Chado or Drupal.
  1812. *
  1813. * @param $schema
  1814. * Wehter you want the schema name for 'chado' or 'drupal'. Chado is the default.
  1815. * @return
  1816. * The name of the PostgreSQL schema housing the $schema specified.
  1817. *
  1818. * @ingroup tripal_chado_query_api
  1819. */
  1820. function tripal_get_schema_name($schema = 'chado') {
  1821. // First we will set our default. This is what will be returned in most cases.
  1822. if ($schema == 'chado') {
  1823. $schema_name = 'chado';
  1824. }
  1825. else {
  1826. $schema_name = 'public';
  1827. }
  1828. // There are cases where modules or admin might need to change the default
  1829. // names for the schema. Thus we provide an alter hook here to allow
  1830. // the names to be changed and ensure that schema names are never hardcoded
  1831. // directly into queries.
  1832. $context = array('schema' => $schema);
  1833. drupal_alter('tripal_get_schema_name', $schema_name, $context);
  1834. return $schema_name;
  1835. }
  1836. /**
  1837. * Alter the name of the schema housing Chado and/or Drupal.
  1838. *
  1839. * This example implementation shows a solution for the case where your chado database
  1840. * was well established in the "public" schema and you added Drupal later in a
  1841. * "drupal" schema. Please note that this has not been tested and while we can ensure
  1842. * that Tripal will work as expected, we have no control over whether Drupal is
  1843. * compatible with not being in the public schema. That's why we recommened the
  1844. * organization we have (ie: Chado in a "chado" schema and Drupal in the "public schema).
  1845. *
  1846. * @param $schema_name
  1847. * The current name of the schema as known by Tripal. This is likely the default
  1848. * set in tripal_get_schema_name() but in the case of multiple alter hooks, it might
  1849. * be different.
  1850. * @param $context
  1851. * This is an array of items to provide context.
  1852. * - schema: this is the schema that was passed to tripal_get_schema_name() and will
  1853. * be either "chado" or "drupal". This should be used to determine you are changing
  1854. * the name of the correct schema.
  1855. *
  1856. * @ingroup tripal_chado_query_api
  1857. */
  1858. function hook_tripal_get_schema_name_alter($schema_name, $context) {
  1859. // First we check which schema was passed to chado_get_schema().
  1860. // Notice that we do not use $schema_name since it may already have
  1861. // been altered by another module.
  1862. if ($context['schema'] == 'chado') {
  1863. $schema_name = 'public';
  1864. }
  1865. // Notice that we use elseif to capture the second case rather than else. This
  1866. // avoids the assumption that there is only one chado and one drupal schema.
  1867. elseif ($context['schema'] == 'drupal') {
  1868. $schema_name = 'drupal';
  1869. }
  1870. }
  1871. /**
  1872. * A replacment for db_select when querying Chado.
  1873. *
  1874. * Use this function instead of db_select when querying Chado tables.
  1875. *
  1876. * @param $table
  1877. * The base table for this query. May be a string or another SelectQuery
  1878. * object. If a query object is passed, it will be used as a subselect.
  1879. * @param $alias
  1880. * The alias for the base table of this query.
  1881. * @param $options
  1882. * An array of options to control how the query operates.
  1883. *
  1884. * @return
  1885. * A new SelectQuery object for this connection.
  1886. *
  1887. * @ingroup tripal_chado_query_api
  1888. */
  1889. function chado_db_select($table, $alias = NULL, array $options = array()) {
  1890. if (empty($options['target'])) {
  1891. $options['target'] = 'default';
  1892. }
  1893. $conninfo = Database::getConnectionInfo();
  1894. $conn = new ChadoDatabaseConnection($conninfo['default']);
  1895. return $conn->select($table, $alias, $options);
  1896. }