github/cool-admin-midway
1
0
Fork 0
mirror of https://github.com/cool-team-official/cool-admin-midway.git synced 2025-12-11 00:22:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

自动校验短信插件

Browse Source
This commit is contained in:
cool 2024-02-21 15:19:51 +08:00
parent 2a7958b7c6
commit ed72f546d4
1 changed files with 21 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/modules/user/service/sms.ts
View File

@ -1,4 +1,4 @@
import { Provide, Config, Inject } from '@midwayjs/decorator';
import { Provide, Config, Inject, Init } from '@midwayjs/decorator';
import { BaseService, CoolCommException } from '@cool-midway/core';
import * as _ from 'lodash';
import { CacheManager } from '@midwayjs/cache';
@ -19,6 +19,23 @@ export class UserSmsService extends BaseService {
@Inject()
pluginService: PluginService;
plugin;
@Init()
async init() {
for (const key of ['sms-tx', 'sms-ali']) {
try {
this.plugin = await this.pluginService.getInstance(key);
if (this.plugin) {
this.config.pluginKey = key;
break;
}
} catch (e) {
continue;
}
}
}
/**
* 发送验证码
* @param phone
@ -27,13 +44,13 @@ export class UserSmsService extends BaseService {
// 随机四位验证码
const code = _.random(1000, 9999);
const pluginKey = this.config.pluginKey;
if (!pluginKey) throw new CoolCommException('未配置短信插件');
if (!this.plugin) throw new CoolCommException('未配置短信插件');
try {
if (pluginKey == 'sms-tx') {
await this.pluginService.invoke('sms-tx', 'send', [code], [code]);
await this.plugin.send([phone], [code]);
}
if (pluginKey == 'sms-ali') {
await this.pluginService.invoke('sms-ali', 'send', [phone], {
await this.plugin.send([phone], {
code,
});
}