| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289 |
- const path = require('path')
- const axios = require('axios')
- const OSS = require('ali-oss')
- const mq = require('../../plugin/mq')
- const { mq: mqName } = require('../../plugin/mq/mqPrefix')
- const { assertRunforgeTaskIngress, TASK_QUEUE } = require('../../plugin/mq/runforgeTaskMq')
- const db = require('../../plugin/DataBase/db')
- const Redis = require('../../plugin/DataBase/Redis')
- const EmailTemplate = require('../../plugin/Email/emailTemplate')
- const { URLSearchParams } = require('url')
- const {
- getPathData,
- selectCheckpoints,
- generateCadence
- } = require('../../plugin/Lepao/Path')
- const {
- dataEncrypt,
- dataDecrypt,
- dataSign
- } = require('../../plugin/Lepao/Crypto')
- const generateGyrFromPath = require('../../plugin/Lepao/generateGyrFromPath')
- const Logger = require('../Logger')
- class Worker {
- constructor() {
- this.logger = new Logger(
- path.join(__dirname, '../logs/LepaoWorker.log'),
- 'INFO'
- )
- this.handlers = {}
- this.running = false
- this.baseUrl = 'https://lepao.ctbu.edu.cn/v3/api.php'
- this.taskQueue = TASK_QUEUE
- this.resultQueue = mqName('runforge_task_result_queue')
- this.deadQueue = mqName('runforge_task_dead_queue')
- this.noticeQueue = mqName('runforge_message_queue')
- this.channelName = 'lepao_worker'
- this.maxRetry = 3
- this.timeout = 15000
- this.maxQueueLength = 2000
- this.defaultUserAgent = 'Mozilla/5.0 (Linux; Android 16; 2211133C Build/BP2A.250605.031.A3; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/138.0.7204.180 Mobile Safari/537.36 XWEB/1380347 MMWEBSDK/20250202 MMWEBID/1020 wxwork/5.0.6.66174 MicroMessenger/8.0.28.48(0x28001c30) MiniProgramEnv/android Luggage/3.0.2.95ef3f83 NetType/WIFI Language/zh_CN ABI/arm64'
- // 调试模式:将 axios 请求走本地代理(例如 charles/fiddler)
- // 开启方式:设置环境变量 LEPAO_DEBUG_PROXY=1
- this.debugProxyEnabled = String(process.env.LEPAO_DEBUG_PROXY || '').trim() === '1'
- this.debugProxyHost = process.env.LEPAO_DEBUG_PROXY_HOST || '127.0.0.1'
- this.debugProxyPort = Number(process.env.LEPAO_DEBUG_PROXY_PORT || 9000)
- }
- /* ================= 工具 ================= */
- api(path) {
- return this.baseUrl + path
- }
- traceId() {
- return Date.now() + '_' + Math.random().toString(36).slice(2, 8)
- }
- sleep(ms) {
- return new Promise(r => setTimeout(r, ms))
- }
- isRunSuccess(bindResponse) {
- const payload = bindResponse?.data
- if (!bindResponse || bindResponse.status !== 1 || !payload) {
- return {
- ok: false,
- reason: bindResponse?.info || '系统繁忙,请联系客服或稍后再试'
- }
- }
- const failedReason = payload.record_failed_reason || ''
- if (failedReason === '' || failedReason === '自动确认有效') {
- return { ok: true, payload }
- }
- return {
- ok: false,
- reason: failedReason,
- payload
- }
- }
- extractApiErrorMessage(name, result) {
- if (!result) {
- this.logger.error(`${name} 接口无响应数据: ${this.safeStringify(result)}`)
- return `系统繁忙,请联系客服或稍后再试`
- }
- const candidates = [
- result.info,
- result.msg,
- result.message,
- result?.data?.info,
- result?.data?.msg,
- result?.data?.message,
- result?.data?.record_failed_reason
- ]
- const reason = candidates.find(v => typeof v === 'string' && v.trim() !== '')
- if (reason) {
- return reason
- }
- if (result.code !== undefined || result.status !== undefined) {
- this.logger.error(`${name} 接口返回异常: ${this.safeStringify(result)}`)
- return `系统繁忙,请联系客服或稍后再试`
- }
- return `系统繁忙,请联系客服或稍后再试`
- }
- async markLoginExpired(account) {
- if (!account) return
- try {
- const sql = 'UPDATE lepao_account SET state = 0 WHERE student_num = ?'
- await db.query(sql, [account])
- this.logger.warn(`${account} 登录状态已失效,已自动更新为未登录`)
- } catch (error) {
- this.logger.error(`更新账号登录状态失败:${error.stack || error}`)
- }
- }
- async writeSuccessRedis(account) {
- if (!account) return
- try {
- const now = new Date()
- const tomorrow = new Date().setHours(24, 0, 0, 0)
- const exp = Math.floor((tomorrow - now) / 1000)
- await Redis.set(`lepaoSuccess:${account}`, account, { EX: exp })
- } catch (error) {
- this.logger.error(`写入乐跑成功缓存失败: ${error.stack || error}`)
- }
- }
- async addLepaoRecord(uuid, account, result, pathId, pointData) {
- if (!uuid || !account || !result || !pathId) return
- try {
- const time = Date.now()
- const sql = 'INSERT INTO lepao_record (uuid, time, lepao_account, result, path_id, point_data, state) VALUES (?, ?, ?, ?, ?, ?, ?)'
- await db.query(sql, [uuid, time, account, result, pathId, JSON.stringify(pointData || []), 1])
- } catch (error) {
- this.logger.error(`写入乐跑记录失败: ${error.stack || error}`)
- }
- }
- async syncRunCount(req, ctx) {
- try {
- const sid = req?.student_id || req?.account
- if (
- req?.uid == null ||
- req?.token == null ||
- String(req.token).trim() === '' ||
- req?.school_id == null ||
- !sid
- ) {
- return { ok: false, reason: '缺少同步乐跑次数所需凭证' }
- }
- const recordData = await this.handlers['lepao.getRecord'](req, ctx)
- const data = recordData?.data
- if (!data) {
- return { ok: false, reason: 'getRecord 无有效 data' }
- }
- const term_num = Number(data.term_num ?? 30)
- const total_num = Number(data.total_num ?? 0)
- const sql = 'UPDATE lepao_account SET term_num = ?, total_num = ? WHERE student_num = ?'
- const rows = await db.query(sql, [term_num, total_num, req.account])
- if (!rows || rows.affectedRows !== 1) {
- this.logger.warn(`${req.account}更新乐跑次数失败`)
- return { ok: false, reason: '数据库更新 lepao_account 失败', term_num, total_num }
- }
- this.logger.info(`${req.account}更新乐跑次数成功 term_num=${term_num}, total_num=${total_num}`)
- return { ok: true, term_num, total_num }
- } catch (error) {
- this.logger.warn(`${req?.account || 'unknown'}同步乐跑次数失败: ${error.message || error}`)
- return { ok: false, reason: error.message || String(error) }
- }
- }
- lepaoTimestamp() {
- return Number((Date.now() / 1000).toFixed(3))
- }
- axiosProxyConfig() {
- if (!this.debugProxyEnabled) {
- return { proxy: false }
- }
- this.logger.info(`使用本地代理: ${this.debugProxyHost}:${this.debugProxyPort}`)
- return {
- proxy: {
- host: this.debugProxyHost,
- port: this.debugProxyPort,
- protocol: 'http'
- }
- }
- }
- async enqueueTask(channel, type, data, options = {}) {
- const payload = {
- id: options.id || this.traceId(),
- type,
- data,
- retry: options.retry ?? 0
- }
- await channel.sendToQueue(
- this.taskQueue,
- Buffer.from(JSON.stringify(payload)),
- { persistent: true, contentType: 'application/json' }
- )
- return payload.id
- }
- async withTimeout(promise, name) {
- return Promise.race([
- promise,
- new Promise((_, reject) =>
- setTimeout(() => reject(new Error(`${name} 超时`)), this.timeout)
- )
- ])
- }
- async retry(fn, name) {
- let lastErr
- for (let i = 0; i < this.maxRetry; i++) {
- try {
- return await fn()
- } catch (err) {
- lastErr = err
- if (!this.isRetryableTaskError(err)) {
- throw err
- }
- this.logger.warn(`[RETRY] ${name} 第${i + 1}次失败`)
- await this.sleep(1000 * (i + 1)) // 指数退避
- }
- }
- throw lastErr
- }
- isNetworkError(err) {
- if (!err) return false
- if (err.code && ['ECONNRESET', 'ECONNABORTED', 'ETIMEDOUT', 'ENOTFOUND', 'EAI_AGAIN'].includes(err.code)) {
- return true
- }
- if (err.isAxiosError && !err.response) return true
- const msg = (err.message || '').toLowerCase()
- return msg.includes('timeout') || msg.includes('network')
- }
- isRetryableTaskError(err) {
- if (!err) return false
- if (err.retryable === true) return true
- if (this.isNetworkError(err)) return true
- return ['PATH_SELECT_FAILED', 'CHECKPOINT_FETCH_FAILED', 'CHECKPOINT_INSUFFICIENT'].includes(err.code)
- }
- safeStringify(obj) {
- const seen = new WeakSet();
- return JSON.stringify(obj, (key, value) => {
- if (typeof value === 'object' && value !== null) {
- if (seen.has(value)) return '[Circular]';
- seen.add(value);
- }
- return value;
- })
- }
- log(traceId, type, msg, data) {
- this.logger.info(`[${traceId}] [${type}] ${msg} ${data ? this.safeStringify(data) : ''}`)
- }
- logErr(traceId, msg, err) {
- this.logger.error(`[${traceId}] ${msg} ${err.stack || err}`)
- }
- async request(traceId, name, url, raw, headers = {}) {
- return this.retry(async () => {
- this.log(traceId, 'REQ', name, raw)
- const mergedHeaders = {
- 'Content-Type': 'application/x-www-form-urlencoded',
- 'Accept': '*/*',
- 'Accept-Language': 'zh-CN,zh-Hans;q=0.9',
- 'Accept-Encoding': 'gzip, deflate, br',
- 'Referer': 'https://servicewechat.com/wxf94c4ddb63d87ede/32/page-frame.html',
- ...headers
- }
- if (!mergedHeaders['User-Agent']) {
- mergedHeaders['User-Agent'] = this.defaultUserAgent
- }
- const form = new URLSearchParams()
- form.append('ostype', '5')
- form.append('data', dataEncrypt(JSON.stringify(raw)))
- const res = await this.withTimeout(
- axios.post(
- url,
- form,
- {
- headers: mergedHeaders,
- ...this.axiosProxyConfig()
- }
- ),
- name
- )
- let result = res.data
- if (result?.data && result?.is_encrypt === 1) {
- result.data = JSON.parse(dataDecrypt(result.data))
- }
- this.log(traceId, 'RES', name, result)
- // 除 bindData 外,其余调用若接口已明确返回失败,直接抛出该失败原因
- // bindData 需要保留完整响应由 isRunSuccess 统一判定。
- if (name !== 'bindData') {
- const hasCode = result && Object.prototype.hasOwnProperty.call(result, 'code')
- const hasStatus = result && Object.prototype.hasOwnProperty.call(result, 'status')
- const failedByCode = hasCode && Number(result.code) !== 1 && Number(result.code) !== 200
- const failedByStatus = hasStatus && Number(result.status) !== 1
- if (failedByCode || failedByStatus) {
- const message = this.extractApiErrorMessage(name, result)
- const err = new Error(message)
- // 学习 Lepao.js:若明确提示重新登录,自动标记账号失效
- if (message.includes('重新登录')) {
- await this.markLoginExpired(raw?.student_num)
- }
- // 接口已返回业务错误,禁止重试
- err.retryable = false
- throw err
- }
- }
- return result
- }, name)
- }
- /**
- * 累计完成次数 >= 跑友目标(且目标>0)时:关闭 auto_run,并发送乐跑目标完成邮件 / Bot 通知
- */
- async handleLepaoTargetComplete(account, user, totalNum, traceId) {
- const target = Number(user?.target_count) || 0
- const total = Number(totalNum) || 0
- if (target <= 0 || total < target) return
- try {
- const up = await db.query(
- 'UPDATE lepao_account SET auto_run = 0 WHERE student_num = ? AND auto_run = 1',
- [account]
- )
- if (up?.affectedRows !== 1) {
- return
- }
- this.logger.info(
- `${account} 已达目标次数(${total}/${target}),关闭自动乐跑`
- )
- } catch (e) {
- this.logger.error(`关闭自动乐跑失败 ${account}: ${e.message || e}`)
- return
- }
- const noticeType = user.notice_type || 'none'
- const overPayload = {
- type: 'lepao_over',
- umo: user.bot_umo,
- name: user.name,
- account,
- total_num: total,
- target_count: target,
- traceId
- }
- if (noticeType === 'email' && user.email) {
- try {
- await EmailTemplate.lepaoOver(user.email, {
- name: user.name,
- account
- })
- } catch (e) {
- this.logger.error(`lepaoOver 邮件发送失败: ${e.message || e}`)
- }
- }
- if (noticeType === 'bot' && user.bot_umo) {
- try {
- const ch = await mq.getChannel(this.noticeQueue)
- await ch.assertQueue(this.noticeQueue, { durable: true })
- ch.sendToQueue(
- this.noticeQueue,
- Buffer.from(JSON.stringify(overPayload)),
- {
- persistent: true,
- contentType: 'application/json'
- }
- )
- } catch (e) {
- this.logger.error(`lepao_over Bot 通知失败: ${e.message || e}`)
- }
- }
- }
- register(type, handler) {
- this.handlers[type] = handler
- this.logger.info(`注册任务: ${type}`)
- }
- /* ================= 业务 ================= */
- initHandlers() {
- /* ---------------- 开始乐跑 ---------------- */
- this.register('lepao.startRun', async (req, ctx) => {
- const traceId = ctx.traceId
- const maxPathRetry = 20 // 自动获取路径失败最大重试次数
- let pathRetry = 0
- let pointData = null
- let pathData = null
- let newPathData = null
- let ossPath = null
- let ossSts = null
- let userData = null
- let pathId = null
- let runZoneId = 0
- let bindRes = null
- try {
- // 检查redis是否存在当天乐跑成功记录
- const isSuccess = await Redis.get(`lepaoSuccess:${req.account}`)
- if (isSuccess)
- throw new Error('该账号当天已乐跑成功!请勿重复乐跑')
- userData = await this.handlers['lepao.getUserData'](req, ctx)
- // 立刻合并账号凭证,保证后续任意 throw 时 finally 里 syncRunCount 不会用空 token 调 getRecord
- req = {
- ...req,
- ...userData,
- student_id: req.account
- }
- // 进入乐跑进程后写入进行中缓存
- const progressKey = `lepaoProgress:${req.account}`
- const inProgress = await Redis.get(progressKey)
- if (inProgress) {
- throw new Error('该账号已进入乐跑任务队列,请等待乐跑完成后再进行乐跑操作')
- }
- await Redis.set(progressKey, req.account, { EX: 1800 })
- ossSts = await this.handlers['lepao.getOssSts'](req, ctx)
- if (!ossSts?.bucket || !ossSts?.AccessKeyId || !ossSts?.AccessKeySecret || !ossSts?.SecurityToken) {
- throw new Error('获取 OSS 凭证失败,请联系客服或稍后再试')
- }
- req = {
- ...req,
- ossSts
- }
- // 晚上10点后提前
- let run_end_time = Math.floor(Date.now() / 1000) - 300 // 提前5分钟
- let hour = new Date().getHours()
- if (hour < 7)
- throw new Error('当前不在有效乐跑时间范围内。RunForge支持乐跑时间段为7:00~24:00')
- if (hour >= 22) {
- this.logger.info(`${req.account}当前时间为${hour}点,调整run_end_time提前5小时`)
- run_end_time -= 18000
- }
- req = {
- ...req,
- run_end_time
- }
- // 1.5️⃣ 乐跑开始前扣减次数(失败会返还,且有幂等保护)
- await this.handlers['lepao.consumeCount']({
- account: req.account,
- uuid: userData?.create_user
- }, ctx)
- while (pathRetry < maxPathRetry) {
- try {
- // 2️⃣ 获取路径(仅路径选择失败时重试)
- const pathRes = await this.handlers['lepao.getPath'](req, ctx)
- pathData = pathRes.pathData
- pathId = pathData?.id || null
- newPathData = getPathData(pathData.data, req.run_end_time, pathData.time)
- // 3️⃣ 切换跑区
- const zoneRes = await this.handlers['lepao.setZone']({ ...req, pathData }, ctx)
- runZoneId = zoneRes?.run_zone_id || 0
- // 4️⃣ 上传 OSS 文件、生成打卡点
- const uploadRes = await this.handlers['lepao.uploadOssFile']({ ...req, pathData, newPathData }, ctx)
- ossPath = uploadRes.oss_path
- pointData = uploadRes.point_data
- if (!pointData) {
- pathRetry++
- this.logger.warn(`[${traceId}] 打卡点不满足要求,重新获取路径 第${pathRetry}次`)
- continue
- }
- // 打卡点符合要求,跳出循环
- break
- } catch (err) {
- if (!this.isRetryableTaskError(err)) {
- throw err
- }
- this.logger.warn(`[${traceId}] 可重试错误,重新获取路径 第${pathRetry + 1}次,原因:${err.message}`)
- pathRetry++
- await this.sleep(1000 * pathRetry)
- }
- }
- if (!pointData) {
- throw new Error('打卡点获取失败,乐跑任务终止')
- }
- // 5️⃣ 提交跑步数据
- bindRes = await this.handlers['lepao.bindData']({
- ...req,
- pathData,
- run_zone_id: runZoneId,
- record_file: ossPath,
- point_data: pointData
- }, ctx)
- // 绑定接口有返回即入库
- if (bindRes && bindRes.data) {
- await this.addLepaoRecord(userData?.create_user, req.account, bindRes.data, pathId, pointData)
- }
- const runResult = this.isRunSuccess(bindRes)
- if (runResult.ok || runResult.reason === '当天关联成绩次数已达到上限') {
- await this.writeSuccessRedis(req.account)
- }
- if (!runResult.ok) {
- throw new Error(runResult.reason)
- }
- if (bindRes && bindRes.data && bindRes.data.record_id) {
- const gyrRes = await this.handlers['lepao.uploadGyrOssFile']({ ...req, newPathData, record_id: bindRes.data.record_id }, ctx)
- if (gyrRes?.status === 1) {
- this.logger.info(`${req.account}上传加速度数据成功!`)
- } else {
- this.logger.error(`${req.account}上传加速度数据失败!原因:${gyrRes.info || '未知错误'}`)
- }
- }
- // 同步乐跑次数(通知里要带 total_num / term_num,与 getRecord 一致)
- const syncResult = await this.syncRunCount(req, ctx)
- if (!syncResult?.ok) {
- this.logger.error(
- `[${traceId}] 同步乐跑次数失败:${syncResult?.reason || 'unknown'}`
- )
- }
- // 6️⃣ 发送通知(把同步后的学期次数、累计完成次数传给 Bot / 邮件)
- if (ctx.channel) {
- await this.enqueueTask(
- ctx.channel,
- 'lepao.sendNotice',
- {
- account: req.account,
- success: true,
- data: runResult.payload,
- traceId,
- total_num: syncResult?.ok ? syncResult.total_num : undefined
- },
- { id: `${traceId}:notice:success` }
- )
- }
- return { traceId, ossPath, pointData, bindRes }
- } catch (err) {
- this.logger.error(`[${traceId}] 乐跑流程失败:`, err)
- // 若已扣减次数,则失败时返还(幂等)
- try {
- await this.handlers['lepao.refundCount']({
- account: req.account,
- uuid: userData?.create_user
- }, ctx)
- } catch (e) {
- this.logger.error(`[${traceId}] 返还乐跑次数失败:${e.stack || e}`)
- }
- if (ctx.channel) {
- await this.enqueueTask(ctx.channel, 'lepao.sendNotice', {
- account: req.account,
- success: false,
- reason: err.message || '未知错误',
- traceId
- }, { id: `${traceId}:notice:fail` })
- }
- throw err
- } finally {
- await Redis.del(`lepaoProgress:${req.account}`)
- }
- })
- /* ---------------- 发送通知(独立 MQ 任务) ---------------- */
- this.register('lepao.sendNotice', async (req, ctx) => {
- const { account, success, data, reason, traceId, total_num: totalNumArg } = req || {}
- if (!account) {
- throw new Error('发送通知失败:缺少 account')
- }
- const emailSql = `
- SELECT
- a.name,
- a.email,
- a.target_count,
- a.notice_type,
- e.bot_umo
- FROM
- lepao_account a
- LEFT JOIN
- lepao_extra e
- ON
- a.student_num = e.student_num
- WHERE
- a.student_num = ?
- `
- const rows = await db.query(emailSql, [account])
- if (!rows || rows.length === 0) {
- throw new Error('发送通知失败:未找到用户通知配置')
- }
- const user = rows[0]
- const noticeType = user.notice_type || 'none'
- let totalForNotice = totalNumArg
- if (
- success &&
- (totalForNotice === undefined || totalForNotice === null)
- ) {
- const accRows = await db.query(
- 'SELECT total_num FROM lepao_account WHERE student_num = ?',
- [account]
- )
- if (accRows && accRows[0]) {
- totalForNotice = accRows[0].total_num
- }
- }
- if (totalForNotice === undefined || totalForNotice === null) {
- totalForNotice = 0
- }
- totalForNotice = Number(totalForNotice)
- const targetCount = Number(user.target_count) || 0
- const payload = success ? {
- ...(data && typeof data === 'object' ? data : {}),
- type: 'lepao_success',
- umo: user.bot_umo,
- total_num: totalForNotice,
- target_count: targetCount,
- name: user.name,
- account,
- traceId
- } : {
- type: 'lepao_fail',
- umo: user.bot_umo,
- name: user.name,
- account,
- reason,
- traceId
- }
- const afterSuccessNotify = async () => {
- if (success) {
- await this.handleLepaoTargetComplete(account, user, totalForNotice, traceId)
- }
- }
- if (noticeType === 'bot' && user.bot_umo) {
- const ch = await mq.getChannel(this.noticeQueue)
- await ch.assertQueue(this.noticeQueue, { durable: true })
- ch.sendToQueue(
- this.noticeQueue,
- Buffer.from(JSON.stringify(payload)),
- {
- persistent: true,
- contentType: 'application/json'
- }
- )
- await afterSuccessNotify()
- return { delivered: true, via: 'bot' }
- }
- if (noticeType === 'email' && user.email) {
- if (success) {
- await EmailTemplate.lepaoSuccess(user.email, payload)
- await afterSuccessNotify()
- return { delivered: true, via: 'email' }
- }
- await EmailTemplate.lepaoFail(user.email, {
- name: user.name,
- account,
- reason: reason || '系统繁忙,请联系客服或稍后再试',
- traceId
- })
- return { delivered: true, via: 'email' }
- }
- await afterSuccessNotify()
- return { delivered: false, via: 'none' }
- })
- /* ---------------- 扣减次数 ---------------- */
- this.register('lepao.consumeCount', async (req, ctx) => {
- const account = req?.account
- const uuid = req?.uuid
- if (!uuid) {
- throw new Error('扣减乐跑次数失败:缺少 uuid')
- }
- // 幂等:同一 taskId 只扣一次
- const consumeKey = `lepao:consume:${ctx?.taskId || ctx?.traceId || account || uuid}`
- const existed = await Redis.get(consumeKey)
- if (existed) {
- return true
- }
- this.logger.info(`${account || uuid}开始扣减乐跑次数`)
- const useLepaoCountSql = 'UPDATE users SET lepao_count = lepao_count - 1 WHERE uuid = ?'
- const r = await db.query(useLepaoCountSql, [uuid])
- if (!r || r.affectedRows !== 1) {
- throw new Error('扣减乐跑次数失败:数据库更新失败')
- }
- this.logger.info(`${account || uuid}扣减乐跑次数完成`)
- await Redis.set(consumeKey, '1', { EX: 3600 })
- return true
- })
- /* ---------------- 返还次数(失败时执行) ---------------- */
- this.register('lepao.refundCount', async (req, ctx) => {
- const account = req?.account
- const uuid = req?.uuid
- if (!uuid) {
- return true
- }
- const baseKey = `${ctx?.taskId || ctx?.traceId || account || uuid}`
- const consumeKey = `lepao:consume:${baseKey}`
- const refundKey = `lepao:refund:${baseKey}`
- const consumed = await Redis.get(consumeKey)
- if (!consumed) {
- return true
- }
- const refunded = await Redis.get(refundKey)
- if (refunded) {
- return true
- }
- this.logger.info(`${account || uuid}开始返还乐跑次数`)
- const sql = 'UPDATE users SET lepao_count = lepao_count + 1 WHERE uuid = ?'
- const r = await db.query(sql, [uuid])
- if (!r || r.affectedRows !== 1) {
- throw new Error('返还乐跑次数失败:数据库更新失败')
- }
- this.logger.info(`${account || uuid}返还乐跑次数完成`)
- await Redis.set(refundKey, '1', { EX: 3600 })
- return true
- })
- this.register('lepao.getUserData', async (req, ctx) => {
- const account = req.account
- this.logger.info(`${account}开始获取用户数据`)
- const accountSql = `
- SELECT
- u.uuid,
- u.lepao_count,
- l.create_user,
- l.name,
- l.student_num,
- l.area,
- l.sex,
- l.state,
- l.token,
- l.uid,
- l.school_id,
- l.userAgent,
- l.deviceModel,
- l.notice_type,
- l.email,
- e.bot_account
- FROM
- lepao_account l
- LEFT JOIN
- users u
- ON
- l.create_user = u.uuid
- LEFT JOIN
- lepao_extra e
- ON
- l.student_num = e.student_num
- WHERE
- l.student_num = ?
- `
- const rows = await db.query(accountSql, [account])
- if (!rows || rows.length === 0) {
- this.logger.error(`${account}无法获取账号数据`)
- throw new Error('无法获取账号数据,请联系客服或稍后再试')
- }
- let userData = rows[0]
- if (!userData.create_user || !userData.uuid) {
- this.logger.warn(`${account}账号状态异常`)
- throw new Error('当前账号状态异常,请联系客服')
- }
- if (userData.state !== 1) {
- this.logger.warn(`${account}登录状态异常 state=${userData.state}`)
- throw new Error('乐跑账号登录已过期,请尝试使用登录器重新登录')
- }
- if (userData.lepao_count < 1) {
- this.logger.warn(`${account}乐跑次数不足`)
- throw new Error('用户乐跑次数不足,请购买乐跑次数后重试!')
- }
- if (!userData.userAgent)
- userData.userAgent = 'Mozilla/5.0 (Linux; Android 16; 2211133C Build/BP2A.250605.031.A3; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/138.0.7204.180 Mobile Safari/537.36 XWEB/1380347 MMWEBSDK/20250202 MMWEBID/1020 wxwork/5.0.6.66174 MicroMessenger/8.0.28.48(0x28001c30) MiniProgramEnv/android Luggage/3.0.2.95ef3f83 NetType/WIFI Language/zh_CN ABI/arm64'
- if (!userData.deviceModel)
- userData.deviceModel = '2211133C'
- return userData
- })
- this.register('lepao.getPath', async (req, ctx) => {
- const account = req.account
- this.logger.info(`${account}开始获取路径`)
- const accountSql = 'SELECT area, sex FROM lepao_account WHERE student_num = ?'
- const rows = await db.query(accountSql, [account])
- if (!rows || rows.length === 0) {
- this.logger.error(`${account}无法获取账号数据`)
- throw new Error('无法获取账号数据')
- }
- const { area, sex } = rows[0]
- let max = 4.00
- let min = 2.00
- if (sex === 2) {
- max = 2.00
- min = 1.60
- }
- this.logger.info(`${account}路径参数: area=${area ?? '随机'}, max_distance=${max}, min_distance=${min}`)
- let pathSql = 'SELECT * FROM path_data WHERE state = 1 AND distance < ? AND distance > ? '
- const pathParams = [max, min]
- if (area) {
- pathSql += ' AND run_zone_name = ?'
- pathParams.push(area)
- }
- pathSql += ' ORDER BY count ASC LIMIT 1'
- const paths = await db.query(pathSql, pathParams)
- if (!paths || paths.length === 0) {
- this.logger.error(`${account}未找到符合条件的路线`)
- const err = new Error('未找到符合条件的路线,请改变路径选择条件')
- err.code = 'PATH_SELECT_FAILED'
- err.retryable = true
- throw err
- }
- const randomPath = paths[0]
- const updateSql = 'UPDATE path_data SET count = count + 1 WHERE id = ?'
- await db.query(updateSql, [randomPath.id])
- this.logger.info(`${account}路径选中id=${randomPath.id},计数加1成功`)
- return { pathData: randomPath }
- })
- /* ---------------- 获取跑步记录 ---------------- */
- this.register('lepao.getRecord', async (req, ctx) => {
- const now = this.lepaoTimestamp()
- const raw = {
- uid: req.uid,
- token: req.token,
- school_id: req.school_id,
- term_id: 0,
- course_id: 0,
- class_id: 0,
- student_num: req.student_id,
- card_id: req.student_id,
- timestamp: now,
- version: 1,
- nonce: String(Math.floor(Math.random() * 900000 + 100000)),
- ostype: 5
- }
- raw.sign = dataSign(raw)
- return this.request(
- ctx.traceId,
- 'getRecord',
- this.api('/Run2/beforeRunV260'),
- raw,
- {
- 'User-Agent': req.userAgent,
- 'charset': 'utf-8',
- 'Referer': 'https://servicewechat.com/wxf94c4ddb63d87ede/32/page-frame.html',
- }
- )
- })
- /* ---------------- 切换跑区 ---------------- */
- this.register('lepao.setZone', async (req, ctx) => {
- const runZoneMap = {
- '兰花湖校区跑区': 2,
- '主校区北跑区': 3,
- '主校区南跑区': 5,
- '重庆工商大学茶园校区': 6
- }
- const { pathData } = req
- const runZoneId = runZoneMap[pathData.run_zone_name]
- if (!runZoneId) throw new Error('跑区不存在')
- const raw = {
- uid: req.uid,
- token: req.token,
- school_id: req.school_id,
- term_id: 0,
- course_id: 0,
- class_id: 0,
- student_num: req.student_id,
- card_id: req.student_id,
- timestamp: this.lepaoTimestamp(),
- version: 1,
- nonce: String(Math.floor(Math.random() * 900000 + 100000)),
- ostype: 5,
- run_zone_id: String(runZoneId)
- }
- raw.sign = dataSign(raw)
- await this.request(
- ctx.traceId,
- 'setZone',
- this.api('/Run/setRunZone'),
- raw
- )
- return { run_zone_id: runZoneId }
- })
- /* ---------------- 获取 OSS STS ---------------- */
- this.register('lepao.getOssSts', async (req, ctx) => {
- const raw = {
- uid: req.uid,
- token: req.token,
- school_id: req.school_id,
- term_id: 0,
- course_id: 0,
- class_id: 0,
- student_num: req.student_id,
- card_id: req.student_id,
- timestamp: this.lepaoTimestamp(),
- version: 1,
- nonce: String(Math.floor(Math.random() * 900000 + 100000)),
- ostype: 5
- }
- raw.sign = dataSign(raw)
- const res = await this.request(
- ctx.traceId,
- 'getOssSts',
- this.api('/WpIndex/getOssSts'),
- raw
- )
- return res.data
- })
- /* ---------------- 上传 OSS 文件 ---------------- */
- this.register('lepao.uploadOssFile', async (req, ctx) => {
- const { pathData, newPathData, ossSts: sts } = req
- // 处理跑步路径
- const pathResult = dataEncrypt(JSON.stringify(newPathData))
- // 获取跑步规则参数
- const runRule = await this.handlers['lepao.getRecord'](req, ctx)
- const ruleData = runRule?.data
- if (!ruleData?.run_line_info?.point_list || !ruleData?.time_rule_arr?.length) {
- const err = new Error('获取打卡点规则失败')
- err.code = 'CHECKPOINT_FETCH_FAILED'
- err.retryable = true
- throw err
- }
- const check_points = ruleData.run_line_info.point_list
- let min_log_num = ruleData.time_rule_arr[0]?.min_log_num || 4
- const point_update_distance = parseFloat(ruleData.run_line_info.point_update_distance || 0) * 1000
- const log_max_distance = Number(ruleData.run_line_info.log_max_distance || 0)
- // 生成打卡点
- const point_data = selectCheckpoints(newPathData, check_points, min_log_num, point_update_distance, log_max_distance, req.run_end_time, pathData.time)
- if (!point_data) {
- this.logger.warn(`[RETRY] 打卡点数量不足,重新更换路径`)
- const err = new Error('打卡点数量不足')
- err.code = 'CHECKPOINT_INSUFFICIENT'
- err.retryable = true
- throw err
- }
- const now = new Date()
- const yyyy = now.getFullYear()
- const mm = String(now.getMonth() + 1).padStart(2, '0')
- const dd = String(now.getDate()).padStart(2, '0')
- const formattedToday = `${yyyy}-${mm}-${dd}`
- const boundary = String(Date.now())
- const timestamp = String(Date.now())
- const ossPath = `Public/Upload/file/run_record/${boundary.slice(-3)}/${formattedToday}/${timestamp}-${Math.floor(Math.random() * 150)}.txt`
- const client = new OSS({
- bucket: sts.bucket,
- region: sts.region || 'oss-cn-hangzhou',
- accessKeyId: sts.AccessKeyId,
- accessKeySecret: sts.AccessKeySecret,
- stsToken: sts.SecurityToken,
- secure: true
- })
- await client.put(ossPath, Buffer.from(pathResult, 'utf-8'))
- return { oss_path: ossPath, point_data: point_data }
- })
- this.register('lepao.uploadGyrOssFile', async (req, ctx) => {
- const { newPathData, ossSts: sts, record_id } = req
- // 生成加速度数据
- const gyrData = generateGyrFromPath(newPathData)
- if (!Array.isArray(gyrData) || gyrData.length === 0) {
- this.logger.error('生成加速度数据失败')
- return { status: 0, info: '生成加速度数据失败' }
- }
- const now = new Date()
- const yyyy = now.getFullYear()
- const mm = String(now.getMonth() + 1).padStart(2, '0')
- const dd = String(now.getDate()).padStart(2, '0')
- const formattedToday = `${yyyy}-${mm}-${dd}`
- const boundary = String(Date.now())
- const timestamp = String(Date.now())
- const ossPath = `Public/Upload/file/run_gyroscope/${boundary.slice(-3)}/${formattedToday}/${timestamp}-${Math.floor(Math.random() * 150)}.txt`
- const client = new OSS({
- bucket: sts.bucket,
- region: sts.region || 'oss-cn-hangzhou',
- accessKeyId: sts.AccessKeyId,
- accessKeySecret: sts.AccessKeySecret,
- stsToken: sts.SecurityToken,
- secure: true
- })
- await client.put(ossPath, Buffer.from(JSON.stringify(gyrData), 'utf-8'))
- const data = {
- uid: req.uid,
- token: req.token,
- school_id: req.school_id,
- term_id: 0,
- course_id: 0,
- class_id: 0,
- student_num: req.student_id,
- card_id: req.student_id,
- timestamp: this.lepaoTimestamp(),
- version: 1,
- nonce: String(Math.floor(Math.random() * 900000 + 100000)),
- ostype: 5,
- record_id: record_id,
- gyroscope_file: ossPath
- }
- data.sign = dataSign(data)
- return this.request(
- ctx.traceId,
- 'bindData',
- this.api('/Run2/gyroscope'),
- data
- )
- })
- /* ---------------- 提交跑步数据 ---------------- */
- this.register('lepao.bindData', async (req, ctx) => {
- const { pathData } = req
- const distance = Number(Number(pathData.distance || 0).toFixed(2))
- const stepData = generateCadence(distance, pathData.time)
- const stepInfo = JSON.stringify({ interval: 60, list: stepData.cadence_list })
- let points = req.point_data.map(({ address, jingwei, ...rest }) => rest)
- points = JSON.stringify(points)
- const data = {
- uid: req.uid,
- token: req.token,
- school_id: req.school_id,
- term_id: 1,
- course_id: 0,
- class_id: 0,
- student_num: req.student_id,
- card_id: req.student_id,
- timestamp: this.lepaoTimestamp(),
- version: 1,
- nonce: String(Math.floor(Math.random() * 900000 + 100000)),
- ostype: 5,
- game_id: String(req.run_zone_id || 0),
- start_time: req.run_end_time - Number(pathData.time),
- end_time: req.run_end_time,
- distance,
- record_img: "",
- log_data: points,
- file_img: "",
- is_running_area_valid: 1,
- mobileDeviceId: 1,
- mobileModel: req.deviceModel,
- step_info: stepInfo,
- step_num: stepData.total_steps,
- used_time: pathData.time,
- mobileOsVersion: 1,
- record_file: req.record_file
- }
- data.sign = dataSign(data)
- return this.request(
- ctx.traceId,
- 'bindData',
- this.api('/Run/stopRunV278'),
- data
- )
- })
- }
- /* ================= Worker核心 ================= */
- async start() {
- if (this.running) return
- this.running = true
- this.logger.info('Worker 启动中...')
- try {
- this.initHandlers()
- const channel = await mq.getChannel(this.channelName)
- await channel.prefetch(5)
- await assertRunforgeTaskIngress(channel, this.logger)
- await channel.assertQueue(this.resultQueue, {
- durable: true,
- arguments: {
- 'x-max-length': this.maxQueueLength
- }
- })
- await channel.assertQueue(this.deadQueue, {
- durable: true,
- arguments: {
- 'x-max-length': this.maxQueueLength
- }
- })
- await channel.consume(this.taskQueue, async (msg) => {
- if (!msg) return
- let content
- try {
- content = JSON.parse(msg.content.toString())
- } catch {
- return channel.ack(msg)
- }
- const { id, type, data, retry = 0 } = content
- const traceId = this.traceId()
- const handler = this.handlers[type]
- if (!handler) {
- this.log(traceId, 'ERROR', '未知任务', { type })
- return channel.ack(msg)
- }
- try {
- const result = await this.withTimeout(
- handler(data, { traceId, channel, taskId: id }),
- type
- )
- await this.sendResult(channel, {
- id,
- success: true,
- result
- })
- this.log(traceId, 'DONE', `任务完成 ${type}`)
- channel.ack(msg)
- } catch (err) {
- this.logErr(traceId, `任务失败 ${type}`, err)
- if (retry < this.maxRetry && this.isRetryableTaskError(err)) {
- // 重试
- await channel.sendToQueue(
- this.taskQueue,
- Buffer.from(JSON.stringify({
- ...content,
- retry: retry + 1
- })),
- { persistent: true }
- )
- this.log(traceId, 'RETRY', `重试第${retry + 1}次`)
- } else {
- // 死信
- await channel.sendToQueue(
- this.deadQueue,
- Buffer.from(JSON.stringify(content)),
- { persistent: true }
- )
- this.log(traceId, 'DEAD', '进入死信队列')
- }
- await this.sendResult(channel, {
- id,
- success: false,
- error: err.message
- })
- channel.ack(msg)
- }
- })
- this.logger.info('RunForge Worker 启动成功')
- } catch (err) {
- this.logger.error('RunForge Worker 启动失败: ' + err.stack)
- }
- }
- async sendResult(channel, data) {
- channel.sendToQueue(
- this.resultQueue,
- Buffer.from(JSON.stringify(data)),
- { persistent: true }
- )
- }
- async stop() {
- this.running = false
- await mq.close()
- this.logger.info('RunForge Worker 已停止')
- }
- }
- module.exports = Worker
|