Merge branch 'issue45' of https://github.com/johnbellone/CodeIgniter …

…into johnbellone-issue45
darkhouse · Aug 21, 2011 · 3932f3d · 3932f3d
2 parents 27a883c + 51e4bca
commit 3932f3d
Showing 1 changed file with 39 additions and 9 deletions.
diff --git a/system/libraries/Cache/drivers/Cache_memcached.php b/system/libraries/Cache/drivers/Cache_memcached.php
@@ -152,8 +152,21 @@ private function _setup_memcached()
 				}				
 			}			
 		}
-
-		$this->_memcached = new Memcached();
+
+                if (class_exists('Memcached')) 
+                {
+                        $this->_memcached = new Memcached();
+                }
+                else if (class_exists('Memcache')) 
+                {
+                        $this->_memcached = new Memcache();
+                }
+                else 
+                {
+                        log_message('error', 'Failed to create object for Memcached Cache; extension not loaded?');
+
+                        return FALSE;        
+                }
 
 		foreach ($this->_memcache_conf as $name => $cache_server)
 		{
@@ -172,10 +185,27 @@ private function _setup_memcached()
 				$cache_server['weight'] = $this->_default_options['default_weight'];
 			}
 
-			$this->_memcached->addServer(
-					$cache_server['hostname'], $cache_server['port'], $cache_server['weight']
-			);
+                        if (get_class($this->_memcached) == 'Memcache') 
+                        {
+                                // Third parameter is persistance and defaults to TRUE.
+                                $this->_memcached->addServer(
+                                        $cache_server['hostname'], 
+                                        $cache_server['port'], 
+                                        TRUE, 
+                                        $cache_server['weight'] 
+                                );
+                        }
+                        else 
+                        {
+                                $this->_memcached->addServer(
+                                        $cache_server['hostname'], 
+                                        $cache_server['port'], 
+                                        $cache_server['weight'] 
+                                );
+                        }
 		}
+
+                return TRUE;
 	}
 
 	// ------------------------------------------------------------------------
@@ -189,15 +219,15 @@ private function _setup_memcached()
 	 */
 	public function is_supported()
 	{
-		if ( ! extension_loaded('memcached'))
+                if ( ! extension_loaded('memcached') && ! extension_loaded('memcache'))
 		{
 			log_message('error', 'The Memcached Extension must be loaded to use Memcached Cache.');
 
 			return FALSE;
 		}
-		
-		$this->_setup_memcached();
-		return TRUE;
+
+                
+		return $this->_setup_memcached();
 	}
 
 	// ------------------------------------------------------------------------