codeigniter使用mongodb/redis
ci2.x版本，使用mongodb，需要安装pecl-php-mongo扩展（github上很多扩展已不可用，找到个可用版本纪录于此），添加到php.ini中
使用如下
public function index() { //var_dump($this->mongo_db->insert("task",array("uid"=>2,"name"=>"ciaos"))); //var_dump($this->mongo_db->get_where("task",array("uid"=>"ciaos"))); var_dump($this->mongo_db->where(array("uid"=>"ciaos"))->update("task",array("task.math"=>array(1,2)))); }
autoload.php增加redis,mongo_db自动加载
mongodb.php放到config目录下
<?php if ( ! defined('BASEPATH')) exit('No direct script access allowed'); // Generally will be localhost if you're querying from the machine that Mongo is installed on $config['mongo_host'] = "localhost"; // Generally will be 27017 unless you've configured Mongo otherwise $config['mongo_port'] = 27017; // The database you want to work from (required) $config['mongo_db'] = "dbname"; // Leave blank if Mongo is not running in auth mode $config['mongo_user'] = ""; $config['mongo_pass'] = ""; // Persistant connections $config['mongo_persist'] = TRUE; $config['mongo_persist_key'] = 'ci_mongo_persist'; // Get results as an object instead of an array $config['mongo_return'] = 'array'; // Set to object // When you run an insert/update/delete how sure do you want to be that the database has received the query? // safe = the database has receieved and executed the query // fysnc = as above + the change has been committed to harddisk <- NOTE: will introduce a performance penalty $config['mongo_query_safety'] = 'safe'; // Supress connection error password display $config['mongo_supress_connect_error'] = TRUE; // If you are having problems connecting try changing this to TRUE $config['host_db_flag'] = FALSE;
Mongo_db.php放到library下
<?php if ( ! defined('BASEPATH')) exit('No direct script access allowed'); /** * CodeIgniter MongoDB Active Record Library * * A library to interface with the NoSQL database MongoDB. For more information see http://www.mongodb.org * Originally created by Alex Bilbie, but extended and updated by Kyle J. Dye * * @package CodeIgniter * @author Alex Bilbie | www.alexbilbie.com | alex@alexbilbie.com * @author Kyle J. Dye | www.kyledye.com | kyle@kyledye.com * @copyright Copyright (c) 2010, Alex Bilbie. * @license http://codeigniter.com/user_guide/license.html * @link http://alexbilbie.com * @link http://kyledye.com * @version Version 0.3.7 */ class Mongo_db { private $CI; private $config_file = 'mongodb'; private $connection; private $db; private $connection_string; private $host; private $port; private $user; private $pass; private $dbname; private $persist; private $persist_key; private $query_safety = 'safe'; private $selects = array(); public $wheres = array(); // Public to make debugging easier private $sorts = array(); private $limit = 999999; private $offset = 0; /** * -------------------------------------------------------------------------------- * CONSTRUCTOR * -------------------------------------------------------------------------------- * * Automatically check if the Mongo PECL extension has been installed/enabled. * Generate the connection string and establish a connection to the MongoDB. */ public function __construct() { if ( ! class_exists('Mongo')) { show_error("The MongoDB PECL extension has not been installed or enabled", 500); } $this->CI =& get_instance(); $this->connection_string(); $this->connect(); } /** * -------------------------------------------------------------------------------- * Switch_db * -------------------------------------------------------------------------------- * * Switch from default database to a different db */ public function switch_db($database = '') { if (empty($database)) { show_error("To switch MongoDB databases, a new database name must be specified", 500); } $this->dbname = $database; try { $this->db = $this->connection->{$this->dbname}; return (TRUE); } catch (Exception $e) { show_error("Unable to switch Mongo Databases: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * Drop_db * -------------------------------------------------------------------------------- * * Drop a Mongo database * @usage: $this->mongo_db->drop_db("foobar"); */ public function drop_db($database = '') { if (empty($database)) { show_error('Failed to drop MongoDB database because name is empty', 500); } else { try { $this->connection->{$database}->drop(); return (TRUE); } catch (Exception $e) { show_error("Unable to drop Mongo database `{$database}`: {$e->getMessage()}", 500); } } } /** * -------------------------------------------------------------------------------- * Drop_collection * -------------------------------------------------------------------------------- * * Drop a Mongo collection * @usage: $this->mongo_db->drop_collection('foo', 'bar'); */ public function drop_collection($db = "", $col = "") { if (empty($db)) { show_error('Failed to drop MongoDB collection because database name is empty', 500); } if (empty($col)) { show_error('Failed to drop MongoDB collection because collection name is empty', 500); } else { try { $this->connection->{$db}->{$col}->drop(); return TRUE; } catch (Exception $e) { show_error("Unable to drop Mongo collection `{$col}`: {$e->getMessage()}", 500); } } return($this); } /** * -------------------------------------------------------------------------------- * SELECT FIELDS * -------------------------------------------------------------------------------- * * Determine which fields to include OR which to exclude during the query process. * Currently, including and excluding at the same time is not available, so the * $includes array will take precedence over the $excludes array. If you want to * only choose fields to exclude, leave $includes an empty array(). * * @usage: $this->mongo_db->select(array('foo', 'bar'))->get('foobar'); */ public function select($includes = array(), $excludes = array()) { if ( ! is_array($includes)) { $includes = array(); } if ( ! is_array($excludes)) { $excludes = array(); } if ( ! empty($includes)) { foreach ($includes as $col) { $this->selects[$col] = 1; } } else { foreach ($excludes as $col) { $this->selects[$col] = 0; } } return ($this); } /** * -------------------------------------------------------------------------------- * WHERE PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents based on these search parameters. The $wheres array should * be an associative array with the field as the key and the value as the search * criteria. * * @usage : $this->mongo_db->where(array('foo' => 'bar'))->get('foobar'); */ public function where($wheres = array()) { foreach ($wheres as $wh => $val) { $this->wheres[$wh] = $val; } return ($this); } /** * -------------------------------------------------------------------------------- * OR_WHERE PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field may be something else * * @usage : $this->mongo_db->or_where(array( array('foo'=>'bar', 'bar'=>'foo' ))->get('foobar'); */ public function or_where($wheres = array()) { if (count($wheres) > 0) { if ( ! isset($this->wheres['$or']) || ! is_array($this->wheres['$or'])) { $this->wheres['$or'] = array(); } foreach ($wheres as $wh => $val) { $this->wheres['$or'][] = array($wh=>$val); } } return ($this); } /** * -------------------------------------------------------------------------------- * WHERE_IN PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is in a given $in array(). * * @usage : $this->mongo_db->where_in('foo', array('bar', 'zoo', 'blah'))->get('foobar'); */ public function where_in($field = "", $in = array()) { $this->_where_init($field); $this->wheres[$field]['$in'] = $in; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE_IN_ALL PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is in all of a given $in array(). * * @usage : $this->mongo_db->where_in('foo', array('bar', 'zoo', 'blah'))->get('foobar'); */ public function where_in_all($field = "", $in = array()) { $this->_where_init($field); $this->wheres[$field]['$all'] = $in; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE_NOT_IN PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is not in a given $in array(). * * @usage : $this->mongo_db->where_not_in('foo', array('bar', 'zoo', 'blah'))->get('foobar'); */ public function where_not_in($field = "", $in = array()) { $this->_where_init($field); $this->wheres[$field]['$nin'] = $in; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE GREATER THAN PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is greater than $x * * @usage : $this->mongo_db->where_gt('foo', 20); */ public function where_gt($field = "", $x) { $this->_where_init($field); $this->wheres[$field]['$gt'] = $x; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE GREATER THAN OR EQUAL TO PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is greater than or equal to $x * * @usage : $this->mongo_db->where_gte('foo', 20); */ public function where_gte($field = "", $x) { $this->_where_init($field); $this->wheres[$field]['$gte'] = $x; return($this); } /** * -------------------------------------------------------------------------------- * WHERE LESS THAN PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is less than $x * * @usage : $this->mongo_db->where_lt('foo', 20); */ public function where_lt($field = "", $x) { $this->_where_init($field); $this->wheres[$field]['$lt'] = $x; return($this); } /** * -------------------------------------------------------------------------------- * WHERE LESS THAN OR EQUAL TO PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is less than or equal to $x * * @usage : $this->mongo_db->where_lte('foo', 20); */ public function where_lte($field = "", $x) { $this->_where_init($field); $this->wheres[$field]['$lte'] = $x; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE BETWEEN PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is between $x and $y * * @usage : $this->mongo_db->where_between('foo', 20, 30); */ public function where_between($field = "", $x, $y) { $this->_where_init($field); $this->wheres[$field]['$gte'] = $x; $this->wheres[$field]['$lte'] = $y; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE BETWEEN AND NOT EQUAL TO PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is between but not equal to $x and $y * * @usage : $this->mongo_db->where_between_ne('foo', 20, 30); */ public function where_between_ne($field = "", $x, $y) { $this->_where_init($field); $this->wheres[$field]['$gt'] = $x; $this->wheres[$field]['$lt'] = $y; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE NOT EQUAL TO PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the value of a $field is not equal to $x * * @usage : $this->mongo_db->where_not_equal('foo', 1)->get('foobar'); */ public function where_ne($field = '', $x) { $this->_where_init($field); $this->wheres[$field]['$ne'] = $x; return ($this); } /** * -------------------------------------------------------------------------------- * WHERE NOT EQUAL TO PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents nearest to an array of coordinates (your collection must have a geospatial index) * * @usage : $this->mongo_db->where_near('foo', array('50','50'))->get('foobar'); */ function where_near($field = '', $co = array()) { $this->__where_init($field); $this->where[$what]['$near'] = $co; return ($this); } /** * -------------------------------------------------------------------------------- * LIKE PARAMETERS * -------------------------------------------------------------------------------- * * Get the documents where the (string) value of a $field is like a value. The defaults * allow for a case-insensitive search. * * @param $flags * Allows for the typical regular expression flags: * i = case insensitive * m = multiline * x = can contain comments * l = locale * s = dotall, "." matches everything, including newlines * u = match unicode * * @param $enable_start_wildcard * If set to anything other than TRUE, a starting line character "^" will be prepended * to the search value, representing only searching for a value at the start of * a new line. * * @param $enable_end_wildcard * If set to anything other than TRUE, an ending line character "$" will be appended * to the search value, representing only searching for a value at the end of * a line. * * @usage : $this->mongo_db->like('foo', 'bar', 'im', FALSE, TRUE); */ public function like($field = "", $value = "", $flags = "i", $enable_start_wildcard = TRUE, $enable_end_wildcard = TRUE) { $field = (string) trim($field); $this->where_init($field); $value = (string) trim($value); $value = quotemeta($value); if ($enable_start_wildcard !== TRUE) { $value = "^" . $value; } if ($enable_end_wildcard !== TRUE) { $value .= "$"; } $regex = "/$value/$flags"; $this->wheres[$field] = new MongoRegex($regex); return ($this); } /** * -------------------------------------------------------------------------------- * ORDER BY PARAMETERS * -------------------------------------------------------------------------------- * * Sort the documents based on the parameters passed. To set values to descending order, * you must pass values of either -1, FALSE, 'desc', or 'DESC', else they will be * set to 1 (ASC). * * @usage : $this->mongo_db->where_between('foo', 20, 30); */ public function order_by($fields = array()) { foreach ($fields as $col => $val) { if ($val == -1 || $val === FALSE || strtolower($val) == 'desc') { $this->sorts[$col] = -1; } else { $this->sorts[$col] = 1; } } return ($this); } /** * -------------------------------------------------------------------------------- * LIMIT DOCUMENTS * -------------------------------------------------------------------------------- * * Limit the result set to $x number of documents * * @usage : $this->mongo_db->limit($x); */ public function limit($x = 99999) { if ($x !== NULL && is_numeric($x) && $x >= 1) { $this->limit = (int) $x; } return ($this); } /** * -------------------------------------------------------------------------------- * OFFSET DOCUMENTS * -------------------------------------------------------------------------------- * * Offset the result set to skip $x number of documents * * @usage : $this->mongo_db->offset($x); */ public function offset($x = 0) { if ($x !== NULL && is_numeric($x) && $x >= 1) { $this->offset = (int) $x; } return ($this); } /** * -------------------------------------------------------------------------------- * GET_WHERE * -------------------------------------------------------------------------------- * * Get the documents based upon the passed parameters * * @usage : $this->mongo_db->get_where('foo', array('bar' => 'something')); */ public function get_where($collection = "", $where = array()) { return ($this->where($where)->get($collection)); } /** * -------------------------------------------------------------------------------- * GET * -------------------------------------------------------------------------------- * * Get the documents based upon the passed parameters * * @usage : $this->mongo_db->get('foo', array('bar' => 'something')); */ public function get($collection = "") { if (empty($collection)) { show_error("In order to retreive documents from MongoDB, a collection name must be passed", 500); } $results = array(); $documents = $this->db->{$collection}->find($this->wheres, $this->selects)->limit((int) $this->limit)->skip((int) $this->offset)->sort($this->sorts); // Clear $this->_clear(); $returns = array(); foreach ($documents as $doc) { $returns[] = $doc; } if ($this->CI->config->item('mongo_return') == 'object') { return (object)$returns; } else { return $returns; } } /** * -------------------------------------------------------------------------------- * COUNT * -------------------------------------------------------------------------------- * * Count the documents based upon the passed parameters * * @usage : $this->mongo_db->get('foo'); */ public function count($collection = "") { if (empty($collection)) { show_error("In order to retreive a count of documents from MongoDB, a collection name must be passed", 500); } $count = $this->db->{$collection}->find($this->wheres)->limit((int) $this->limit)->skip((int) $this->offset)->count(); $this->_clear(); return ($count); } /** * -------------------------------------------------------------------------------- * INSERT * -------------------------------------------------------------------------------- * * Insert a new document into the passed collection * * @usage : $this->mongo_db->insert('foo', $data = array()); */ public function insert($collection = "", $insert = array()) { if (empty($collection)) { show_error("No Mongo collection selected to insert into", 500); } if (count($insert) == 0 || !is_array($insert)) { show_error("Nothing to insert into Mongo collection or insert is not an array", 500); } try { $this->db->{$collection}->insert($insert, array( "w" => 1)); if (isset($insert['_id'])) { return ($insert['_id']); } else { return (FALSE); } } catch (MongoCursorException $e) { show_error("Insert of data into MongoDB failed: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * UPDATE * -------------------------------------------------------------------------------- * * Updates a single document * * @usage: $this->mongo_db->update('foo', $data = array()); */ public function update($collection = "", $data = array(), $options = array()) { if (empty($collection)) { show_error("No Mongo collection selected to update", 500); } if (count($data) == 0 || ! is_array($data)) { show_error("Nothing to update in Mongo collection or update is not an array", 500); } try { $options = array_merge($options, array("w" => 1, 'multiple' => FALSE)); $this->db->{$collection}->update($this->wheres, array('$set' => $data), $options); $this->_clear(); return (TRUE); } catch (MongoCursorException $e) { show_error("Update of data into MongoDB failed: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * UPDATE_ALL * -------------------------------------------------------------------------------- * * Updates a collection of documents * * @usage: $this->mongo_db->update_all('foo', $data = array()); */ public function update_all($collection = "", $data = array()) { if (empty($collection)) { show_error("No Mongo collection selected to update", 500); } if (count($data) == 0 || ! is_array($data)) { show_error("Nothing to update in Mongo collection or update is not an array", 500); } try { $this->db->{$collection}->update($this->wheres, array('$set' => $data), array($this->query_safety => TRUE, 'multiple' => TRUE)); $this->_clear(); return (TRUE); } catch (MongoCursorException $e) { show_error("Update of data into MongoDB failed: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * DELETE * -------------------------------------------------------------------------------- * * delete document from the passed collection based upon certain criteria * * @usage : $this->mongo_db->delete('foo', $data = array()); */ public function delete($collection = "") { if (empty($collection)) { show_error("No Mongo collection selected to delete from", 500); } try { $this->db->{$collection}->remove($this->wheres, array($this->query_safety => TRUE, 'justOne' => TRUE)); $this->_clear(); return (TRUE); } catch (MongoCursorException $e) { show_error("Delete of data into MongoDB failed: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * DELETE_ALL * -------------------------------------------------------------------------------- * * Delete all documents from the passed collection based upon certain criteria * * @usage : $this->mongo_db->delete_all('foo', $data = array()); */ public function delete_all($collection = "") { if (empty($collection)) { show_error("No Mongo collection selected to delete from", 500); } try { $this->db->{$collection}->remove($this->wheres, array($this->query_safety => TRUE, 'justOne' => FALSE)); $this->_clear(); return (TRUE); } catch (MongoCursorException $e) { show_error("Delete of data into MongoDB failed: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * COMMAND * -------------------------------------------------------------------------------- * * Runs a MongoDB command (such as GeoNear). See the MongoDB documentation for more usage scenarios: * http://dochub.mongodb.org/core/commands * * @usage : $this->mongo_db->command(array('geoNear'=>'buildings', 'near'=>array(53.228482, -0.547847), 'num' => 10, 'nearSphere'=>true)); */ public function command($query = array()) { try { $run = $this->db->command($query); return $run; } catch (MongoCursorException $e) { show_error("MongoDB command failed to execute: {$e->getMessage()}", 500); } } /** * -------------------------------------------------------------------------------- * ADD_INDEX * -------------------------------------------------------------------------------- * * Ensure an index of the keys in a collection with optional parameters. To set values to descending order, * you must pass values of either -1, FALSE, 'desc', or 'DESC', else they will be * set to 1 (ASC). * * @usage : $this->mongo_db->add_index($collection, array('first_name' => 'ASC', 'last_name' => -1), array('unique' => TRUE)); */ public function add_index($collection = "", $keys = array(), $options = array()) { if (empty($collection)) { show_error("No Mongo collection specified to add index to", 500); } if (empty($keys) || ! is_array($keys)) { show_error("Index could not be created to MongoDB Collection because no keys were specified", 500); } foreach ($keys as $col => $val) { if($val == -1 || $val === FALSE || strtolower($val) == 'desc') { $keys[$col] = -1; } else { $keys[$col] = 1; } } if ($this->db->{$collection}->ensureIndex($keys, $options) == TRUE) { $this->_clear(); return ($this); } else { show_error("An error occured when trying to add an index to MongoDB Collection", 500); } } /** * -------------------------------------------------------------------------------- * REMOVE_INDEX * -------------------------------------------------------------------------------- * * Remove an index of the keys in a collection. To set values to descending order, * you must pass values of either -1, FALSE, 'desc', or 'DESC', else they will be * set to 1 (ASC). * * @usage : $this->mongo_db->remove_index($collection, array('first_name' => 'ASC', 'last_name' => -1)); */ public function remove_index($collection = "", $keys = array()) { if (empty($collection)) { show_error("No Mongo collection specified to remove index from", 500); } if (empty($keys) || ! is_array($keys)) { show_error("Index could not be removed from MongoDB Collection because no keys were specified", 500); } if ($this->db->{$collection}->deleteIndex($keys, $options) == TRUE) { $this->_clear(); return ($this); } else { show_error("An error occured when trying to remove an index from MongoDB Collection", 500); } } /** * -------------------------------------------------------------------------------- * REMOVE_ALL_INDEXES * -------------------------------------------------------------------------------- * * Remove all indexes from a collection. * * @usage : $this->mongo_db->remove_all_index($collection); */ public function remove_all_indexes($collection = "") { if (empty($collection)) { show_error("No Mongo collection specified to remove all indexes from", 500); } $this->db->{$collection}->deleteIndexes(); $this->_clear(); return ($this); } /** * -------------------------------------------------------------------------------- * LIST_INDEXES * -------------------------------------------------------------------------------- * * Lists all indexes in a collection. * * @usage : $this->mongo_db->list_indexes($collection); */ public function list_indexes($collection = "") { if (empty($collection)) { show_error("No Mongo collection specified to remove all indexes from", 500); } return ($this->db->{$collection}->getIndexInfo()); } /** * -------------------------------------------------------------------------------- * CONNECT TO MONGODB * -------------------------------------------------------------------------------- * * Establish a connection to MongoDB using the connection string generated in * the connection_string() method. If 'mongo_persist_key' was set to true in the * config file, establish a persistent connection. We allow for only the 'persist' * option to be set because we want to establish a connection immediately. */ private function connect() { $options = array(); if ($this->persist === TRUE) { $options['persist'] = isset($this->persist_key) && !empty($this->persist_key) ? $this->persist_key : 'ci_mongo_persist'; } try { $this->connection = new MongoClient($this->connection_string, $options); $this->db = $this->connection->{$this->dbname}; return ($this); } catch (MongoConnectionException $e) { if($this->CI->config->item('mongo_supress_connect_error')) { show_error("Unable to connect to MongoDB", 500); } else { show_error("Unable to connect to MongoDB: {$e->getMessage()}", 500); } } } /** * -------------------------------------------------------------------------------- * BUILD CONNECTION STRING * -------------------------------------------------------------------------------- * * Build the connection string from the config file. */ private function connection_string() { $this->CI->config->load($this->config_file); $this->host = trim($this->CI->config->item('mongo_host')); $this->port = trim($this->CI->config->item('mongo_port')); $this->user = trim($this->CI->config->item('mongo_user')); $this->pass = trim($this->CI->config->item('mongo_pass')); $this->dbname = trim($this->CI->config->item('mongo_db')); $this->persist = trim($this->CI->config->item('mongo_persist')); $this->persist_key = trim($this->CI->config->item('mongo_persist_key')); $this->query_safety = trim($this->CI->config->item('mongo_query_safety')); $dbhostflag = (bool)$this->CI->config->item('host_db_flag'); $connection_string = "mongodb://"; if (empty($this->host)) { show_error("The Host must be set to connect to MongoDB", 500); } if (empty($this->dbname)) { show_error("The Database must be set to connect to MongoDB", 500); } if ( ! empty($this->user) && ! empty($this->pass)) { $connection_string .= "{$this->user}:{$this->pass}@"; } if (isset($this->port) && ! empty($this->port)) { $connection_string .= "{$this->host}:{$this->port}"; } else { $connection_string .= "{$this->host}"; } if ($dbhostflag === TRUE) { $this->connection_string = trim($connection_string) . '/' . $this->dbname; } else { $this->connection_string = trim($connection_string); } } /** * -------------------------------------------------------------------------------- * _clear * -------------------------------------------------------------------------------- * * Resets the class variables to default settings */ private function _clear() { $this->selects = array(); $this->wheres = array(); $this->limit = 999999; $this->offset = 0; $this->sorts = array(); } /** * -------------------------------------------------------------------------------- * WHERE INITIALIZER * -------------------------------------------------------------------------------- * * Prepares parameters for insertion in $wheres array(). */ private function _where_init($param) { if ( ! isset($this->wheres[$param])) { $this->wheres[ $param ] = array(); } } } // EOF
使用redis也需要安装redis扩展
redis.php放config目录
<?php if ( ! defined('BASEPATH')) exit('No direct script access allowed'); /** * Config for the CodeIgniter Redis library * * @see ../libraries/Redis.php */ // Default connection group $config['redis_default']['host'] = '127.0.0.1‘; $config['redis_default']['port'] = '6379'; // Default Redis port is 6379 $config['redis_default']['password'] = ''; // Can be left empty when the server does not require AUTH $config['redis_slave']['host'] = ''; $config['redis_slave']['port'] = '6379'; $config['redis_slave']['password'] = '';
Redis.php放library目录
<?php defined('BASEPATH') OR exit('No direct script access allowed'); /** * CodeIgniter Redis * * A CodeIgniter library to interact with Redis * * @package CodeIgniter * @category Libraries * @author Joël Cox * @version v0.4 * @link https://github.com/joelcox/codeigniter-redis * @link http://joelcox.nl * @license http://www.opensource.org/licenses/mit-license.html */ class CI_Redis { /** * CI * * CodeIgniter instance * @var object */ private $_ci; /** * Connection * * Socket handle to the Redis server * @var handle */ private $_connection; /** * Debug * * Whether we're in debug mode * @var bool */ public $debug = TRUE; /** * CRLF * * User to delimiter arguments in the Redis unified request protocol * @var string */ const CRLF = "\r\n"; /** * Constructor */ public function __construct($params = array()) { log_message('debug', 'Redis Class Initialized'); $this->_ci = get_instance(); $this->_ci->load->config('redis'); // Check for the different styles of configs if (isset($params['connection_group'])) { // Specific connection group $config = $this->_ci->config->item('redis_' . $params['connection_group']); } elseif (is_array($this->_ci->config->item('redis_default'))) { // Default connection group $config = $this->_ci->config->item('redis_default'); } else { // Original config style $config = array( 'host' => $this->_ci->config->item('redis_host'), 'port' => $this->_ci->config->item('redis_port'), 'password' => $this->_ci->config->item('redis_password'), ); } // Connect to Redis $this->_connection = @fsockopen($config['host'], $config['port'], $errno, $errstr, 3); // Display an error message if connection failed if ( ! $this->_connection) { show_error('Could not connect to Redis at ' . $config['host'] . ':' . $config['port']); } // Authenticate when needed $this->_auth($config['password']); } /** * Call * * Catches all undefined methods * @param string method that was called * @param mixed arguments that were passed * @return mixed */ public function __call($method, $arguments) { $request = $this->_encode_request($method, $arguments); return $this->_write_request($request); } /** * Command * * Generic command function, just like redis-cli * @param string full command as a string * @return mixed */ public function command($string) { $slices = explode(' ', $string); $request = $this->_encode_request($slices[0], array_slice($slices, 1)); return $this->_write_request($request); } /** * Auth * * Runs the AUTH command when password is set * @param string password for the Redis server * @return void */ private function _auth($password = NULL) { // Authenticate when password is set if ( ! empty($password)) { // See if we authenticated successfully if ($this->command('AUTH ' . $password) !== 'OK') { show_error('Could not connect to Redis, invalid password'); } } } /** * Clear Socket * * Empty the socket buffer of theconnection so data does not bleed over * to the next message. * @return NULL */ public function _clear_socket() { // Read one character at a time fflush($this->_connection); return NULL; } /** * Write request * * Write the formatted request to the socket * @param string request to be written * @return mixed */ private function _write_request($request) { if ($this->debug === TRUE) { //log_message('debug', 'Redis unified request: ' . $request); } // How long is the data we are sending? $value_length = strlen($request); // If there isn't any data, just return if ($value_length <= 0) return NULL; // Handle reply if data is less than or equal to 8192 bytes, just send it over if ($value_length <= 8192) { fwrite($this->_connection, $request); } else { while ($value_length > 0) { // If we have more than 8192, only take what we can handle if ($value_length > 8192) { $send_size = 8192; } // Send our chunk fwrite($this->_connection, $request, $send_size); // How much is left to send? $value_length = $value_length - $send_size; // Remove data sent from outgoing data $request = substr($request, $send_size, $value_length); } } // Read our request into a variable $return = $this->_read_request(); // Clear the socket so no data remains in the buffer $this->_clear_socket(); return $return; } /** * Read request * * Route each response to the appropriate interpreter * @return mixed */ private function _read_request() { $type = fgetc($this->_connection); // Times we will attempt to trash bad data in search of a // valid type indicator $response_types = array('+', '-', ':', '$', '*'); $type_error_limit = 50; $try = 0; while ( ! in_array($type, $response_types) && $try < $type_error_limit) { $type = fgetc($this->_connection); $try++; } if ($this->debug === TRUE) { log_message('debug', 'Redis response type: ' . $type); } switch ($type) { case '+': return $this->_single_line_reply(); break; case '-': return $this->_error_reply(); break; case ':': return $this->_integer_reply(); break; case '$': return $this->_bulk_reply(); break; case '*': return $this->_multi_bulk_reply(); break; default: return FALSE; } } /** * Single line reply * * Reads the reply before the EOF * @return mixed */ private function _single_line_reply() { $value = rtrim(fgets($this->_connection)); $this->_clear_socket(); return $value; } /** * Error reply * * Write error to log and return false * @return bool */ private function _error_reply() { // Extract the error message $error = substr(rtrim(fgets($this->_connection)), 4); log_message('error', 'Redis server returned an error: ' . $error); $this->_clear_socket(); return FALSE; } /** * Integer reply * * Returns an integer reply * @return int */ private function _integer_reply() { return (int) rtrim(fgets($this->_connection)); } /** * Bulk reply * * Reads to amount of bits to be read and returns value within * the pointer and the ending delimiter * @return string */ private function _bulk_reply() { // How long is the data we are reading? Support waiting for data to // fully return from redis and enter into socket. $value_length = (int) fgets($this->_connection); if ($this->debug === TRUE) { log_message('debug', 'Redis response length: ' . $value_length); } if ($value_length <= 0) return NULL; $response = ''; // Handle reply if data is less than or equal to 8192 bytes, just read it if ($value_length <= 8192) { //$response = fread($this->_connection, $value_length); //fix read_size error // Read our chunk $read_size = $value_length; $chunk = fread($this->_connection, $read_size); // Support reading very long responses that don't come through // in one fread $chunk_length = strlen($chunk); while ($chunk_length < $read_size) { $keep_reading = $read_size - $chunk_length; $chunk .= fread($this->_connection, $keep_reading); $chunk_length = strlen($chunk); } $response = $chunk; } else { $data_left = $value_length; // If the data left is greater than 0, keep reading while ($data_left > 0 ) { // If we have more than 8192, only take what we can handle if ($data_left > 8192) { $read_size = 8192; } else { $read_size = $data_left; } // Read our chunk $chunk = fread($this->_connection, $read_size); // Support reading very long responses that don't come through // in one fread $chunk_length = strlen($chunk); while ($chunk_length < $read_size) { $keep_reading = $read_size - $chunk_length; $chunk .= fread($this->_connection, $keep_reading); $chunk_length = strlen($chunk); } $response .= $chunk; // Re-calculate how much data is left to read $data_left = $data_left - $read_size; } } if ($this->debug === TRUE) { log_message('debug', 'Redis response real_length is: ' . strlen($response)); if ($value_length != strlen($response)) { log_message('debug', 'Redis response get a output-error'); } } // Clear the socket in case anything remains in there $this->_clear_socket(); return isset($response) ? $response : FALSE; } /** * Multi bulk reply * * Reads n bulk replies and return them as an array * @return array */ private function _multi_bulk_reply() { // Get the amount of values in the response $response = array(); $total_values = (int) fgets($this->_connection); // Loop all values and add them to the response array for ($i = 0; $i < $total_values; $i++) { // Remove the new line and carriage return before reading // another bulk reply fgets($this->_connection, 2); // If this is a second or later pass, we also need to get rid // of the $ indicating a new bulk reply and its length. if ($i > 0) { fgets($this->_connection); fgets($this->_connection, 2); } $response[] = $this->_bulk_reply(); } // Clear the socket $this->_clear_socket(); return isset($response) ? $response : FALSE; } /** * Encode request * * Encode plain-text request to Redis protocol format * @link http://redis.io/topics/protocol * @param string request in plain-text * @param string additional data (string or array, depending on the request) * @return string encoded according to Redis protocol */ private function _encode_request($method, $arguments = array()) { $request = '$' . strlen($method) . self::CRLF . $method . self::CRLF; $_args = 1; // Append all the arguments in the request string foreach ($arguments as $argument) { if (is_array($argument)) { foreach ($argument as $key => $value) { // Prepend the key if we're dealing with a hash if (!is_int($key)) { $request .= '$' . strlen($key) . self::CRLF . $key . self::CRLF; $_args++; } $request .= '$' . strlen($value) . self::CRLF . $value . self::CRLF; $_args++; } } else { $request .= '$' . strlen($argument) . self::CRLF . $argument . self::CRLF; $_args++; } } $request = '*' . $_args . self::CRLF . $request; return $request; } /** * Info * * Overrides the default Redis response, so we can return a nice array * of the server info instead of a nasty string. * @return array */ public function info($section = FALSE) { if ($section !== FALSE) { $response = $this->command('INFO '. $section); } else { $response = $this->command('INFO'); } $data = array(); $lines = explode(self::CRLF, $response); // Extract the key and value foreach ($lines as $line) { $parts = explode(':', $line); if (isset($parts[1])) $data[$parts[0]] = $parts[1]; } return $data; } /** * Debug * * Set debug mode * @param bool set the debug mode on or off * @return void */ public function debug($bool) { $this->debug = (bool) $bool; } /** * Destructor * * Kill the connection * @return void */ function __destruct() { log_message('debug', 'Redis Class destruct'); if ($this->_connection) fclose($this->_connection); } }
