hrms-bangkok/hrms-api-backend
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages 7 Wiki Activity Actions

feat: Change RAbbitMQ Conenction to Pool and Add k6 Test Script for max request per second.

Browse source
This commit is contained in:
Suphonchai Phoonsawat 2024-08-20 10:37:47 +07:00
parent 0bac3630ec
commit f5355728ab
7 changed files with 186 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

2
BMA.EHR.Leave/BMA.EHR.Leave.csproj
View file

@ -37,6 +37,7 @@
<PrivateAssets>all</PrivateAssets>
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
</PackageReference>
<PackageReference Include="Microsoft.Extensions.ObjectPool" Version="8.0.8" />
<PackageReference Include="Microsoft.IdentityModel.Logging" Version="6.31.0" />
<PackageReference Include="Microsoft.VisualStudio.Azure.Containers.Tools.Targets" Version="1.18.1" />
<PackageReference Include="runtime.osx.10.10-x64.CoreCompat.System.Drawing" Version="6.0.5.128" />
@ -49,6 +50,7 @@
<PackageReference Include="Serilog.Sinks.Elasticsearch" Version="9.0.3" />
<PackageReference Include="Swashbuckle.AspNetCore.Annotations" Version="6.5.0" />
<PackageReference Include="Swashbuckle.AspNetCore" Version="6.5.0" />
<PackageReference Include="System.Threading.Channels" Version="8.0.0" />
</ItemGroup>
<ItemGroup>

105
BMA.EHR.Leave/Controllers/LeaveController.cs
View file

@ -1,5 +1,4 @@
using Amazon.S3.Model;
using BMA.EHR.Application.Repositories;
using BMA.EHR.Application.Repositories;
using BMA.EHR.Application.Repositories.Commands;
using BMA.EHR.Application.Repositories.Leaves.LeaveRequests;
using BMA.EHR.Application.Repositories.Leaves.TimeAttendants;
@ -15,12 +14,10 @@ using BMA.EHR.Leave.Service.DTOs.CheckIn;
using BMA.EHR.Leave.Service.DTOs.DutyTime;
using BMA.EHR.Leave.Service.DTOs.LeaveRequest;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.ObjectPool;
using Newtonsoft.Json;
using Org.BouncyCastle.Ocsp;
using RabbitMQ.Client;
using Serilog;
using Swashbuckle.AspNetCore.Annotations;
using System.ComponentModel.DataAnnotations;
using System.Security.Claims;
@ -58,6 +55,10 @@ namespace BMA.EHR.Leave.Service.Controllers
private readonly string _bucketName = "check-in";
private readonly ObjectPool<IModel> _objectPool;
private readonly string _fakeCheckInQueue = "fake-bma-checkin-queue";
private readonly string _realCheckInQueue = "bma-checkin-queue";
#endregion
#region " Constuctor and Destructor "
@ -75,7 +76,8 @@ namespace BMA.EHR.Leave.Service.Controllers
AdditionalCheckRequestRepository additionalCheckRequestRepository,
UserCalendarRepository userCalendarRepository,
CommandRepository commandRepository,
LeaveRequestRepository leaveRequestRepository)
LeaveRequestRepository leaveRequestRepository,
ObjectPool<IModel> objectPool)
{
_dutyTimeRepository = dutyTimeRepository;
_context = context;
@ -92,6 +94,7 @@ namespace BMA.EHR.Leave.Service.Controllers
_commandRepository = commandRepository;
_leaveRequestRepository = leaveRequestRepository;
_objectPool = objectPool;
}
#endregion
@ -428,6 +431,7 @@ namespace BMA.EHR.Leave.Service.Controllers
[ProducesResponseType(StatusCodes.Status500InternalServerError)]
public async Task<ActionResult<ResponseObject>> CheckInAsync([FromForm] CheckTimeDto data)
{
// prepare data and convert request body and send to queue
var userId = UserId == null ? Guid.Empty : Guid.Parse(UserId);
var currentDate = DateTime.Now;
var checkFileBytes = new byte[0];
@ -458,29 +462,48 @@ namespace BMA.EHR.Leave.Service.Controllers
Token = AccessToken ?? ""
};
// create connection
var factory = new ConnectionFactory()
var channel = _objectPool.Get();
try
{
HostName = _configuration["Rabbit:Host"],
UserName = _configuration["Rabbit:User"],
Password = _configuration["Rabbit:Password"],
};
channel.QueueDeclare(queue: _realCheckInQueue, durable: true, exclusive: false, autoDelete: false, arguments: null);
var serializedObject = JsonConvert.SerializeObject(checkData);
var body = Encoding.UTF8.GetBytes(serializedObject);
// create channel
using var connection = factory.CreateConnection();
using var channel = connection.CreateModel();
channel.QueueDeclare(queue: "checkin-queue", durable: false, exclusive: false, autoDelete: false, arguments: null);
channel.BasicPublish(exchange: "",
routingKey: _realCheckInQueue,
basicProperties: null,
body: body);
// แปลง Object เป็น JSON สตริง
var serializedObject = JsonConvert.SerializeObject(checkData);
return Success(new { date = currentDate });
}
finally
{
_objectPool.Return(channel);
}
// แปลง JSON สตริงเป็น byte array
var body = Encoding.UTF8.GetBytes(serializedObject);
//// create connection
//var factory = new ConnectionFactory()
//{
// HostName = _configuration["Rabbit:Host"],
// UserName = _configuration["Rabbit:User"],
// Password = _configuration["Rabbit:Password"],
//};
channel.BasicPublish(exchange: "", routingKey: "checkin-queue", basicProperties: null, body: body);
Console.WriteLine($"Send to Queue: {serializedObject}");
//// create channel
//using var connection = factory.CreateConnection();
//using var channel = connection.CreateModel();
//channel.QueueDeclare(queue: "checkin-queue", durable: false, exclusive: false, autoDelete: false, arguments: null);
return Success(new { date = currentDate });
//// แปลง Object เป็น JSON สตริง
//var serializedObject = JsonConvert.SerializeObject(checkData);
//// แปลง JSON สตริงเป็น byte array
//var body = Encoding.UTF8.GetBytes(serializedObject);
//channel.BasicPublish(exchange: "", routingKey: "checkin-queue", basicProperties: null, body: body);
//Console.WriteLine($"Send to Queue: {serializedObject}");
//return Success(new { date = currentDate });
}
/// <summary>
@ -491,7 +514,7 @@ namespace BMA.EHR.Leave.Service.Controllers
/// <response code="200">เมื่อทำรายการสำเร็จ</response>
/// <response code="401">ไม่ได้ Login เข้าระบบ</response>
/// <response code="500">เมื่อเกิดข้อผิดพลาดในการทำงาน</response>
[HttpPost("check-in"), DisableRequestSizeLimit]
[HttpPost("fake-check-in"), DisableRequestSizeLimit]
[ProducesResponseType(StatusCodes.Status200OK)]
[ProducesResponseType(StatusCodes.Status401Unauthorized)]
[ProducesResponseType(StatusCodes.Status500InternalServerError)]
@ -499,30 +522,26 @@ namespace BMA.EHR.Leave.Service.Controllers
public ActionResult<ResponseObject> FakeCheckIn([FromBody] FakeCheckTimeDto data)
{
var currentDate = DateTime.Now;
// create connection
var factory = new ConnectionFactory()
var channel = _objectPool.Get();
try
{
HostName = _configuration["Rabbit:Host"],
UserName = _configuration["Rabbit:User"],
Password = _configuration["Rabbit:Password"],
};
channel.QueueDeclare(queue: _fakeCheckInQueue, durable: true, exclusive: false, autoDelete: false, arguments: null);
// create channel
using var connection = factory.CreateConnection();
using var channel = connection.CreateModel();
channel.QueueDeclare(queue: "fake-checkin-queue", durable: false, exclusive: false, autoDelete: false, arguments: null);
var serializedObject = JsonConvert.SerializeObject(data);
// แปลง Object เป็น JSON สตริง
var serializedObject = JsonConvert.SerializeObject(data);
var body = Encoding.UTF8.GetBytes(serializedObject);
// แปลง JSON สตริงเป็น byte array
var body = Encoding.UTF8.GetBytes(serializedObject);
channel.BasicPublish(exchange: "",
routingKey: _fakeCheckInQueue,
basicProperties: null,
body: body);
channel.BasicPublish(exchange: "", routingKey: "fake-checkin-queue", basicProperties: null, body: body);
Console.WriteLine($"Send to Queue: {serializedObject}");
return Success(new { date = currentDate });
return Success(new { date = currentDate });
}
finally
{
_objectPool.Return(channel);
}
}
/// <summary>

20
BMA.EHR.Leave/Extensions/RabbitMqServiceCollectionExtensions.cs Normal file
View file

@ -0,0 +1,20 @@
using BMA.EHR.Leave.Service.Services;
using Microsoft.Extensions.ObjectPool;
using RabbitMQ.Client;
namespace BMA.EHR.Leave.Service.Extensions
{
public static class RabbitMqServiceCollectionExtensions
{
public static IServiceCollection AddRabbitMqConnectionPooling(this IServiceCollection services, IConfiguration configuration)
{
services.AddSingleton<ObjectPoolProvider, DefaultObjectPoolProvider>();
services.AddSingleton<ObjectPool<IModel>>(sp =>
{
var provider = sp.GetRequiredService<ObjectPoolProvider>();
return provider.Create(new RabbitMqConnectionPoolPolicy(configuration));
});
return services;
}
}
}

3
BMA.EHR.Leave/Program.cs
View file

@ -21,6 +21,7 @@ using System.Transactions;
using BMA.EHR.Leave.Service.Filters;
using Hangfire.Common;
using BMA.EHR.Application.Repositories.Leaves.TimeAttendants;
using BMA.EHR.Leave.Service.Extensions;
var builder = WebApplication.CreateBuilder(args);
{
@ -100,6 +101,8 @@ var builder = WebApplication.CreateBuilder(args);
builder.Services.AddHealthChecks();
builder.Services.AddRabbitMqConnectionPooling(builder.Configuration);
// Add Hangfire services.
var defaultConnection = builder.Configuration.GetConnectionString("DefaultConnection");

53
BMA.EHR.Leave/Services/RabbitMqConnectionPoolPolicy.cs Normal file
View file

@ -0,0 +1,53 @@
using Microsoft.Extensions.ObjectPool;
using RabbitMQ.Client;
namespace BMA.EHR.Leave.Service.Services;
public class RabbitMqConnectionPoolPolicy : IPooledObjectPolicy<IModel>
{
#region " Fields "
private readonly IConnection _connection;
private readonly IConfiguration _configuration;
#endregion
#region " Constructor and Destructor "
public RabbitMqConnectionPoolPolicy(IConfiguration configuration)
{
_configuration = configuration;
var factory = new ConnectionFactory()
{
HostName = _configuration["Rabbit:Host"],
UserName = _configuration["Rabbit:User"],
Password = _configuration["Rabbit:Password"]
};
_connection = factory.CreateConnection();
}
#endregion
#region " Methods "
public IModel Create()
{
return _connection.CreateModel();
}
public bool Return(IModel obj)
{
if (obj.IsOpen)
{
return true;
}
else
{
obj.Dispose();
return false;
}
}
#endregion
}

7
BMA.EHR.Solution.sln
View file

@ -31,7 +31,12 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "BMA.EHR.Insignia", "BMA.EHR
EndProject
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "BMA.EHR.Leave", "BMA.EHR.Leave\BMA.EHR.Leave.csproj", "{B06C5612-2346-44B1-9D50-F8373885BD88}"
EndProject
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "BMA.EHR.CheckInConsumer", "BMA.EHR.CheckInConsumer\BMA.EHR.CheckInConsumer.csproj", "{B36F7E5A-2745-42B5-8493-D648EF8CC43C}"
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "BMA.EHR.CheckInConsumer", "BMA.EHR.CheckInConsumer\BMA.EHR.CheckInConsumer.csproj", "{B36F7E5A-2745-42B5-8493-D648EF8CC43C}"
EndProject
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "k6_scripts", "k6_scripts", "{367D82BE-0AF9-4739-A45D-C88194E7820C}"
ProjectSection(SolutionItems) = preProject
checkin_load_test.js = checkin_load_test.js
EndProjectSection
EndProject
Global
GlobalSection(SolutionConfigurationPlatforms) = preSolution

40
checkin_load_test.js Normal file
View file

@ -0,0 +1,40 @@
import http from 'k6/http';
import { check, sleep } from 'k6';
export let options = {
scenarios: {
constant_rate: {
executor: 'constant-arrival-rate',
rate: 1000, // จำกัดที่ 1000 requests ต่อวินาที
timeUnit: '1s',
duration: '5m', // ใช้เวลารันทดสอบ 5 นาที
preAllocatedVUs: 50,
},
},
};
export default function () {
// ข้อมูล payload ที่จะส่งไปยัง server
let payload = JSON.stringify({
lat: 1,
lon: 1,
poi: 'FPT',
isLocation: true
});
// ตัวเลือก headers
let headers = {
'Content-Type': 'application/json',
};
// ส่ง POST request
let response = http.post('https://localhost:7283/api/v1/leave/fake-check-in', payload, { headers: headers });
// ตรวจสอบการตอบสนอง
check(response, {
'is status 200': (r) => r.status === 200,
});
// หน่วงเวลา 1 วินาที
sleep(1);
}