PHP基于共享内存的顺序号发生器
本文地址：http://tongxinmao.com/Article/Detail/id/126
/*
* 信号量(Semaphore)。
* 这是一个包装类，用于解决不同平台下对“信号量”的不同实现方式。
* 目前这个类只是象征性的，在 Windows 平台下实际是空跑（并没有真的实现互斥）。
*/
class SemWrapper
{
private $hasSemSupport;
private $sem;
const SEM_KEY = 1;
public function __construct()
{
$this->hasSemSupport = function_exists( 'sem_get' );
if ( $this->hasSemSupport ) {
$this->sem = sem_get( self::SEM_KEY );
}
}
public function acquire() {
if ( $this->hasSemSupport ) {
return sem_acquire( $this->sem );
}
return true;
}
public function release() {
if ( $this->hasSemSupport ) {
return sem_release( $this->sem );
}
return true;
}
}
/*
* 顺序号发生器。
*/
class SeqGenerator
{
const SHM_KEY = 1;
/**
* 对顺序号发生器进行初始化。
* 仅在服务器启动后的第一次调用有效，此后再调用此方法没有实际作用。
* @param int $start 产生顺序号的起始值。
* @return boolean 返回 true 表示成功。
*/
static public function init( $start = 1 )
{
// 通过信号量实现互斥，避免对共享内存的访问冲突
$sw = new SemWrapper;
if ( ! $sw->acquire() ) {
return false;
}
// 打开共享内存
$shm_id = shmop_open( self::SHM_KEY, 'n', 0644, 4 );
if ( empty($shm_id) ) {
// 因使用了 'n' 模式，如果无法打开共享内存，可以认为该共享内存已经创建，无需再次初始化
$sw->release();
return true;
}
// 在共享内存中写入初始值
$size = shmop_write( $shm_id, pack( 'L', $start ), 0 );
if ( $size != 4 ) {
shmop_close( $shm_id );
$sw->release();
return false;
}
// 关闭共享内存，释放信号量
shmop_close( $shm_id );
$sw->release();
return true;
}
/**
* 产生下一个顺序号。
* @return int 产生的顺序号
*/
static public function next()
{
// 通过信号量实现互斥，避免对共享内存的访问冲突
$sw = new SemWrapper;
if ( ! $sw->acquire() ) {
return 0;
}
// 打开共享内存
$shm_id = shmop_open( self::SHM_KEY, 'w', 0, 0 );
if ( empty($shm_id) ) {
$sw->release();
return 0;
}
// 从共享内存中读出顺序号
$data = shmop_read( $shm_id, 0, 4 );
if ( empty($data) ) {
$sw->release();
return 0;
}
$arr = unpack( 'L', $data );
$seq = $arr[1];
// 把下一个顺序号写入共享内存
$size = shmop_write( $shm_id, pack( 'L', $seq + 1 ), 0 );
if ( $size != 4 ) {
$sw->release();
return 0;
}
// 关闭共享内存，释放信号量
shmop_close( $shm_id );
$sw->release();
return $seq;
}
}
$a = SeqGenerator::init( time() );
var_dump($a);
for ( $i=0; $i < 10; $i++ ) {
$seq = SeqGenerator::next();
var_dump($seq);
}
上一篇：使用 watchdog 构建高可用性的 Linux 系统及应用
下一篇：爱宝a-58USB打印机描述符