update:重复创建也返回成功

创建时可直接返回二维码
This commit is contained in:
2021-03-04 10:52:30 +08:00
parent 5234d3358b
commit 65dcef0f64
3 changed files with 28 additions and 7 deletions

View File

@@ -35,7 +35,7 @@ namespace QRCodeService.Controllers.Api
}
[Route("i/{shortCode}/")]
[HttpGet]
public async Task<IActionResult> GetQrCode(string shortCode,int img=0)
public async Task<IActionResult> GetQrCodeImage(string shortCode)
{
var link = await linkQueries.GetLinkAsync(shortCode);
if (link == null)
@@ -51,7 +51,11 @@ namespace QRCodeService.Controllers.Api
return File(stream.ToArray(), "image/png");
}
}
/// <summary>
/// 创建二维码 如果请求头的Accept类型是image/png 则会返回二维码图片 否则返回二维码地址
/// </summary>
/// <param name="input"></param>
/// <returns></returns>
[CheckSign]
[HttpPost]
public async Task<IActionResult> Create(CreateLinkModel input)
@@ -62,7 +66,19 @@ namespace QRCodeService.Controllers.Api
{
return BadRequest();
}
return Created($"{option.Value.BaseUrl}r/{link.ShortCode}",link.ShortCode);
var accept = Request.Headers["Accept"];
if(accept == "image/png")
{
var qrCodeGenerator = new QRCodeGenerator();
var data = qrCodeGenerator.CreateQrCode($"{option.Value.BaseUrl}r/{link.ShortCode}", QRCodeGenerator.ECCLevel.Q);
var qrCode = new QRCode(data);
using (var stream = new MemoryStream())
{
qrCode.GetGraphic(10, Color.Black, Color.White, null, 15, 6, false).Save(stream, System.Drawing.Imaging.ImageFormat.Png);
return File(stream.ToArray(), "image/png");
}
}
return Ok($"{option.Value.BaseUrl}r/{link.ShortCode}");
}
}
}