Jerry/crystelf-core
1
0
Fork 0
mirror of https://github.com/crystelf/crystelf-core.git synced 2025-12-05 10:31:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:增加get表情包端口

Browse Source
This commit is contained in:
Jerry 2025-09-09 13:41:34 +08:00
parent 797dd94717
commit 97d536766e
1 changed files with 41 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
src/modules/meme/meme.controller.ts
View File

@ -1,7 +1,9 @@
import { import {
Controller, Controller,
Post, Post,
Get,
Body, Body,
Query,
Res, Res,
HttpException, HttpException,
HttpStatus, HttpStatus,
@ -9,7 +11,7 @@ import {
Inject, Inject,
Ip, Ip,
} from '@nestjs/common'; } from '@nestjs/common';
import { ApiTags, ApiOperation, ApiBody, ApiProperty } from '@nestjs/swagger'; import { ApiTags, ApiOperation, ApiBody, ApiQuery } from '@nestjs/swagger';
import { MemeService } from './meme.service'; import { MemeService } from './meme.service';
import { Response } from 'express'; import { Response } from 'express';
import * as fs from 'fs'; import * as fs from 'fs';
@ -18,17 +20,8 @@ import { ToolsService } from '../../core/tools/tools.service';
import { RedisService } from '../../core/redis/redis.service'; import { RedisService } from '../../core/redis/redis.service';
class MemeRequestDto { class MemeRequestDto {
@ApiProperty({ description: '角色名称', example: 'zhenxun', required: false })
character?: string; character?: string;
@ApiProperty({ description: '状态', example: 'happy', required: false })
status?: string; status?: string;
@ApiProperty({
description: '可选访问令牌',
example: 'token',
required: false,
})
token?: string; token?: string;
} }
@ -46,13 +39,46 @@ export class MemeController {
private readonly redisService: RedisService, private readonly redisService: RedisService,
) {} ) {}
@Post('getRandom') @Post('get')
@ApiOperation({ summary: '获取随机表情包' }) @ApiOperation({ summary: '获取随机表情包' })
@ApiQuery({ name: 'character', required: false, description: '角色名称' })
@ApiQuery({ name: 'status', required: false, description: '状态' })
@ApiQuery({ name: 'token', required: false, description: '可选访问令牌' })
@ApiBody({ type: MemeRequestDto }) @ApiBody({ type: MemeRequestDto })
public async getRandomMeme( public async getRandomMemePost(
@Body() dto: MemeRequestDto, @Body() dto: MemeRequestDto,
@Res() res: Response, @Res() res: Response,
@Ip() ip: string, @Ip() ip: string,
) {
return this.handleMemeRequest(dto, res, ip, 'POST');
}
@Get()
@ApiOperation({ summary: '获取随机表情包' })
@ApiQuery({ name: 'character', required: false, description: '角色名称' })
@ApiQuery({ name: 'status', required: false, description: '状态' })
@ApiQuery({ name: 'token', required: false, description: '可选访问令牌' })
public async getRandomMemeGet(
@Query() query: MemeRequestDto,
@Res() res: Response,
@Ip() ip: string,
) {
return this.handleMemeRequest(query, res, ip, 'GET');
}
/**
* 处理请求
* @param dto
* @param res
* @param ip
* @param method
* @private
*/
private async handleMemeRequest(
dto: MemeRequestDto,
res: Response,
ip: string,
method: string,
) { ) {
try { try {
const realToken = dto.token; const realToken = dto.token;
@ -88,7 +114,7 @@ export class MemeController {
}); });
if (hasValidToken) { if (hasValidToken) {
this.logger.log(`有token入不限速: ${memePath}`); this.logger.log(`[${method}] 有token的入不限速 => ${memePath}`);
stream.pipe(res); stream.pipe(res);
} else { } else {
stream.on('data', async (chunk) => { stream.on('data', async (chunk) => {
@ -99,14 +125,14 @@ export class MemeController {
1, 1,
); );
if (total > 100 * 1024) { if (total > 100 * 1024) {
this.logger.warn(`IP ${ip} 超过速率限制,断开连接..`); this.logger.warn(`[${method}]${ip} 超过速率限制,断开连接..`);
stream.destroy(); stream.destroy();
res.end(); res.end();
} }
}); });
const throttle = new Throttle({ rate: 100 * 1024 }); const throttle = new Throttle({ rate: 100 * 1024 });
this.logger.log(`白嫖的入限速!(${ip}) => ${memePath}`); this.logger.log(`[${method}] 白嫖入限速! (${ip}) => ${memePath}`);
stream.pipe(throttle).pipe(res); stream.pipe(throttle).pipe(res);
} }
} catch (e) { } catch (e) {