thinkphp socket to support sso (php socket 不稳定,不建议使用)

cli code

#!/bin/bash

filepath=$(cd "$(dirname "$0")"; pwd)
cd $filepath
#running status:sso.lock,stop status:ssostop.lock
TMPFILE=$filepath/sso.lock
STOPFILE=$filepath/ssostop.lock
runCLI(){
touch $TMPFILE
echo "sso server process started.\n"
logname="sso_"$(date +%Y%m%d);
su www-data -c "php index.php sso start 1>>./Log/$logname.log 2>&1"
rm -f $STOPFILE;
}

restartCLI(){
stopCLI
echo "Waiting for stop sso process."
sleep 1
echo "Exit prev sso process."
runCLI &
}

stopCLI(){
if [ -f $TMPFILE ];then
touch $STOPFILE
fi
rm -f ${TMPFILE};
logname="sso_"$(date +%Y%m%d);
su www-data -c "php index.php sso stop 1>>./Log/$logname.log 2>&1"
echo "Sent stop sso server signal."
}

if [ "$1" = "-stop" ] ;then
stopCLI
elif [ "$1" = "-start" -o "$1" = "" ] ;then
if [ -f $TMPFILE ];then
echo "Another process is running."
else
runCLI &
fi
elif [ "$1" = "-restart" ] ;then
restartCLI
else
echo "Please select option -start -restart -stop"
fi



socket server code

<?php
namespace Admin\Action;
use Think\Action;
class SsoAction extends Action {
private $sock;
private $clients = array();
// 停止socket。kill 后,有可能socket临时文件还是没删除。会导致同一个端口无法启动
//check port status:netstat -ano |grep port
//search pid: lsof -i:port
//kill pid
//rm -f sso.lock
public function run($taskType)
{
switch ($taskType) {
case 'start':
$this->_start();
break;
default:
echo datetime() . " unknow task type!" . PHP_EOL;
break;
}
}

public function index() {
echo datetime() . " server msg: ". $_SERVER['argv'][1] . " " . $_SERVER['argv'][2] . PHP_EOL;
$this->run($_SERVER['argv'][2]);
}

private function _start()
{
set_time_limit(0);
ob_implicit_flush();

if (($this->sock = socket_create(AF_INET, SOCK_STREAM, SOL_TCP)) === false) {
echo datetime() . " socket_create() failed: reason:" . socket_strerror(socket_last_error()) . "\n";
exit;
}

// start
if (socket_bind($this->sock, C('SSO_ADDRESS'), C('SSO_PORT')) === false) {
echo datetime() . " socket_bind failed, reason:" . socket_strerror(socket_last_error($sock)) . PHP_EOL;
exit;
}

// listen
if (socket_listen($this->sock, 5) === false) {
echo datetime() . " socket_listen() failed: reason: " . socket_strerror(socket_last_error($sock)) . PHP_EOL;
exit;
}
echo datetime() . " start socket server success!" . PHP_EOL;
do {
// accept socket msg failed
if (($msgsock = socket_accept($this->sock)) === false) {
echo datetime() . " socket_accepty() failed :reason:" . socket_strerror(socket_last_error($this->sock)) . PHP_EOL;
break;
}
//read client message
$buf = socket_read($msgsock, 1024);
$data = json_decode($buf, true);
$userID = $data['userID'];
if ($data['msg'] == 'logout') {
// clear session table
$this->_logout($userID);
} elseif ($data['msg'] == "check") {
// check cache with user ID
$back = S($userID) ? "success" : "fail";
echo datetime() . " no cache of {$userID} " . PHP_EOL;
if (false === socket_write($msgsock, $back)) {
echo datetime() . " socket_write() failed reason:" . socket_strerror(socket_last_error($this->sock)) . PHP_EOL;
}
}
socket_close($msgsock);
} while (true);
}

private function _logout($userID) {
if (!$userID) return false;
S($userID, null);
echo datetime() . " userID:{$userID} logout by client. " . PHP_EOL;
}
}
?>


socket client code

<?php
namespace Admin\Action;
use Think\Action;
use Org\Util\Socket;
class SsoClientAction extends Action {
private $SocketClient;
/**
* loop check sso server auth status
*/
public function run()
{
$result = true;
while($result===true) {
$this->_connect();
$result = $this->send();
if (!$result) {
// logout
session_destroy();
}
sleep(3);
}
//$this->_connect();
//$result = $this->send();
}

private function _connect()
{
$config = array('persistent' => false,
'host' => C("SSO_ADDRESS"),
'protocol' => 'tcp',
'port' => C("SSO_PORT"),
'timeout' => 5);
$this->SocketClient = new Socket($config);
$this->SocketClient->connect();
}

/**
* @param msg check or logout string
*/
public function send($msg="check")
{
$userID = $_SESSION[C('USER_AUTH_KEY')];
if (!$userID) $msg = "logout";
$arr = array("msg"=>$msg, $userID=>$userID);
$data = json_encode($arr);
$result = $this->SocketClient->write($data);
//echo datetime() . " send msg:{$msg} \r\n" . PHP_EOL;
$response = $this->SocketClient->read();
echo datetime() . " " . $response . " from sso server \r\n" . PHP_EOL;
$this->SocketClient->disconnect();
if ($response == "logout" || $response == "fail") {
echo datetime . " logout or fail \r\n" . PHP_EOL;
session_destroy();
return false;
}
return true;
}
}
?>
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值