Adding support for mysql_data_seek()

1 files changed, 6 insertions, 0 deletions

diff --git a/taxonomy_node_tree.module b/taxonomy_node_tree.module
index 0905ac8..5b9ab7f 100644
--- a/taxonomy_node_tree.module
+++ b/taxonomy_node_tree.module
@@ -52,6 +52,9 @@ function taxonomy_node_tree_parents($vid) {
  * @todo
  *   It is assumed that nodes are just associated with a single term.
  *   This could be changed to support multiple relationships.
+ *
+ *   Also, current result rewind method should support postgresql
+ *   data structures.
  */
 function taxonomy_node_tree_index($nodes, $terms) {
   while ($node = db_fetch_object($nodes)) {
@@ -69,6 +72,9 @@ function taxonomy_node_tree_index($nodes, $terms) {
   if (is_object($nodes)) {
     $nodes->data_seek(0);
   }
+  else {
+    mysql_data_seek($nodes, 0);
+  }
 
   return $tree;
 }