基于memcache接口的统一存储工具类设计

刚开始处理的时候，临时想做一个统一接口的key-value数据的存储方式。然后我模拟memcached接口，写了个简单的k-v数据存储的工具类。
临时想到，暂且记下。

1. 接口类 /**

* 存储工具抽象接口类

*

*/

abstract class StoreTool

{

abstract public function set($key, $val);

abstract public function get($key);

abstract public function replace($key, $val);

abstract public function delete($key);

abstract public function flush();

abstract public function increment($key, $val=1);

abstract public function add($key, $val);

}

复制代码
2. memcached和mysql处理（mysql用的heap engine） /**

* mysql heap engin 存储工具类

*

*/

class DbStoreTool extends StoreTool

{

static private $_instance;

private $_dbh;

private $_tablename = 'mmouc_memory_kv';

static public function getInstance($config) {

if (self::$_instance == null) {

   self::$_instance = new self($config);

}

return self::$_instance;

}

private function __construct($config) {

$conn = $config['persistency'] ? mysql_pconnect($config['host'].':'.$config['port'], $config['user'], $config['password'])

     : mysql_connect($config['host'].':'.$config['port'], $config['user'], $config['password']);

if($conn) {

   if($config['charset']) mysql_query("SET NAMES '" . $config['charset'] . "'", $conn);

   if(!empty($config['database'])) {

    $dbselect = mysql_select_db($config['database'], $conn);

    if(!$dbselect) {

     mysql_close($conn);

     $conn = $dbselect;

    }

   }

   $this->_dbh = $conn;

   $this->_tablename = $config['tablename'];

}

}

/**

* Store data at the server

* stores an item var with key on the memcached server

*

* @param unknown_type $key

* @param unknown_type $val

* @return unknown

*/

public function set($key, $val) {

$res = $this->update($key, $val);

if (0 === $res) {

   return $this->add($key, $val);

}

return true;

}

/**

* Replace value of the existing item

* should be used to replace value of existing item with key.

* In case if item with such key doesn't exists, This function returns FALSE

*

* @param unknown_type $key

* @param unknown_type $val

* @return unknown

*/

public function replace($key, $val) {

$res = $this->update($key, $val);

if (0 === $res) {

   return false;

}

return true;

}

public function get($key) {

if (is_array($key)) {

   $in_keys = "'" . implode("','", $key) . "'";

   $sql = "

    SELECT `k`, `v` FROM `" . $this->_tablename . "`

    WHERE `k` IN ({$in_keys})

   ";

   $res = mysql_query($sql, $this->_dbh);

   if (empty($res)) {

    return MMO_STORE_OP_ERROR;

   }

   $_arr_res = array();

   while ($row = mysql_fetch_assoc($res)) {

    $row['v'] = unserialize($row['v']);

    $_arr_res[$row['k']] = $row;

   }

   $out = array();

   foreach ($key as $_k) {

    $out[] = $_arr_res[$_k]['v'];

   }

   return $out;

} else if (is_string($key)) {

   $sql = "

    SELECT `v` FROM `" . $this->_tablename . "`

    WHERE `k`='{$key}'

   ";

   $res = mysql_query($sql, $this->_dbh);

   if (empty($res)) {

    return -1;

   }

   $row = mysql_fetch_assoc($res);

   if (empty($row)) {

    return MMO_STORE_ITEM_NOT_EXIST;

   }

   return unserialize($row['v']);

} else {

   return false;

}

}

public function delete($key) {

$sql = "

   DELETE FROM `" . $this->_tablename . "`

   WHERE `k`='$key'

   ";

$res = mysql_query($sql, $this->_dbh);

if (!$res) {

   return MMO_STORE_OP_ERROR;

}

return mysql_affected_rows($this->_dbh);

}

public function flush() {

$sql = " TRUNCATE TABLE `" . $this->_tablename . "` ";

$res = mysql_query($sql, $this->_dbh);

return $res ? true : false;

}

/**

*

* TODO:

* 修改这里的并发访问问题

*

* @param unknown_type $key

* @param unknown_type $val

* @return unknown

*/

public function increment($key, $val=1) {

$_db_val = $this->get($key);

if (MMO_STORE_ITEM_NOT_EXIST == $_db_val) {

   //不存在

   return false;

}

$val = intval($_db_val) + intval($val);

$this->update($key, $val);

return $val;

}

/**

* Add an item to the server

*

* stores variable var with key only if such key doesn't exist at the server yet

*

* @param unknown_type $key

* @param unknown_type $val

* @return unknown

*/

public function add($key, $val) {

if (!$this->_isExist($key)) {

   $val = serialize($val);

   $time = time();

   $sql = "

    INSERT INTO `" . $this->_tablename . "`

    SET `k`='{$key}',

        `v`='{$val}',

        `t`='{$time}'

   ";

   $res = mysql_query($sql, $this->_dbh);

   return $res ? true : MMO_STORE_OP_ERROR;

} else {

   return false;

}

}

private function _isExist($key, $val='') {

$sql = "

   SELECT COUNT(`k`) as 'num'

   FROM `" . $this->_tablename . "`

   WHERE `k`='{$key}'

";

!empty($val) && $sql .= ", `v`='" . serialize($val) . "'";

$res = mysql_query($sql, $this->_dbh);

if (empty($res)) {

   return -1;

}

$row = mysql_fetch_assoc($res);

return $row['num'] ? true : false;

}

private function update($key, $val) {

$val = serialize($val);

$time = time();

$sql = "

   UPDATE `" . $this->_tablename . "`

   SET `v`='{$val}',

       `t`='{$time}'

   WHERE `k`='$key'

   ";

$res = mysql_query($sql, $this->_dbh);

if (!$res) {

   return MMO_STORE_OP_ERROR;

}

return mysql_affected_rows($this->_dbh);

}

}

class FileStoreTool

{

}

class MemcacheStoreTool extends StoreTool

{

static private $_instance;

private $_memcacheHandler;

static public function getInstance($config) {

if (self::$_instance == null) {

   self::$_instance = new self($config);

}

return self::$_instance;

}

private function __construct($config) {

$this->_memServers = $config;

$this->_initMemcacheObj();

}

public function set($key, $val) {

return $this->_memcacheHandler->set($key, $val);

}

public function get($key) {

return $this->_memcacheHandler->get($key);

}

public function replace($key, $val) {

return $this->_memcacheHandler->replace($key, $val);

}

public function delete($key) {

return $this->_memcacheHandler->delete($key);

}

public function flush() {

return $this->_memcacheHandler->flush();

}

public function increment($key, $val=1) {

return $this->_memcacheHandler->increment($key, $val);

}

public function add($key, $val) {

return $this->_memcacheHandler->add($key, $val);

}

/**

    * 检查保存Session数据的路径是否存在

    *

    * @return bool 成功返回true

    */

   private function _initMemcacheObj(){

       if (!class_exists('Memcache') || !function_exists('memcache_connect')){

           die('Failed: Memcache extension not install, please from http://pecl.php.net download and install');

       }

       if ($this->_memcacheHandler && is_object($this->_memcacheHandler)){

           return true;

       }

       $this->_memcacheHandler = new Memcache;

       if (!empty($this->_memServers)) {

          foreach ($this->_memServers as $_host => $_port) {

            $this->_memcacheHandler->addServer($_host, $_port);

        }

       }

       return true;

   }

}

复制代码
3. mysql和file相关

mysql 记录方式，对应表建表语句如下：
CREATE TABLE IF NOT EXISTS `mmouc_memory_kv` (
`k` varchar(40) NOT NULL COMMENT '键名',
`v` varchar(2048) NOT NULL COMMENT '键值的serialize值',
`t` int(10) NOT NULL DEFAULT '0' COMMENT '更新时间',
PRIMARY KEY (`k`)
) ENGINE=MEMORY DEFAULT CHARSET=utf8 COMMENT='代替memcache数据记录表';
对于file，暂未处理，以后补上。

4. todo:

a. 对于memcached数据的遍历方式策略，需要做一个完备而统一接口的策略算法处理类，可以在多种分级存储模式，和存储策略间任意切换；
b. file存储工具类补上

基于memcache接口的统一存储工具类设计

猜你喜欢