Added constructor and associated support function to Zvol class.

[omvzfs.git] / src / Zpool.php

diff --git a/src/Zpool.php b/src/Zpool.php
index a2fee739939e34a540a5c2c7bbd753fb69d7092a..61a31d293b33b52abfaacc99440c5a451b098400 100644 (file)
--- a/src/Zpool.php
+++ b/src/Zpool.php
@@ -1,208 +1,289 @@
 <?php
 <?php

-require_once 'Vdev.php';
-require_once 'Snapshot.php';
-require_once 'Dataset.php';
-require_once 'Zvol.php';
+require_once("Vdev.php");
+require_once("Snapshot.php");
+require_once("Dataset.php");
+require_once("Zvol.php");
+require_once("Exception.php");

 
 /**
 
 /**

- * XXX detailed description
+ * Class containing information about the pool

  *
  *

- * @author    XXX
- * @version   XXX
- * @copyright XXX
+ * @author    Michael Rasmussen
+ * @version   0.1
+ * @copyright Michael Rasmussen <mir@datanom.net>

  */
  */

-class Zpool {
+class OMVModuleZFSZpool extends OMVModuleAbstract
+               implements OMVNotifyListener {

     // Attributes
     /**
     // Attributes
     /**

-     * XXX
+     * Name of pool

      *
      * @var    string $name
      * @access private
      */
      *
      * @var    string $name
      * @access private
      */

-    private $_name;
+    private $name;

 
     /**
 
     /**

-     * XXX
+     * List of Vdev

      *
      *

-     * @var    list<Vdev> $vdevs
+     * @var    array $vdevs

      * @access private
      * @access private

+     * @accociation OMVModuleZFSVdev to vdevs

      */
      */

-    private $_vdevs;
+    private $vdevs;

 
     /**
 
     /**

-     * XXX
+     * List of spares

      *
      *

-     * @var    list<Disk> $spare
+     * @var    array $spare

      * @access private
      * @access private

+     * @accociation OMVModuleZFSVdev to spare

      */
      */

-    private $_spare;
+    private $spare;

 
     /**
 
     /**

-     * XXX
+     * List of log

      *
      *

-     * @var    Log $log
+     * @var    array $log

      * @access private
      * @access private

+     * @accociation OMVModuleZFSVdev to log

      */
      */

-    private $_log;
+    private $log;

 
     /**
 
     /**

-     * XXX
+     * List of cache

      *
      *

-     * @var    Cache $cache
+     * @var    array $cache

      * @access private
      * @access private

+     * @accociation OMVModuleZFSVdev to cache

      */
      */

-    private $_cache;
+    private $cache;

 
     /**
 
     /**

-     * XXX
+     * Pool size

      *
      * @var    int $size
      * @access private
      */
      *
      * @var    int $size
      * @access private
      */

-    private $_size;
+    private $size;

 
     /**
 
     /**

-     * XXX
+     * Pool's mountpoint

      *
      * @var    string $mountPoint
      * @access private
      */
      *
      * @var    string $mountPoint
      * @access private
      */

-    private $_mountPoint;
+    private $mountPoint;

 
     /**
 
     /**

-     * XXX
+     * List of features

      *
      *

-     * @var    list<Feature> $features
+     * @var    array $features

      * @access private
      */
      * @access private
      */

-    private $_features;
+    private $features;

 
     // Associations
     /**
 
     // Associations
     /**

-     * XXX
-     *
-     * @var    Snapshot $unnamed
-     * @access private
-     * @accociation Snapshot to unnamed
-     */
-    #var $unnamed;
-
-    /**
-     * XXX
+     * Array of OMVModuleZFSSnapshot.

      *
      *

-     * @var    Dataset $unnamed
+     * @var    array $snapshot

      * @access private
      * @access private

-     * @accociation Dataset to unnamed
+     * @accociation OMVModuleZFSSnapshot to snapshot

      */
      */

-    #var $unnamed;
+    private $snapshot;

 
     /**
 
     /**

-     * XXX
+     * Array of OMVModuleZFSDataset

      *
      *

-     * @var    Zvol $unnamed
+     * @var    Dataset $dataset

      * @access private
      * @access private

-     * @accociation Zvol to unnamed
+     * @accociation OMVModuleZFSDataset to dataset

      */
      */

-    #var $unnamed;
+    private $dataset;

 
     /**
 
     /**

-     * XXX
+     * Array of OMVModuleZFSZvol

      *
      *

-     * @var    Vdev $unnamed
+     * @var    Zvol $zvol

      * @access private
      * @access private

-     * @accociation Vdev to unnamed
+     * @accociation OMVModuleZFSZvol to zvol

      */
      */

-    #var $unnamed;
+    private $zvol;

 
     // Operations
 
     // Operations

+       /**
+        * Constructor
+        *
+        * @param $pool pool this mirror belongs to
+     * @throws OMVModuleZFSException
+        */
+
+       public function __construct($vdev) {
+               if (is_array($vdev)) {
+                       $cmd = $this->getCommandString($vdev);
+                       $name = $vdev[0]->getPool();
+                       $type = $vdev[0]->getType();
+               }
+               else {
+                       $cmd = $this->getCommandString(array($vdev));
+                       $name = $vdev->getPool();
+                       $type = $vdev->getType();
+               }
+               $cmd = "zpool create $name $cmd";
+
+               OMVUtil::exec($cmd, $output, $result);
+               if ($result)
+                       throw new OMVModuleZFSException($output);
+               else {
+                       $this->vdevs = array();
+                       $this->spare = array();
+                       $this->log = array();
+                       $this->cache = array();
+                       $this->features = array();
+                       $this->name = $name;
+                       $this->type = $type;
+                       if (is_array($vdev))
+                               $this->vdevs = $vdev;
+                       else
+                               array_push ($this->vdevs, $vdev);
+                       $this->size = $this->getAttribute("size");
+                       $this->mountPoint = $this->getAttribute("mountpoint");
+               }
+       }
+

     /**
     /**

-     * XXX
+     * Get pool name

      *
      *

-     * @return string XXX
+     * @return string

      * @access public
      */
     public function getName() {
      * @access public
      */
     public function getName() {

-        trigger_error('Not Implemented!', E_USER_WARNING);
+        return $this->name;

     }
 
     /**
     }
 
     /**

-     * XXX
+     * Get array of Vdev

      *
      *

-     * @return list<Vdev> XXX
+     * @return array

      * @access public
      */
     public function getVdevs() {
      * @access public
      */
     public function getVdevs() {

-        trigger_error('Not Implemented!', E_USER_WARNING);
+        return $this->vdevs;

     }
 
     /**
     }
 
     /**

-     * XXX
+     * Add Vdev to pool

      *
      *

-     * @param  Vdev $vdev XXX
-     * @return void XXX
+     * @param  array $vdev array of OMVModuleZFSVdev
+     * @return void
+     * @throws OMVModuleZFSException

      * @access public
      */
      * @access public
      */

-    public function addVdev($vdev) {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+    public function addVdev(array $vdevs) {
+               $cmd = "zpool add " . $this->getName() . " " . $this->getCommandString($vdevs);
+               OMVUtil::exec($cmd, $output, $result);
+               if ($result)
+                       throw new OMVModuleZFSException($output);
+               else
+                       $this->vdevs = array_merge($this->vdevs, $vdevs);

     }
 
     /**
      * XXX
      *
     }
 
     /**
      * XXX
      *

-     * @param  Vdev $vdev XXX
-     * @return void XXX
+     * @param  OMVModuleZFSVdev $vdev
+     * @return void
+     * @throws OMVModuleZFSException

      * @access public
      */
      * @access public
      */

-    public function removeVdev($vdev) {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+    public function removeVdev(OMVModuleZFSVdev $vdev) {
+        throw new OMVModuleZFSException("Cannot remove vdevs from a pool");

     }
 
     /**
      * XXX
      *
     }
 
     /**
      * XXX
      *

-     * @param  Cache $cache XXX
-     * @return void XXX
+     * @param  OMVModuleZFSVdev $cache
+     * @return void
+     * @throws OMVModuleZFSException

      * @access public
      */
      * @access public
      */

-    public function addCache($cache) {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+    public function addCache(OMVModuleZFSVdev $cache) {
+        if ($cache->getType() != OMVModuleZFSVdevType::OMVMODULEZFSPLAIN)
+                       throw new OMVModuleZFSException("Only a plain Vdev can be added as cache");
+
+               $cmd = "zpool add " . $this->getName() . " cache " . $this->getCommandString($vdevs);
+               OMVUtil::exec($cmd, $output, $result);
+               if ($result)
+                       throw new OMVModuleZFSException($output);
+
+               $disks = $cache->getDisks();
+               foreach ($disks as $disk) {
+                       array_push ($this->cache, $disk);
+               }

     }
 
     /**
      * XXX
      *
     }
 
     /**
      * XXX
      *

-     * @return void XXX
+     * @param array $disks
+     * @return void
+     * @throws OMVModuleZFSException

      * @access public
      */
      * @access public
      */

-    public function removeCache() {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+    public function removeCache(array $disks = null) {
+               $errors = array();
+               $exception = null;
+
+        if (! $disks)
+                       $disks = $this->cache;
+
+               foreach ($disks as $disk) {
+                       $cmd = "zpool remove " . $this->getName() . " $disk";
+                       OMVUtil::exec($cmd, $output, $result);
+                       if ($result)
+                               array_push ($errors, $output);
+                       else
+                               $this->cache = $this->removeDisk($this->cache, $disk);
+               }
+
+               foreach ($errors as $error) {
+                       if ($exception)
+                               $exception .= "\n$error";
+                       else
+                               $exception = $error;
+               }
+
+               if ($exception)
+                       throw new OMVModuleZFSException($exception);

     }
 
     /**
      * XXX
      *
     }
 
     /**
      * XXX
      *

-     * @return Cache XXX
+     * @return Cache

      * @access public
      */
     public function getCache() {
      * @access public
      */
     public function getCache() {

-        trigger_error('Not Implemented!', E_USER_WARNING);
+        return $this->cache;

     }
 
     /**
      * XXX
      *
     }
 
     /**
      * XXX
      *

-     * @param  Log $log XXX
-     * @return void XXX
+     * @param  OMVModuleZFSVdev $log
+     * @return void

      * @access public
      */
      * @access public
      */

-    public function addLog($log) {
+    public function addLog(OMVModuleZFSVdev $log) {

         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
     /**
      * XXX
      *
         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
     /**
      * XXX
      *

-     * @return void XXX
+     * @return void

      * @access public
      */
     public function removeLog() {
      * @access public
      */
     public function removeLog() {


@@ -212,7 +293,7 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return Log XXX
+     * @return Log

      * @access public
      */
     public function getLog() {
      * @access public
      */
     public function getLog() {


@@ -222,19 +303,19 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @param  Disk $spare XXX
-     * @return void XXX
+     * @param  array $spares
+     * @return void

      * @access public
      */
      * @access public
      */

-    public function addSpare($spare) {
+    public function addSpare(array $spares) {

         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
     /**
      * XXX
      *
         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
     /**
      * XXX
      *

-     * @param  Disk $spare XXX
-     * @return void XXX
+     * @param  Disk $spare
+     * @return void

      * @access public
      */
     public function removeSpare($spare) {
      * @access public
      */
     public function removeSpare($spare) {


@@ -244,7 +325,7 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return list<Disk> XXX
+     * @return list<Disk>

      * @access public
      */
     public function getSpares() {
      * @access public
      */
     public function getSpares() {


@@ -254,7 +335,7 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return int XXX
+     * @return int

      * @access public
      */
     public function getSize() {
      * @access public
      */
     public function getSize() {


@@ -264,7 +345,7 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return string XXX
+     * @return string

      * @access public
      */
     public function getMountPoint() {
      * @access public
      */
     public function getMountPoint() {


@@ -274,18 +355,18 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @param  list<Feature> $features XXX
-     * @return void XXX
+     * @param  array $features
+     * @return void

      * @access public
      */
      * @access public
      */

-    public function setFeatures($features) {
+    public function setFeatures(array $features) {

         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
     /**
      * XXX
      *
         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
     /**
      * XXX
      *

-     * @return list<Feature> XXX
+     * @return list<Feature>

      * @access public
      */
     public function getFeatures() {
      * @access public
      */
     public function getFeatures() {


@@ -295,7 +376,7 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return void XXX
+     * @return void

      * @access public
      */
     public function export() {
      * @access public
      */
     public function export() {


@@ -305,8 +386,8 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @param  string $name XXX
-     * @return void XXX
+     * @param  string $name
+     * @return void

      * @access public
      */
     public function import($name) {
      * @access public
      */
     public function import($name) {


@@ -316,7 +397,7 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return void XXX
+     * @return void

      * @access public
      */
     public function scrub() {
      * @access public
      */
     public function scrub() {


@@ -326,13 +407,28 @@ class Zpool {
     /**
      * XXX
      *
     /**
      * XXX
      *

-     * @return string XXX
+     * @return string

      * @access public
      */
     public function status() {
         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 
      * @access public
      */
     public function status() {
         trigger_error('Not Implemented!', E_USER_WARNING);
     }
 

+    public function bindListeners(OMVNotifyDispatcher $dispatcher) {
+        $dispatcher->addListener(
+          OMV_NOTIFY_EVENT,
+          "org.openmediavault.module.service.nfs.start",
+          array($this, "onNotify"));
+        $dispatcher->addListener(
+          OMV_NOTIFY_EVENT,
+          "org.openmediavault.module.service.nfs.stop",
+          array($this, "onNotify"));
+        $dispatcher->addListener(
+          OMV_NOTIFY_EVENT,
+          "org.openmediavault.module.service.nfs.applyconfig",
+          array($this, "onNotify"));
+    }
+

        /**
         * XXX
         * org.openmediavault.module.service.<servicename>.start
        /**
         * XXX
         * org.openmediavault.module.service.<servicename>.start


@@ -347,30 +443,75 @@ class Zpool {
        }
 
        /**
        }
 
        /**

-        * XXX
+        * Convert array of Vdev to command string

         *
         *

-        * @access public
+        * @param array $vdevs
+        * @return string
+        * @throws OMVMODULEZFSException

         */
         */

-       public function applyConfig() {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+       private function getCommandString(array $vdevs) {
+               $adds = array();
+
+               foreach ($vdevs as $vdev) {
+                       $type = $vdev->getType();
+                       $command = "";
+
+                       switch ($type) {
+                               case OMVModuleZFSVdevType::OMVMODULEZFSPLAIN: break;
+                               case OMVModuleZFSVdevType::OMVMODULEZFSMIRROR: $command = "mirror"; break;
+                               case OMVModuleZFSVdevType::OMVMODULEZFSRAIDZ1: $command = "raidz1"; break;
+                               case OMVModuleZFSVdevType::OMVMODULEZFSRAIDZ2: $command = "raidz2"; break;
+                               case OMVModuleZFSVdevType::OMVMODULEZFSRAIDZ3: $command = "raidz3"; break;
+                               default:
+                                       throw new OMVMODULEZFSException("Unknown Vdev type");
+                       }
+                       $disks = $vdev->getDisks();
+                       $diskStr = "";
+                       foreach($disks as $disk) {
+                               $diskStr .= " $disk";
+                       }
+
+                       array_push ($adds, $command . $diskStr);
+               }
+
+               return join(" ", $adds);

        }
 
        /**
        }
 
        /**

-        * XXX
+        * Get an attribute from pool

         *
         *

-        * @access public
+        * @param string $attribute
+        * @return string value

         */
         */

-       public function stopService() {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+       private function getAttribute($attribute) {
+               $cmd = "zpool list -H -o $attribute {$this->name}";
+               OMVUtil::exec($cmd, $output, $result);
+               if ($result) {
+                       $cmd = "zfs list -H -o $attribute {$this->name}";
+                       OMVUtil::exec($cmd, $output, $result);
+                       if ($result)
+                               return null;
+               }
+
+               return $output;

        }
 
        /**
        }
 
        /**

-        * XXX
+        * Remove a disk from array

         *
         *

-        * @access public
+        * @param array $array
+        * @param string $disk
+        * @return array

         */
         */

-       public function startService() {
-        trigger_error('Not Implemented!', E_USER_WARNING);
+       private function removeDisk(array $array, $disk) {
+               $new_disks = array();
+
+               foreach ($array as $item) {
+                       if (strcmp($item, $disk) != 0)
+                               array_push ($new_disks, $item);
+               }
+
+               return $new_disks;

        }
 }
 
        }
 }