PHP中使用Memache作为进程锁的操作类分享

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

<?php

// 使用Memache 作为进程锁

class lock_processlock{

// key 的前缀

protected $sLockKeyPre;

// 重试间隔

protected $iLockRetryInterval;

//重试次数

protected $iLockRetryCount;

//锁的过期时间

protected $iLockCacheTimeout;

// 锁过期后的回调函数

protected $onLockTimeoutFunc;

// memache 的实例

protected $oMemcache;

// 存储memcache失败后重试次数

protected $iMemcacheRetryCount;

public function __construct ($onLockTimeoutFunc=NULL) {

$aLockConfig = get_config('', 'lock');

$this->sLockKeyPre = self::LOCK_KEY_PRE;

$this->iLockRetryInterval = self::LOCK_RETRY_INTERVAL;

$this->iLockRetryCount =self::LOCK_RETRY_COUNT;

$this->iLockCacheTimeout = self::LOCK_CACHE_TIMEOUT;

$this->iMemcacheRetryCount = self::LOCK_CACHE_TIMEOUT;

if(!$onLockTimeoutFunc){

// 如果加锁不成功则调用回调函数,如果没有回调函数,使用本类中所带的

$onLockTimeoutFunc ='onLockTimeout' ;

}

$this->onLockTimeoutFunc = $onLockTimeoutFunc;

}

/**

连接memcache 服务器

*/

public function connect() {

if (! isset ( $this->oMemcache )) {

$this->oMemcache = new Memcache ();

$this->oMemcache->connect ( '127.0.0.1', 11211 );

}

return $this->oMemcache;

}

/*

向MeMcache中添加 key

*/

public addMemcache($sKey, $sValue, $iTimeout){

for($i= 0 ; $i<$this->iMemcacheRetryCount){

$bRes = $this->oMemcache->add($sKey, $sValue, $iTimeout);

if($bRes){

return true ;

}

// 如果加锁不成功,sleep 之后,从新加锁

usleep($this->iLockRetryInterval*1000);

}

return false ;

}

/*

加锁

*/

public function lock($sLockID){

$oMemcache = $this->connect();

$sKey = $this->sLockKeyPre . $sLockID;

// 加锁如果不成功可以多试几次

for($i = 0 ; $i <$this->iLockRetryCount ; $i++){

// 这里设置value 的值可以随便设置

if($this->addMemcache($sKey,'1',$this->iLockCacheTimeout)){

return true ;

}

// 如果加锁不成功,sleep 之后,从新加锁

usleep($this->iLockRetryInterval*1000);

}

// 若还不成功,则加锁失败,调用回调函数,.也就是失败后需要处理的操作

if(is_callable($this->onLockTimeoutFunc)){

// 调用函数

call_user_func($this->onLockTimeoutFunc);

}

}

/*

解锁操作

*/

public function unlock($sLockID){

$oMemcache = $this->connect();

$sKey = $this->sLockKeyPre . $sLockID;

// 删除key

return $this->oMemcache->delete($sKey);

}

/**

如果加锁不成功,则执行如下操作

*/

public function onLockTimeout(){

echo ("加锁超时");

}

}

// 应用实例

$oLock = new lock_processlock();

$lockResource = "test";

// 加锁

$oLock->lock($lockResource);

// 解锁