[hw-5] concurrency, graceful shutdown, concurrent tests

This commit is contained in:
Никита Шубин
2025-07-06 20:52:27 +00:00
parent dbf8aaedcf
commit 84201fe495
23 changed files with 742 additions and 157 deletions

View File

@@ -1,37 +1,16 @@
[
{
"sku": 139275865,
"total_count": 65534,
"reserved": 0
},
{
"sku": 2956315,
"total_count": 100,
"reserved": 30
},
{
"sku": 1076963,
"total_count": 100,
"reserved": 35
},
{
"sku": 135717466,
"total_count": 100,
"reserved": 20
},
{
"sku": 135937324,
"total_count": 100,
"reserved": 30
},
{
"sku": 1625903,
"total_count": 10000,
"reserved": 0
},
{
"sku": 1148162,
"total_count": 100,
"reserved": 0
}
{"sku": 139275865,"total_count": 65534,"reserved": 0},
{"sku": 2956315,"total_count": 100,"reserved": 30},
{"sku": 1076963,"total_count": 100,"reserved": 35},
{"sku": 135717466,"total_count": 100,"reserved": 20},
{"sku": 135937324,"total_count": 100,"reserved": 30},
{"sku": 1625903,"total_count": 10000,"reserved": 0},
{"sku": 1148162,"total_count": 100,"reserved": 0},
{"sku": 2958025,"total_count": 100,"reserved": 0},
{"sku": 3596599,"total_count": 100,"reserved": 0},
{"sku": 3618852,"total_count": 100,"reserved": 0},
{"sku": 4288068,"total_count": 100,"reserved": 0},
{"sku": 4465995,"total_count": 100,"reserved": 0},
{"sku": 30816475,"total_count": 100,"reserved": 0},
{"sku": 2618151,"total_count": 100,"reserved": 0}
]

View File

@@ -27,7 +27,7 @@ func NewStockRepository(write, read *pgxpool.Pool) service.StockRepository {
}
}
func (s *stockRepo) GetQuerier(ctx context.Context) *Queries {
func (s *stockRepo) GetWriter(ctx context.Context) *Queries {
tx, ok := postgres.TxFromCtx(ctx)
if ok {
return New(tx)
@@ -36,8 +36,17 @@ func (s *stockRepo) GetQuerier(ctx context.Context) *Queries {
return New(s.write)
}
func (s *stockRepo) GetReader(ctx context.Context) *Queries {
tx, ok := postgres.TxFromCtx(ctx)
if ok {
return New(tx)
}
return New(s.read)
}
func (s *stockRepo) StockReserve(ctx context.Context, stock *entity.Stock) error {
querier := s.GetQuerier(ctx)
querier := s.GetWriter(ctx)
rows, err := querier.StockReserve(ctx, &StockReserveParams{
Sku: int64(stock.Item.ID),
@@ -55,7 +64,7 @@ func (s *stockRepo) StockReserve(ctx context.Context, stock *entity.Stock) error
}
func (s *stockRepo) StockReserveRemove(ctx context.Context, stock *entity.Stock) error {
querier := s.GetQuerier(ctx)
querier := s.GetWriter(ctx)
rows, err := querier.StockReserveRemove(ctx, &StockReserveRemoveParams{
Sku: int64(stock.Item.ID),
@@ -81,7 +90,7 @@ func (s *stockRepo) StockReserveRemove(ctx context.Context, stock *entity.Stock)
}
func (s *stockRepo) StockCancel(ctx context.Context, stock *entity.Stock) error {
querier := s.GetQuerier(ctx)
querier := s.GetWriter(ctx)
rows, err := querier.StockCancel(ctx, &StockCancelParams{
Sku: int64(stock.Item.ID),
@@ -107,7 +116,7 @@ func (s *stockRepo) StockCancel(ctx context.Context, stock *entity.Stock) error
}
func (s *stockRepo) StockGetByID(ctx context.Context, sku entity.Sku) (*entity.Stock, error) {
querier := s.GetQuerier(ctx)
querier := s.GetReader(ctx)
stock, err := querier.StockGetByID(ctx, int64(sku))
switch {