分表
This commit is contained in:
221
application/common/library/GiftTableManager.php
Normal file
221
application/common/library/GiftTableManager.php
Normal file
@@ -0,0 +1,221 @@
|
||||
<?php
|
||||
// application/common/library/GiftTableManager.php
|
||||
|
||||
namespace app\common\library;
|
||||
|
||||
use think\Db;
|
||||
use think\Exception;
|
||||
use think\facade\Log;
|
||||
|
||||
class GiftTableManager
|
||||
{
|
||||
/**
|
||||
* 根据时间戳获取分表名
|
||||
* @param int $timestamp 时间戳
|
||||
* @return string 表名
|
||||
*/
|
||||
public static function getTableName($timestamp = null)
|
||||
{
|
||||
$timestamp = $timestamp ?: time();
|
||||
$month = date('Ym', $timestamp);
|
||||
return 'fa_vs_give_gift_' . $month;
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建月份分表（纯PHP实现，不使用存储过程）
|
||||
* @param string $yearMonth 年月格式：202401
|
||||
* @return bool
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function createMonthTable($yearMonth)
|
||||
{
|
||||
if (!preg_match('/^\d{6}$/', $yearMonth)) {
|
||||
throw new Exception('年月格式错误');
|
||||
}
|
||||
|
||||
$tableName = 'fa_vs_give_gift_' . $yearMonth;
|
||||
|
||||
// 检查表是否已存在
|
||||
try {
|
||||
$checkSql = "SHOW TABLES LIKE '{$tableName}'";
|
||||
$exists = Db::query($checkSql);
|
||||
if ($exists) {
|
||||
return true;
|
||||
}
|
||||
} catch (\Exception $e) {
|
||||
// 表不存在，继续创建
|
||||
}
|
||||
|
||||
// 创建分表SQL
|
||||
$sql = "
|
||||
CREATE TABLE IF NOT EXISTS `{$tableName}` (
|
||||
`id` bigint NOT NULL COMMENT '雪花ID',
|
||||
`user_id` int NOT NULL DEFAULT '0' COMMENT '用户id 送礼物的人',
|
||||
`gift_id` int NOT NULL DEFAULT '0' COMMENT '礼物id',
|
||||
`gift_type` tinyint(1) NOT NULL DEFAULT '1' COMMENT '礼物类型，1-正常礼物，2-盲盒开出来的，3-签约金币',
|
||||
`number` int unsigned NOT NULL DEFAULT '0' COMMENT '礼物数量',
|
||||
`gift_user` int NOT NULL DEFAULT '0' COMMENT '收礼物的人',
|
||||
`from_id` int NOT NULL DEFAULT '0' COMMENT '来源ID，（房间ID）',
|
||||
`pit_number` int NOT NULL DEFAULT '0' COMMENT '麦位编号（废弃）',
|
||||
`total_price` decimal(16,2) NOT NULL COMMENT '礼物总价值',
|
||||
`type` int NOT NULL DEFAULT '1' COMMENT '1金币购买 2送背包礼物',
|
||||
`from` int NOT NULL DEFAULT '1' COMMENT '1私聊送礼物 2语聊房送礼 3视频直播房送礼（暂时不用） 4动态打赏 6私密房间送礼',
|
||||
`createtime` bigint NOT NULL COMMENT '创建时间（秒时间戳）',
|
||||
`updatetime` bigint DEFAULT NULL COMMENT '更新时间（秒时间戳）',
|
||||
PRIMARY KEY (`id`),
|
||||
KEY `idx_createtime` (`createtime`) USING BTREE,
|
||||
KEY `idx_user_id` (`user_id`) USING BTREE,
|
||||
KEY `idx_gift_user` (`gift_user`) USING BTREE,
|
||||
KEY `idx_from_id` (`from_id`) USING BTREE,
|
||||
KEY `idx_room_user_time` (`from_id`, `user_id`, `createtime`) USING BTREE,
|
||||
KEY `idx_room_receiver_time` (`from_id`, `gift_user`, `createtime`) USING BTREE,
|
||||
KEY `idx_user_time` (`user_id`, `createtime`) USING BTREE,
|
||||
KEY `idx_receiver_time` (`gift_user`, `createtime`) USING BTREE,
|
||||
KEY `idx_combined_search` (`from_id`, `user_id`, `gift_user`, `createtime`) USING BTREE
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='用户送礼表_{$yearMonth}'
|
||||
";
|
||||
|
||||
try {
|
||||
Db::execute($sql);
|
||||
|
||||
// 计算时间范围
|
||||
$startTime = strtotime($yearMonth . '01 00:00:00');
|
||||
$endTime = strtotime(date('Y-m-01', strtotime('+1 month', $startTime)));
|
||||
|
||||
// 记录分表信息
|
||||
$tableInfo = Db::name('vs_give_gift_tables')
|
||||
->where('table_name', $tableName)
|
||||
->find();
|
||||
|
||||
if (!$tableInfo) {
|
||||
Db::name('vs_give_gift_tables')->insert([
|
||||
'table_name' => $tableName,
|
||||
'start_time' => $startTime,
|
||||
'end_time' => $endTime,
|
||||
'create_time' => time()
|
||||
]);
|
||||
}
|
||||
|
||||
Log::info("创建分表成功: {$tableName}");
|
||||
return true;
|
||||
} catch (\Exception $e) {
|
||||
Log::error("创建分表失败 {$tableName}: " . $e->getMessage());
|
||||
throw new Exception('创建分表失败：' . $e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量创建分表
|
||||
* @param int $months 创建未来几个月
|
||||
* @return array 创建结果
|
||||
*/
|
||||
public static function batchCreateTables($months = 12)
|
||||
{
|
||||
$results = [];
|
||||
|
||||
for ($i = 0; $i < $months; $i++) {
|
||||
$yearMonth = date('Ym', strtotime("+{$i} month"));
|
||||
|
||||
try {
|
||||
$result = self::createMonthTable($yearMonth);
|
||||
$results[$yearMonth] = [
|
||||
'success' => $result,
|
||||
'message' => $result ? '创建成功' : '创建失败'
|
||||
];
|
||||
} catch (\Exception $e) {
|
||||
$results[$yearMonth] = [
|
||||
'success' => false,
|
||||
'message' => $e->getMessage()
|
||||
];
|
||||
}
|
||||
}
|
||||
|
||||
return $results;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取所有分表
|
||||
* @return array
|
||||
*/
|
||||
public static function getAllTables()
|
||||
{
|
||||
try {
|
||||
return Db::name('vs_give_gift_tables')
|
||||
->order('start_time', 'asc')
|
||||
->select();
|
||||
} catch (\Exception $e) {
|
||||
Log::error("获取分表列表失败: " . $e->getMessage());
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据时间范围获取需要查询的表
|
||||
* @param int $startTime 开始时间戳
|
||||
* @param int $endTime 结束时间戳
|
||||
* @return array 表名数组
|
||||
*/
|
||||
public static function getTablesByTimeRange($startTime = null, $endTime = null)
|
||||
{
|
||||
try {
|
||||
$query = Db::name('vs_give_gift_tables');
|
||||
|
||||
if ($startTime !== null) {
|
||||
$query->where('end_time', '>=', $startTime);
|
||||
}
|
||||
|
||||
if ($endTime !== null) {
|
||||
$query->where('start_time', '<=', $endTime);
|
||||
}
|
||||
|
||||
$tables = $query->order('start_time', 'asc')
|
||||
->column('table_name');
|
||||
|
||||
return $tables ?: [];
|
||||
} catch (\Exception $e) {
|
||||
Log::error("获取时间范围分表失败: " . $e->getMessage());
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化分表系统
|
||||
* @param string $startMonth 起始年月 202401
|
||||
* @param int $months 创建几个月
|
||||
* @return array
|
||||
*/
|
||||
public static function initTables($startMonth = null, $months = 24)
|
||||
{
|
||||
if (!$startMonth) {
|
||||
$startMonth = date('Ym');
|
||||
}
|
||||
|
||||
$results = [];
|
||||
$startDate = new \DateTime($startMonth . '01');
|
||||
|
||||
for ($i = 0; $i < $months; $i++) {
|
||||
$currentDate = clone $startDate;
|
||||
$currentDate->modify("+{$i} month");
|
||||
$yearMonth = $currentDate->format('Ym');
|
||||
|
||||
try {
|
||||
$result = self::createMonthTable($yearMonth);
|
||||
$results[$yearMonth] = $result;
|
||||
} catch (\Exception $e) {
|
||||
$results[$yearMonth] = false;
|
||||
}
|
||||
}
|
||||
|
||||
return $results;
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查并创建当前月份的表
|
||||
* @return bool
|
||||
*/
|
||||
public static function checkAndCreateCurrentMonthTable()
|
||||
{
|
||||
$currentMonth = date('Ym');
|
||||
return self::createMonthTable($currentMonth);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user