ky_zhangq/qrcodeService
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

创建签名验证自定义中间件 ,优化代码结构

Browse Source
This commit is contained in:
zhangqi
2021-02-26 16:40:51 +08:00
parent 7c3ce74a3b
commit d6617f47c0
8 changed files with 128 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
QRCodeService/Controllers/Api/LinkController.cs
View File

@@ -10,6 +10,7 @@ using System.Threading.Tasks;
using MediatR;
using QRCodeService.Application.Queries;
using QRCodeService.Application.Commands;
using QRCodeService.Infrastructure.Middlewares;
namespace QRCodeService.Controllers.Api
{
@@ -19,32 +20,28 @@ namespace QRCodeService.Controllers.Api
{
readonly IMediator mediator;
readonly ILinkQueries linkQueries;
readonly IAppQueries appQueries;
public LinkController(IMediator mediator, ILinkQueries queries, IAppQueries appQueries)
public LinkController(IMediator mediator, ILinkQueries queries)
{
this.mediator = mediator;
this.linkQueries = queries;
this.appQueries = appQueries;
}
[Route("{shortCode}")]
[HttpGet]
public async Task<IActionResult> Get(string shortCode)
{
var link = await linkQueries.GetLinkAsync(shortCode);
if (link == null)
{
return NotFound();
}
return Ok(link);
}
[HttpGet]
public IActionResult Get()
{
return Ok();
}
[CheckSign(typeof(CreateLinkModel))]
[HttpPost]
public async Task<IActionResult> Create(CreateLinkModel input)
{
var app = await appQueries.GetAppAsync(input.AppId);
if (app == null)
{
return BadRequest();
}
if (! await input.CheckValidAsync(app.Appkey))
{
return BadRequest();
}
var command = new CreateLinkCommand(input.SuffixUrl,1);
var link = await mediator.Send(command);
if (link==null)

12
QRCodeService/Controllers/ImageController.cs
View File

@@ -9,6 +9,7 @@ using System.IO;
using QRCodeService.Application.Queries;
using QRCodeService.Models;
using QRCodeService.Options;
using QRCodeService.Infrastructure.Middlewares;
namespace QRCodeService.Controllers
{
@@ -25,20 +26,11 @@ namespace QRCodeService.Controllers
this.appQueries = appQueries;
this.option = option;
}
[CheckSign(typeof(GetQRCodeModel))]
[Route("qrcode")]
[HttpPost]
public async Task<IActionResult> GetImage(GetQRCodeModel input)
{
var app = await appQueries.GetAppAsync(input.AppId);
if (app == null)
{
return BadRequest();
}
if (!await input.CheckValidAsync(app.Appkey))
{
return BadRequest();
}
var link = await linkQueries.GetLinkAsync(input.ShortCode);
if (link.AppId != input.AppId)
{