phptsmadmin/application/classes/ormtsm.php

<?php defined('SYSPATH') or die('No direct access allowed.');

/**
 * This class extends Kohana's [ORM] class to create defaults for TSM.
 *
 * @package    PTA
 * @subpackage Core
 * @category   ORM
 * @author     Deon George
 * @copyright  (c) 2010 phpTSMadmin Development Team
 * @license    http://dev.osbill.net/license.html
 */
class ORMTSM extends ORM {
	// Suppress ORMs inclusion of <table_name>.*
	protected $_disable_wild_select = TRUE;
	// Suppress ORMs inclusion of <table_name>. to column joins
	protected $_disable_join_table_name = TRUE;
	// Suppress ORMs use of limit
	protected $_disable_limit = TRUE;
	// To enable effective caching, this must disabled.
	protected $_reload_on_wakeup = FALSE;

	// Enable the formating of columns
	protected $_object_formated = array();
	protected $_formated = FALSE;
	protected $_formats = array();

	protected $_tsm = array();

	/**
	 *
	 * Configuration to enable PLA to work with DSMADMC & DB2 connections
	 * This is required, because the schema is different between each
	 * connection?
	 */
	protected function _initialize() {
		// Set out DB connection configuration.
		$this->_db_group = Kohana::config('config.client_type');

		// Adjustments for DSMADMC or DB2 connections
		if (array_key_exists(Kohana::config('config.client_type'),$this->_tsm))
			foreach ($this->_tsm[Kohana::config('config.client_type')] as $k => $v)
				if (preg_match('/^_/',$k))
					$this->{$k} = $v;

		if ($this->_db_group == 'db2')
			$this->_disable_join_table_name = FALSE;

		parent::_initialize();
	}

	public function __get($column) {
		// Get a substited column name - need for DB2/DSMADMC schema differences
		if (isset($this->_tsm[Kohana::config('config.client_type')]['translate']) AND array_key_exists($column,$this->_tsm[Kohana::config('config.client_type')]['translate'])) {
			return is_null($c=$this->_tsm[Kohana::config('config.client_type')]['translate'][$column]) ? NULL : parent::__get($c);
}
		else
			return parent::__get($column);
	}

	public function find() {
		// Check if we can preload our data and havent already done it
		if ($time = $this->isCacheable() AND is_null(Cache::instance()->get($cache_key = 'PRELOAD:'.$this->_table_name))) {

			// Firstly set our cache, so that we dont get in a loop
			Cache::instance()->set($cache_key,TRUE,$time-1);

			// Find all records of this type
			$c = get_class($this);
			$x = new $c;

			foreach ($x->find_all() as $record) {
				// Simulate loading the record so that we can get the SQL to use as our cache key
				$y = new $c;
				$y->where($y->_primary_key,'=',(string)$record);

				// Code, as extracted from ORM to complete building the SQL
				$y->_build(Database::SELECT);
				$y->_db_builder->from($y->_table_name);
				if (! isset($y->_db_applied['order_by']) AND ! empty($y->_sorting))
					foreach ($y->_sorting as $column => $direction) {
						if (strpos($column, '.') === FALSE)
							// Sorting column for use in JOINs
							$column = ($y->_disable_join_table_name ? '' : $y->_table_name.'.').$column;

						$y->_db_builder->order_by($column, $direction);
					}

				// Set the cache key based on the database instance name and SQL
				$cache_key = 'Database::query("'.$y->_db.'", "'.(string)$y->_db_builder.'")';
				unset($y);

				// Cache the record, our subsequent find should get a cache hit now.
				Kohana::cache($cache_key, array($record->as_array()), $time);
			}

			unset($x);
		}

		// Contiue as normal
		return parent::find();
	}

	private function isCacheable() {
		$preload = array('NODES','VOLUMES');

		$config = Kohana::config('database')->{Database::$default};

		if ($config['caching'] AND isset($config['cachepreload'][$this->_table_name]) AND count($this->_db_pending) == 1 AND $this->_db_pending[0]['name'] == 'where' AND $this->_db_pending[0]['args'][0] == $this->_primary_key AND $this->_db_pending[0]['args'][1] == '=')
			return $config['cachepreload'][$this->_table_name];
		else
			return FALSE;
	}

	protected function _load_result($multiple = FALSE) {
		// We'll cache our query results
		if ($c = $this->_db->caching($this->_table_name))
			$this->_db_builder->cached($c);

		return parent::_load_result($multiple);
	}

	/**
	 * Proxy method to Database list_columns.
	 * This enables caching of the list_columns queries. Since this doesnt
	 * we hard code the cache to 7 days.
	 *
	 * @return array
	 */
	public function list_columns() { 
		// We'll cache our query results
		if ($this->_db->caching('SCHEMA')) {
			// Set the cache key based on the database instance name and SQL
			$cache_key = 'Database::query(LC:'.$this->_table_name.')';
 
			if (! is_null($result = Cache::instance()->get($cache_key)))
				// Return a cached result
				return $result;
		}

		// Proxy to database
		$result = $this->_db->list_columns($this->_table_name);

		// Cache the result array
		if (isset($cache_key))
			Cache::instance()->set($cache_key, $result, 604800);

		return $result;
	}

	// Load our values into the ORM object
	public function load_object(array $values) {
		return parent::_load_values($values);
	}

	public static function date($date,$format) {
		return $date ? date($format,strtotime($date)) : '';
	}
}
?>