TimeTrex/classes/modules/other/Redis_Cache_Lite.class.php

434 lines
14 KiB
PHP
Raw Normal View History

2022-12-13 07:10:06 +01:00
<?php
/*********************************************************************************
*
* TimeTrex is a Workforce Management program developed by
* TimeTrex Software Inc. Copyright (C) 2003 - 2021 TimeTrex Software Inc.
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU Affero General Public License version 3 as published by
* the Free Software Foundation with the addition of the following permission
* added to Section 15 as permitted in Section 7(a): FOR ANY PART OF THE COVERED
* WORK IN WHICH THE COPYRIGHT IS OWNED BY TIMETREX, TIMETREX DISCLAIMS THE
* WARRANTY OF NON INFRINGEMENT OF THIRD PARTY RIGHTS.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
* details.
*
*
* You should have received a copy of the GNU Affero General Public License along
* with this program; if not, see http://www.gnu.org/licenses or write to the Free
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA.
*
*
* You can contact TimeTrex headquarters at Unit 22 - 2475 Dobbin Rd. Suite
* #292 West Kelowna, BC V4T 2E9, Canada or at email address info@timetrex.com.
*
*
* The interactive user interfaces in modified source and object code versions
* of this program must display Appropriate Legal Notices, as required under
* Section 5 of the GNU Affero General Public License version 3.
*
*
* In accordance with Section 7(b) of the GNU Affero General Public License
* version 3, these Appropriate Legal Notices must retain the display of the
* "Powered by TimeTrex" logo. If the display of the logo is not reasonably
* feasible for technical reasons, the Appropriate Legal Notices must display
* the words "Powered by TimeTrex".
*
********************************************************************************/
/**
* Class Redis_Cache_Lite
*/
class Redis_Cache_Lite extends Cache_Lite {
private $_redisHost;
private $_redisHostConn;
/**
* Redis_Cache_Lite constructor.
* @param array $options
*/
function __construct( $options = [ null ] ) {
parent::__construct( $options );
if ( defined( 'ADODB_DIR' ) ) {
include_once( ADODB_DIR . '/adodb-csvlib.inc.php' );
}
$this->redisConnectMaster();
return true;
}
/**
* @param $key
* @return bool|Redis
*/
function redisConnect( $key ) {
if ( isset( $this->_redisHostConn[$key] ) && $this->_redisHostConn[$key] === false ) {
Debug::Text( 'Previous error connecting to the Redis database, not attempting again during this request...', __FILE__, __LINE__, __METHOD__, 1 );
return false;
}
try {
global $config_vars;
if ( !isset( $config_vars['database']['persistent_connections'] ) ) {
$config_vars['database']['persistent_connections'] = false;
}
if ( extension_loaded( 'redis' ) ) { //Make sure REDIS PHP extension is enabled to avoid PHP FATAL error.
$this->_redisHostConn[$key] = new Redis();
//Try with 2 second timeout, we don't want redis to block requests if its down.
if ( $config_vars['database']['persistent_connections'] == true ) {
$connection_retval = @$this->_redisHostConn[$key]->pconnect( trim( $this->_redisHostHost[$key] ), 6379, 2 );
} else {
$connection_retval = @$this->_redisHostConn[$key]->connect( trim( $this->_redisHostHost[$key] ), 6379, 2 );
}
if ( $connection_retval === true ) {
if ( isset( $this->_redisDB ) && $this->_redisDB != '' ) {
if ( $this->_redisHostConn[$key]->select( $this->_redisDB ) === false ) {
//return $this->raiseError('Cache_Lite : Unable to switch redis DB to: '. $this->_redisDB, -2); //In order to catch these we need to include PEAR.php all the time.
return false;
}
//else {
// Debug::text('Switched REDIS DB to: '. $this->_redisDB, __FILE__, __LINE__, __METHOD__, 10);
//}
}
return $this->_redisHostConn[$key];
} else {
$this->_redisHostConn[$key] = false; //Prevent further connections from timing out during this request...
Debug::Text( 'Error connecting to the Redis database! (a) Host: ' . $this->_redisHostHost[$key], __FILE__, __LINE__, __METHOD__, 1 );
}
} else {
$this->_redisHostConn[$key] = false; //Prevent further connections from timing out during this request...
Debug::Text( 'REDIS PHP extension is not installed/enabled!', __FILE__, __LINE__, __METHOD__, 1 );
unset( $e );
//throw new DBError($e);
}
} catch ( Exception $e ) {
$this->_redisHostConn[$key] = false; //Prevent further connections from timing out during this request...
Debug::Text( 'Error connecting to the Redis database! (b) Host: ' . $this->_redisHostHost[$key] . ' Message: ' . $e->getMessage(), __FILE__, __LINE__, __METHOD__, 1 );
unset( $e );
//throw new DBError($e);
}
//return $this->raiseError('Cache_Lite : Unable to connect to redis host: '. $key, -2); //In order to catch these we need to include PEAR.php all the time.
return false;
}
/**
* @return bool
*/
function redisConnectMaster() {
Debug::text( 'Connecting to REDIS Host: ' . $this->_redisHost, __FILE__, __LINE__, __METHOD__, 10 );
if ( isset( $this->_redisHost ) && $this->_redisHost != '' ) {
$split_server = explode( ',', $this->_redisHost );
$i = 0;
foreach ( $split_server as $server ) {
$server = trim( $server );
if ( $i == 0 ) {
$this->_redisHostHost['master'] = $server;
} else {
$this->_redisHostHost['slave_' . $i] = $server;
}
$i++;
}
//Debug::Arr($this->_redisHostHost, 'REDIS Hosts: ', __FILE__, __LINE__, __METHOD__, 10);
return $this->redisConnect( 'master' );
}
return false;
}
/**
* @param string $name
* @param mixed $value
*/
function setOption( $name, $value ) {
$availableOptions = [ 'redisHost', 'redisDB', 'errorHandlingAPIBreak', 'hashedDirectoryUmask', 'hashedDirectoryLevel', 'automaticCleaningFactor', 'automaticSerialization', 'fileNameProtection', 'memoryCaching', 'onlyMemoryCaching', 'memoryCachingLimit', 'cacheDir', 'caching', 'lifeTime', 'fileLocking', 'writeControl', 'readControl', 'readControlType', 'pearErrorMode', 'hashedDirectoryGroup', 'cacheFileMode', 'cacheFileGroup' ];
if ( in_array( $name, $availableOptions ) ) {
$property = '_' . $name;
$this->$property = $value;
}
}
/**
* @param string $id
* @param string $group
* @param bool $doNotTestCacheValidity
* @return bool|mixed
*/
function get( $id, $group = 'default', $doNotTestCacheValidity = false ) {
$this->_id = $id;
$this->_group = $group;
if ( $this->_caching ) {
$this->_setRefreshTime();
$this->_setFileName( $id, $group );
clearstatcache();
if ( $this->_memoryCaching ) {
if ( isset( $this->_memoryCachingArray[$this->_file] ) ) {
if ( $this->_automaticSerialization ) {
return unserialize( $this->_memoryCachingArray[$this->_file] );
}
return $this->_memoryCachingArray[$this->_file];
}
if ( $this->_onlyMemoryCaching ) {
return false;
}
}
$data = $this->_read();
if ( ( $data ) && ( $this->_memoryCaching ) ) {
$this->_memoryCacheAdd( $data );
}
if ( ( $this->_automaticSerialization ) && ( is_string( $data ) ) ) {
$data = unserialize( $data );
}
return $data;
}
return false;
}
/**
* @return bool
*/
function _read() {
$redis = $this->redisConnect( 'master' );
//if ( !PEAR::isError($redis) ) {
if ( is_object( $redis ) && get_class( $redis ) == 'Redis' ) {
try {
return $redis->get( $this->_file );
} catch ( Exception $e ) {
Debug::Text( 'Redis Error: Message: ' . $e->getMessage(), __FILE__, __LINE__, __METHOD__, 1 );
}
}
//return $this->raiseError('Cache_Lite : Unable to read cache !', -2); //In order to catch these we need to include PEAR.php all the time.
return false;
}
/**
* @param string $data
* @param string $id UUID
* @param string $group
* @return bool
*/
function save( $data, $id = null, $group = 'default' ) {
if ( $this->_caching ) {
if ( $this->_automaticSerialization ) {
$data = serialize( $data );
}
if ( isset( $id ) ) {
$this->_setFileName( $id, $group );
}
if ( $this->_memoryCaching ) {
$this->_memoryCacheAdd( $data );
if ( $this->_onlyMemoryCaching ) {
return true;
}
}
if ( $this->_automaticCleaningFactor > 0 && ( $this->_automaticCleaningFactor == 1 || mt_rand( 1, $this->_automaticCleaningFactor ) == 1 ) ) {
$this->clean( false, 'old' );
}
$res = $this->_write( $data );
if ( is_object( $res ) ) {
// $res is a PEAR_Error object
if ( !( $this->_errorHandlingAPIBreak ) ) {
return false; // we return false (old API)
}
}
return $res;
}
return false;
}
/**
* @param string $id
* @param string $group
*/
function _setFileName( $id, $group ) {
//if ($this->_fileNameProtection) {
// $suffix = md5($group).'_'.md5($id);
//} else {
$suffix = $group . '_' . $id;
//}
$this->_fileName = $suffix;
$this->_file = $suffix;
}
/**
* @param string $data
* @return bool
*/
function _write( $data ) {
$redis = $this->redisConnect( 'master' );
//if ( !PEAR::isError($redis) ) {
if ( is_object( $redis ) && get_class( $redis ) == 'Redis' ) {
//Debug::text('Writing to REDIS as KEY: '. $this->_file, __FILE__, __LINE__, __METHOD__, 10);
try {
//In the case where the cache data could exist on read-only (slave) servers, remove it from
// the other servers while skipping master, then cache it on the master server.
//We have to do it in this order, otherwise if the same server is referenced twice in the .ini config, the cache will be set, then deleted immediately after.
// **NOTE: This greatly increases the load across all REDIS servers, so instead just do this manually when needed outside this class. (ie: Authentication->Update() )
//$this->_unlink( $this->_file, true );
return $redis->set( $this->_file, $data, $this->_lifeTime );
} catch ( Exception $e ) {
Debug::Text( 'Redis Error: Message: ' . $e->getMessage(), __FILE__, __LINE__, __METHOD__, 1 );
}
}
//return $this->raiseError('Cache_Lite : Unable to write cache file : '.$this->_file, -1); //In order to catch these we need to include PEAR.php all the time.
return false;
}
/**
* @param string $file
* @param bool $skip_master
* @return bool
*/
function _unlink( $file, $skip_master = false ) {
//When multiple redis servers are specified, we need to expire cache on them all.
foreach ( $this->_redisHostHost as $server_key => $value ) {
if ( $skip_master == false || ( $skip_master == true && $server_key != 'master' ) ) {
$redis = $this->redisConnect( $server_key );
//if ( !PEAR::isError($redis) ) {
if ( is_object( $redis ) && get_class( $redis ) == 'Redis' ) {
//Debug::text('Deleting REDIS as KEY: '. $this->_file .' Server Key: '. $server_key, __FILE__, __LINE__, __METHOD__, 10);
try {
if ( $redis->unlink( $file ) === false ) { //Use 'UNLINK' instead of 'DEL' as it frees memory asynchronously and can be slightly faster in some cases.
//return $this->raiseError('Cache_Lite : Unable to delete cache file : '.$this->_file, -1); //In order to catch these we need to include PEAR.php all the time.
return false;
}
} catch ( Exception $e ) {
Debug::Text( 'Redis Error: Message: ' . $e->getMessage(), __FILE__, __LINE__, __METHOD__, 1 );
}
}
}
}
return true;
}
//When using redis, no need to touch cache files ever.
function _touchCacheFile() {
return true;
}
/**
* @param bool $group
* @param string $mode
* @param bool $skip_master
* @return bool
*/
function clean( $group = false, $mode = 'ingroup', $skip_master = false ) {
//Make sure we still clear local PHP memory cache too.
if ( $this->_memoryCaching ) {
foreach ( $this->_memoryCachingArray as $key => $v ) {
if ( $group == false || strpos( $key, $group . '_' ) !== false ) {
unset( $this->_memoryCachingArray[$key] );
$this->_memoryCachingCounter = $this->_memoryCachingCounter - 1;
}
}
if ( $this->_onlyMemoryCaching ) {
return true;
}
}
//When multiple redis servers are specified, we need to expire cache on them all.
foreach ( $this->_redisHostHost as $server_key => $value ) {
if ( $skip_master == false || ( $skip_master == true && $server_key != 'master' ) ) {
$redis = $this->redisConnect( $server_key );
//if ( !PEAR::isError($redis) ) {
if ( is_object( $redis ) && get_class( $redis ) == 'Redis' ) {
try {
if ( $group != '' ) {
$redis->eval( 'return redis.call(\'del\', unpack(redis.call(\'keys\', ARGV[1])))', [ $group . '_*' ] );
} else {
$redis->flushdb(); //If no group is specified, flush all keys in DB.
}
} catch ( Exception $e ) {
Debug::Text( 'Redis Error: Message: ' . $e->getMessage(), __FILE__, __LINE__, __METHOD__, 1 );
}
}
}
}
return true;
}
/*
* Support ADODB Cache module.
*/
var $createdir = false; // do not set this to true unless you use temp directories in cache path
/**
* @param $filename
* @param $contents
* @param bool $debug
* @return bool
*/
function writecache( $filename, $contents, $debug = false ) {
return $this->save( $contents, $filename, 'adodb' );
}
/**
* @param $filename
* @param $err
* @param $secs2cache
* @param $rsClass
* @return mixed
*/
function readcache( $filename, &$err, $secs2cache, $rsClass ) {
$rs = explode( "\n", $this->get( $filename, 'adodb' ) );
unset( $rs[0] );
$rs = join( "\n", $rs );
return unserialize( $rs );
}
/**
* @param bool $debug
* @return bool
*/
function flushall( $debug = false ) {
return $this->clean( 'adodb' );
}
/**
* @param $filename
* @param bool $debug
* @return bool
*/
function flushcache( $filename, $debug = false ) {
return $this->remove( $filename, 'adodb' );
}
/**
* @param $dir
* @param $hash
* @return bool
*/
function createdir( $dir, $hash ) {
return true;
}
}
?>