当前位置：Gxlcms > PHP教程 > PHPmemcache环形队列

PHPmemcache环形队列

时间：2021-07-01 10:21:17 帮助过：22人阅读

PHP memcache 环形队列类。新手，没咋学过数据结构，因为业务需要，所以只是硬着头皮模拟的！ 原形是 oschina上 lusi 分享的PHP memcache 队列代码。为使队列随时可入可出，且不受int长度越界危险（单链采取Head自增的话不作处理有越界可能），所以索性改写成环形队列。可能还有BUG，忘见谅！
  2. /**
  3. * PHP memcache 环形队列类
  4. * 原作者 LKK/lianq.net
  5. * 修改 FoxHunter
  6. * 因业务需要只保留的队列中的Pop和Push,修改过期时间为0即永久
  7. */
  8. class MQueue
  9. {
  10. public static $client;
  12. private $expire; //过期时间,秒,1~2592000,即30天内
  13. private $sleepTime; //等待解锁时间,微秒
  14. private $queueName; //队列名称,唯一值
  15. private $retryNum; //尝试次数
  16. private $MAXNUM; //最大队列容量
  17. private $canRewrite; //是否可以覆写开关，满出来的内容从头部开始覆盖重写原来的数据
  19. private $HEAD; //下一步要进入的指针位置
  20. private $TAIL; //下一步要进入的指针位置
  21. private $LEN; //队列现有长度
  23. const LOCK_KEY = '_Fox_MQ_LOCK_'; //锁存储标示
  24. const LENGTH_KEY = '_Fox_MQ_LENGTH_'; //队列现长度存储标示
  25. const VALU_KEY = '_Fox_MQ_VAL_'; //队列键值存储标示
  26. const HEAD_KEY = '_Fox_MQ_HEAD_'; //队列HEAD指针位置标示
  27. const TAIL_KEY = '_Fox_MQ_TAIL_'; //队列TAIL指针位置标示
  29. /*
  30. * 构造函数
  31. * 对于同一个$queueName，实例化时必须保障构造函数的参数值一致，否则pop和push会导队列顺序混乱
  32. */
  33. public function __construct($queueName = '', $maxqueue = 1, $canRewrite = false, $expire = 0, $config = '')
  34. {
  35. if (empty($config)) {
  36. self::$client = memcache_pconnect('127.0.0.1', 11211);
  37. } elseif (is_array($config)) { //array('host'=>'127.0.0.1','port'=>'11211')
  38. self::$client = memcache_pconnect($config['host'], $config['port']);
  39. } elseif (is_string($config)) { //"127.0.0.1:11211"
  40. $tmp = explode(':', $config);
  41. $conf['host'] = isset($tmp[0]) ? $tmp[0] : '127.0.0.1';
  42. $conf['port'] = isset($tmp[1]) ? $tmp[1] : '11211';
  43. self::$client = memcache_pconnect($conf['host'], $conf['port']);
  44. }
  45. if (!self::$client)
  46. return false;
  47. ignore_user_abort(true); //当客户断开连接,允许继续执行
  48. set_time_limit(0); //取消脚本执行延时上限
  49. $this->access = false;
  50. $this->sleepTime = 1000;
  51. $expire = (empty($expire)) ? 0 : (int) $expire + 1;
  52. $this->expire = $expire;
  53. $this->queueName = $queueName;
  54. $this->retryNum = 20000;
  55. $this->MAXNUM = $maxqueue != null ? $maxqueue : 1;
  56. $this->canRewrite = $canRewrite;
  57. $this->getHeadAndTail();
  58. if (!isset($this->HEAD) || empty($this->HEAD))
  59. $this->HEAD = 0;
  60. if (!isset($this->TAIL) || empty($this->TAIL))
  61. $this->TAIL = 0;
  62. if (!isset($this->LEN) || empty($this->LEN))
  63. $this->LEN = 0;
  65. }
  67. //获取队列首尾指针信息和长度
  68. private function getHeadAndTail()
  69. {
  70. $this->HEAD = (int) memcache_get(self::$client, $this->queueName . self::HEAD_KEY);
  71. $this->TAIL = (int) memcache_get(self::$client, $this->queueName . self::TAIL_KEY);
  72. $this->LEN = (int) memcache_get(self::$client, $this->queueName . self::LENGTH_KEY);
  73. }
  76. // 利用memcache_add原子性加锁
  77. private function lock()
  78. {
  79. if ($this->access === false) {
  80. $i = 0;
  81. while (!memcache_add(self::$client, $this->queueName . self::LOCK_KEY, 1, false, $this->expire)) {
  82. usleep($this->sleepTime);
  83. @$i++;
  84. if ($i > $this->retryNum) { //尝试等待N次
  85. return false;
  86. break;
  87. }
  88. }
  89. return $this->access = true;
  90. }
  91. return false;
  92. }
  94. //更新头部指针指向,指向下一个位置
  95. private function incrHead()
  96. {
  97. //$this->getHeadAndTail(); //获取最新指针信息 ,由于本方法体均在锁内调用，其锁内已调用了此方法，本行注释
  98. $this->HEAD++; //头部指针下移
  99. if ($this->HEAD >= $this->MAXNUM) {
  100. $this->HEAD = 0; //边界值修正
  101. }
  102. ;
  103. $this->LEN--; //Head的移动由Pop触发，所以相当于数量减少
  104. if ($this->LEN < 0) {
  105. $this->LEN = 0; //边界值修正
  106. }
  107. ;
  108. memcache_set(self::$client, $this->queueName . self::HEAD_KEY, $this->HEAD, false, $this->expire); //更新
  109. memcache_set(self::$client, $this->queueName . self::LENGTH_KEY, $this->LEN, false, $this->expire); //更新
  111. }
  113. //更新尾部指针指向，指向下一个位置
  114. private function incrTail()
  115. {
  117. //$this->getHeadAndTail(); //获取最新指针信息，由于本方法体均在锁内调用，其锁内已调用了此方法，本行注释
  118. $this->TAIL++; //尾部指针下移
  119. if ($this->TAIL >= $this->MAXNUM) {
  120. $this->TAIL = 0; //边界值修正
  121. }
  122. ;
  123. $this->LEN++; //Head的移动由Push触发，所以相当于数量增加
  124. if ($this->LEN >= $this->MAXNUM) {
  125. $this->LEN = $this->MAXNUM; //边界值长度修正
  126. }
  127. ;
  128. memcache_set(self::$client, $this->queueName . self::TAIL_KEY, $this->TAIL, false, $this->expire); //更新
  129. memcache_set(self::$client, $this->queueName . self::LENGTH_KEY, $this->LEN, false, $this->expire); //更新
  130. }
  133. // 解锁
  134. private function unLock()
  135. {
  136. memcache_delete(self::$client, $this->queueName . self::LOCK_KEY);
  137. $this->access = false;
  138. }
  140. //判断是否满队列
  141. public function isFull()
  142. {
  143. //外部直接调用的时候由于没有锁所以此处的值是个大概值，并不很准确，但是内部调用由于在前面有lock，所以可信
  144. if ($this->canRewrite)
  145. return false;
  146. return $this->LEN == $this->MAXNUM ? true : false;
  147. }
  149. //判断是否为空
  150. public function isEmpty()
  151. {
  152. //外部直接调用的时候由于没有锁所以此处的值是个大概值，并不很准确，但是内部调用由于在前面有lock，所以可信
  153. return $this->LEN == 0 ? true : false;
  154. }
  156. public function getLen()
  157. {
  158. //外部直接调用的时候由于没有锁所以此处的值是个大概值，并不很准确，但是内部调用由于在前面有lock，所以可信
  159. return $this->LEN;
  160. }
  162. /*
  163. * push值
  164. * @param mixed 值
  165. * @return bool
  166. */
  167. public function push($data = '')
  168. {
  170. $result = false;
  171. if (empty($data))
  172. return $result;
  174. if (!$this->lock()) {
  175. return $result;
  176. }
  178. $this->getHeadAndTail(); //获取最新指针信息
  180. if ($this->isFull()) { //只有在非覆写下才有Full概念
  181. $this->unLock();
  182. return false;
  183. }
  185. if (memcache_set(self::$client, $this->queueName . self::VALU_KEY . $this->TAIL, $data, MEMCACHE_COMPRESSED, $this->expire)) {
  186. //当推送后，发现尾部和头部重合（此时指针还未移动），且右边仍有未由Head读取的数据，那么移动Head指针，避免尾部指针跨越Head
  187. if ($this->TAIL == $this->HEAD && $this->LEN >= 1) {
  188. $this->incrHead();
  189. }
  190. $this->incrTail(); //移动尾部指针
  191. $result = true;
  192. }
  194. $this->unLock();
  195. return $result;
  196. }
  199. /*
  200. * Pop一个值
  201. * @param [length] int 队列长度
  202. * @return array
  203. */
  204. public function pop($length = 0)
  205. {
  206. if (!is_numeric($length))
  207. return false;
  209. if (!$this->lock())
  210. return false;
  212. $this->getHeadAndTail();
  214. if (empty($length))
  215. $length = $this->LEN; //默认读取所有
  217. if ($this->isEmpty()) {
  218. $this->unLock();
  219. return false;
  220. }
  221. //获取长度超出队列长度后进行修正
  222. if ($length > $this->LEN)
  223. $length = $this->LEN;
  225. $data = $this->popKeyArray($length);
  226. $this->unLock();
  227. return $data;
  228. }
  231. /*
  232. * pop某段长度的值
  233. * @param [length] int 队列长度
  234. * @return array
  235. */
  236. private function popKeyArray($length)
  237. {
  239. $result = array();
  240. if (empty($length))
  241. return $result;
  242. for ($k = 0; $k < $length; $k++) {
  243. $result[] = @memcache_get(self::$client, $this->queueName . self::VALU_KEY . $this->HEAD);
  244. @memcache_delete(self::$client, $this->queueName . self::VALU_KEY . $this->HEAD, 0);
  245. //当提取值后，发现头部和尾部重合（此时指针还未移动），且右边没有数据，即队列中最后一个数据被完全掏空，此时指针停留在本地不移动，队列长度变为0
  246. if ($this->TAIL == $this->HEAD && $this->LEN <= 1) {
  247. $this->LEN = 0;
  248. memcache_set(self::$client, $this->queueName . self::LENGTH_KEY, $this->LEN, false, $this->expire); //更新
  249. break;
  250. } else {
  251. $this->incrHead(); //首尾未重合，或者重合但是仍有未读取出的数据，均移动HEAD指针到下一处待读取位置
  252. }
  253. }
  254. return $result;
  255. }
  257. /*
  258. * 重置队列
  259. * * @return NULL
  260. */
  261. private function reset($all = false)
  262. {
  264. if ($all) {
  265. memcache_delete(self::$client, $this->queueName . self::HEAD_KEY, 0);
  266. memcache_delete(self::$client, $this->queueName . self::TAIL_KEY, 0);
  267. memcache_delete(self::$client, $this->queueName . self::LENGTH_KEY, 0);
  268. } else {
  269. $this->HEAD = $this->TAIL = $this->LEN = 0;
  270. memcache_set(self::$client, $this->queueName . self::HEAD_KEY, 0, false, $this->expire);
  271. memcache_set(self::$client, $this->queueName . self::TAIL_KEY, 0, false, $this->expire);
  272. memcache_set(self::$client, $this->queueName . self::LENGTH_KEY, 0, false, $this->expire);
  273. }
  274. }
  276. /*
  277. * 清除所有memcache缓存数据
  278. * @return NULL
  279. */
  280. public function memFlush()
  281. {
  282. memcache_flush(self::$client);
  283. }
  286. public function clear($all = false)
  287. {
  288. if (!$this->lock())
  289. return false;
  290. $this->getHeadAndTail();
  291. $Head = $this->HEAD;
  292. $Length = $this->LEN;
  293. $curr = 0;
  294. for ($i = 0; $i < $Length; $i++) {
  295. $curr = $this->$Head + $i;
  296. if ($curr >= $this->MAXNUM) {
  297. $this->HEAD = $curr = 0;
  298. }
  299. @memcache_delete(self::$client, $this->queueName . self::VALU_KEY . $curr, 0);
  300. }
  303. $this->unLock();
  304. $this->reset($all);
  305. return true;
  306. }
  309. }

人气教程排行

本站所有资源全部来源于网络，若本站发布的内容侵害到您的隐私或者利益，请联系我们删除！