1: <?php
 2: /**
 3:  * Magento
 4:  *
 5:  * NOTICE OF LICENSE
 6:  *
 7:  * This source file is subject to the Open Software License (OSL 3.0)
 8:  * that is bundled with this package in the file LICENSE.txt.
 9:  * It is also available through the world-wide-web at this URL:
10:  * http://opensource.org/licenses/osl-3.0.php
11:  * If you did not receive a copy of the license and are unable to
12:  * obtain it through the world-wide-web, please send an email
13:  * to license@magentocommerce.com so we can send you a copy immediately.
14:  *
15:  * DISCLAIMER
16:  *
17:  * Do not edit or add to this file if you wish to upgrade Magento to newer
18:  * versions in the future. If you wish to customize Magento for your
19:  * needs please refer to http://www.magentocommerce.com for more information.
20:  *
21:  * @category    Mage
22:  * @package     Mage_Dataflow
23:  * @copyright   Copyright (c) 2012 Magento Inc. (http://www.magentocommerce.com)
24:  * @license     http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
25:  */
26: 
27: 
28: /**
29:  * Dataflow Batch abstract resource model
30:  *
31:  * @category    Mage
32:  * @package     Mage_Dataflow
33:  * @author      Magento Core Team <core@magentocommerce.com>
34:  */
35: abstract class Mage_Dataflow_Model_Resource_Batch_Abstract extends Mage_Core_Model_Resource_Db_Abstract
36: {
37:     /**
38:      * Retrieve Id collection
39:      *
40:      * @param Mage_Dataflow_Model_Batch_Abstract $object
41:      * @return array
42:      */
43:     public function getIdCollection(Mage_Dataflow_Model_Batch_Abstract $object)
44:     {
45:         if (!$object->getBatchId()) {
46:             return array();
47:         }
48: 
49:         $ids = array();
50:         $select = $this->_getWriteAdapter()->select()
51:             ->from($this->getMainTable(), array($this->getIdFieldName()))
52:             ->where('batch_id = :batch_id');
53:         $ids = $this->_getWriteAdapter()->fetchCol($select, array('batch_id' => $object->getBatchId()));
54:         return $ids;
55:     }
56: 
57:     /**
58:      * Delete current Batch collection
59:      *
60:      * @param Mage_Dataflow_Model_Batch_Abstract $object
61:      * @return Mage_Dataflow_Model_Resource_Batch_Abstract
62:      */
63:     public function deleteCollection(Mage_Dataflow_Model_Batch_Abstract $object)
64:     {
65:         if (!$object->getBatchId()) {
66:             return $this;
67:         }
68: 
69:         $this->_getWriteAdapter()->delete($this->getMainTable(), array('batch_id=?' => $object->getBatchId()));
70:         return $this;
71:     }
72: }
73: