soul-yongping/scripts/auto-unbind-expired-simple.js

#!/usr/bin/env node
/**
 * 自动解绑定时任务 - 简化版（直接连接MySQL）
 * 
 * 功能：定时检查并解绑过期的推荐关系
 * 
 * 解绑条件：
 * 1. 绑定状态为 active
 * 2. 过期时间已到（expiry_date < NOW）
 * 3. 期间没有任何购买（purchase_count = 0）
 * 
 * 使用方式：
 * 1. 确保已安装 mysql2: npm install mysql2
 * 2. 配置环境变量或修改下方 DB_CONFIG
 * 3. 手动执行：node scripts/auto-unbind-expired-simple.js
 * 4. 宝塔定时任务：每天 02:00 执行
 */

const mysql = require('mysql2/promise')
require('dotenv').config()

// 数据库配置（从环境变量读取）
const DB_CONFIG = {
  host: process.env.DB_HOST || 'localhost',
  port: parseInt(process.env.DB_PORT) || 3306,
  user: process.env.DB_USER || 'root',
  password: process.env.DB_PASSWORD || '',
  database: process.env.DB_NAME || 'mycontent_db',
  charset: 'utf8mb4'
}

async function autoUnbind() {
  console.log('=' .repeat(60))
  console.log('自动解绑定时任务')
  console.log('执行时间:', new Date().toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai' }))
  console.log('=' .repeat(60))
  console.log()
  
  let connection
  
  try {
    // 连接数据库
    connection = await mysql.createConnection(DB_CONFIG)
    console.log(`✅ 已连接到数据库: ${DB_CONFIG.database}`)
    console.log()
    
    // 1. 查询需要解绑的记录
    console.log('步骤 1: 查询需要解绑的记录...')
    console.log('-' .repeat(60))
    
    const [expiredBindings] = await connection.execute(`
      SELECT 
        id, 
        referee_id, 
        referrer_id, 
        binding_date, 
        expiry_date,
        purchase_count,
        total_commission
      FROM referral_bindings
      WHERE status = 'active'
        AND expiry_date < NOW()
        AND purchase_count = 0
      ORDER BY expiry_date ASC
    `)
    
    if (expiredBindings.length === 0) {
      console.log('✅ 无需解绑的记录')
      console.log()
      console.log('=' .repeat(60))
      console.log('任务完成')
      console.log('=' .repeat(60))
      return
    }
    
    console.log(`找到 ${expiredBindings.length} 条需要解绑的记录`)
    console.log()
    
    // 2. 输出明细
    console.log('步骤 2: 解绑明细')
    console.log('-' .repeat(60))
    
    expiredBindings.forEach((binding, index) => {
      const bindingDate = new Date(binding.binding_date).toLocaleDateString('zh-CN')
      const expiryDate = new Date(binding.expiry_date).toLocaleDateString('zh-CN')
      const daysExpired = Math.floor((Date.now() - new Date(binding.expiry_date).getTime()) / (1000 * 60 * 60 * 24))
      
      console.log(`${index + 1}. 用户 ${binding.referee_id}`)
      console.log(`   推荐人: ${binding.referrer_id}`)
      console.log(`   绑定时间: ${bindingDate}`)
      console.log(`   过期时间: ${expiryDate} (已过期 ${daysExpired} 天)`)
      console.log(`   购买次数: ${binding.purchase_count}`)
      console.log(`   累计佣金: ¥${(binding.total_commission || 0).toFixed(2)}`)
      console.log()
    })
    
    // 3. 批量更新为 expired
    console.log('步骤 3: 执行解绑操作...')
    console.log('-' .repeat(60))
    
    const ids = expiredBindings.map(b => b.id)
    const placeholders = ids.map(() => '?').join(',')
    
    const [result] = await connection.execute(
      `UPDATE referral_bindings SET status = 'expired' WHERE id IN (${placeholders})`,
      ids
    )
    
    console.log(`✅ 已成功解绑 ${result.affectedRows} 条记录`)
    console.log()
    
    // 4. 更新推荐人的推广数量
    console.log('步骤 4: 更新推荐人统计...')
    console.log('-' .repeat(60))
    
    const referrerCounts = {}
    expiredBindings.forEach(binding => {
      referrerCounts[binding.referrer_id] = (referrerCounts[binding.referrer_id] || 0) + 1
    })
    
    for (const [referrerId, count] of Object.entries(referrerCounts)) {
      await connection.execute(
        `UPDATE users SET referral_count = GREATEST(referral_count - ?, 0) WHERE id = ?`,
        [count, referrerId]
      )
      console.log(`  - ${referrerId}: -${count} 个绑定`)
    }
    
    console.log(`✅ 已更新 ${Object.keys(referrerCounts).length} 个推荐人的统计数据`)
    console.log()
    
    // 5. 总结
    console.log('=' .repeat(60))
    console.log('✅ 任务完成')
    console.log(`   - 解绑记录数: ${expiredBindings.length}`)
    console.log(`   - 受影响推荐人: ${Object.keys(referrerCounts).length}`)
    console.log('=' .repeat(60))
    
  } catch (error) {
    console.error('❌ 任务执行失败:', error.message)
    console.error(error.stack)
    throw error
  } finally {
    // 关闭数据库连接
    if (connection) {
      await connection.end()
      console.log('\n数据库连接已关闭')
    }
  }
}

// 主函数
async function main() {
  try {
    await autoUnbind()
    process.exit(0)
  } catch (error) {
    console.error('\n❌ 脚本执行失败')
    process.exit(1)
  }
}

// 运行
if (require.main === module) {
  main()
}

module.exports = { autoUnbind }