Changeset 19 for trunk/DNSDB.pm

Timestamp:

10/09/09 17:49:14 (15 years ago)

Author:

Kris Deugau

Message:

/trunk

checkpoint

File:

• trunk/DNSDB.pm (modified) (3 diffs)

trunk/DNSDB.pm

@@ -23 +23 @@
 @ISA            = qw(Exporter);
 @EXPORT_OK      = qw(
-        &initGlobals &connectDB &finish &addDomain &delDomain &domainName &addGroup &grpName &getSOA
+        &initGlobals &connectDB &finish &addDomain &delDomain &domainName &addGroup &getChildren &grpName &getSOA
         &getRecLine &getDomRecs &addRec &updateRec &delRec &domStatus
         %typemap %reverse_typemap
@@ -30 +30 @@
 @EXPORT         = (); # Export nothing by default.
 %EXPORT_TAGS    = ( ALL => [qw(
-                &initGlobals &connectDB &finish &addDomain &delDomain &domainName &addGroup &grpName &getSOA
+                &initGlobals &connectDB &finish &addDomain &delDomain &domainName &addGroup &getChildren &grpName 
+&getSOA
                 &getRecLine &getDomRecs &addRec &updateRec &delRec &domStatus
                 %typemap %reverse_typemap
@@ -318 +319 @@
 
 
+## DNSDB::getChildren()
+# Get a list of all groups whose parent^n is group <n>
+# Takes a database handle, group ID, and reference to an array to put the group IDs in
+# Calls itself
+sub getChildren {
+  $errstr = '';
+  my $dbh = shift;
+  my $rootgrp = shift;
+  my $grpdest = shift;
+
+  # special break for default group;  otherwise we get stuck.
+  if ($rootgrp == 1) {
+    # by definition, group 1 is the Root Of All Groups
+    my $sth = $dbh->prepare("SELECT group_id FROM groups");
+    $sth->execute;
+    my @grplist;
+    while (my @this = $sth->fetchrow_array) {
+      push @$grpdest, @this;
+    }
+  } else {
+    my $sth = $dbh->prepare("SELECT group_id FROM groups WHERE parent_group_id=?");
+    $sth->execute($rootgrp);
+    return if $sth->rows == 0;
+    my @grplist;
+    while (my ($grp) = $sth->fetchrow_array) {
+      push @$grpdest, $grp;
+      getChildren($dbh,$grp,$grpdest);
+    }
+  }
+} # end getChildren()
+
+
 ## DNSDB::grpName()
 # Return the group name based on a group ID