format + lint
This commit is contained in:
@@ -4,18 +4,15 @@ import { AuthGuard } from '@nestjs/passport';
|
||||
|
||||
@Controller('oss')
|
||||
export class OssController {
|
||||
constructor(private readonly ossService: OssService) {}
|
||||
|
||||
constructor(
|
||||
private readonly ossService: OssService,
|
||||
) { }
|
||||
|
||||
@UseGuards(AuthGuard('jwt'))
|
||||
@Get('sts')
|
||||
async getStsToken(@Request() req) {
|
||||
const { userId, sessionId } = req.user;
|
||||
return {
|
||||
...await this.ossService.getStsToken(`${userId}`),
|
||||
userId,
|
||||
}
|
||||
}
|
||||
@UseGuards(AuthGuard('jwt'))
|
||||
@Get('sts')
|
||||
async getStsToken(@Request() req) {
|
||||
const { userId } = req.user;
|
||||
return {
|
||||
...(await this.ossService.getStsToken(`${userId}`)),
|
||||
userId,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
@@ -4,6 +4,6 @@ import { OssController } from './oss.controller';
|
||||
|
||||
@Module({
|
||||
providers: [OssService],
|
||||
controllers: [OssController]
|
||||
controllers: [OssController],
|
||||
})
|
||||
export class OssModule {}
|
||||
|
||||
@@ -3,51 +3,51 @@ import { STS } from 'ali-oss';
|
||||
|
||||
@Injectable()
|
||||
export class OssService {
|
||||
private sts = new STS({
|
||||
accessKeyId: process.env.ALIYUN_ACCESS_KEY_ID,
|
||||
accessKeySecret: process.env.ALIYUN_ACCESS_KEY_SECRET,
|
||||
});
|
||||
|
||||
private sts = new STS({
|
||||
accessKeyId: process.env.ALIYUN_ACCESS_KEY_ID,
|
||||
accessKeySecret: process.env.ALIYUN_ACCESS_KEY_SECRET,
|
||||
});
|
||||
private stsCache: {
|
||||
[session: string]: {
|
||||
credentials: {
|
||||
AccessKeyId: string;
|
||||
AccessKeySecret: string;
|
||||
SecurityToken: string;
|
||||
Expiration: string;
|
||||
};
|
||||
expireTime: number; // 时间戳,单位为毫秒
|
||||
};
|
||||
} = {};
|
||||
|
||||
private stsCache: {
|
||||
[session: string]: {
|
||||
credentials: {
|
||||
AccessKeyId: string;
|
||||
AccessKeySecret: string;
|
||||
SecurityToken: string;
|
||||
Expiration: string;
|
||||
};
|
||||
expireTime: number; // 时间戳,单位为毫秒
|
||||
}
|
||||
} = {};
|
||||
|
||||
/** @todo 该方法存在缓存穿透问题,待优化 */
|
||||
async getStsToken(session: string) {
|
||||
if (this.stsCache[session]) {
|
||||
const cached = this.stsCache[session];
|
||||
// 检查缓存是否过期
|
||||
if (cached.expireTime > Date.now()) {
|
||||
return cached.credentials;
|
||||
} else {
|
||||
// 如果过期,删除缓存
|
||||
delete this.stsCache[session];
|
||||
}
|
||||
}
|
||||
|
||||
return this.sts.assumeRole(
|
||||
process.env.ALIYUN_OSS_STS_ROLE_ARN, ``, 3600, `${session}`,
|
||||
).then((res) => {
|
||||
// 缓存
|
||||
this.stsCache[session] = {
|
||||
credentials: res.credentials,
|
||||
expireTime: new Date(res.credentials.Expiration).getTime() - 5 * 60 * 1000, // 提前5分钟过期,
|
||||
};
|
||||
|
||||
return res.credentials;
|
||||
}).catch(err => {
|
||||
console.error('获取STS Token失败:', err);
|
||||
throw new Error('获取STS Token失败');
|
||||
})
|
||||
/** @todo 该方法存在缓存穿透问题,待优化 */
|
||||
async getStsToken(session: string) {
|
||||
if (this.stsCache[session]) {
|
||||
const cached = this.stsCache[session];
|
||||
// 检查缓存是否过期
|
||||
if (cached.expireTime > Date.now()) {
|
||||
return cached.credentials;
|
||||
} else {
|
||||
// 如果过期,删除缓存
|
||||
delete this.stsCache[session];
|
||||
}
|
||||
}
|
||||
|
||||
return this.sts
|
||||
.assumeRole(process.env.ALIYUN_OSS_STS_ROLE_ARN, ``, 3600, `${session}`)
|
||||
.then((res) => {
|
||||
// 缓存
|
||||
this.stsCache[session] = {
|
||||
credentials: res.credentials,
|
||||
expireTime:
|
||||
new Date(res.credentials.Expiration).getTime() - 5 * 60 * 1000, // 提前5分钟过期,
|
||||
};
|
||||
|
||||
return res.credentials;
|
||||
})
|
||||
.catch((err) => {
|
||||
console.error('获取STS Token失败:', err);
|
||||
throw new Error('获取STS Token失败');
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user