PlannedTrajectory fix

This commit is contained in:
ngfrolov 2023-01-12 12:07:31 +05:00
parent 2b258b8e74
commit 7c6e00fe51
Signed by: ng.frolov
GPG Key ID: E99907A0357B29A7
3 changed files with 26 additions and 27 deletions

View File

@ -21,7 +21,7 @@ namespace AsbCloudInfrastructure.Services.PlannedTrajectory
private readonly IPlannedTrajectoryService plannedTrajectoryService; private readonly IPlannedTrajectoryService plannedTrajectoryService;
private const string templateFileName = "PlannedTrajectoryTemplate.xlsx"; private const string templateFileName = "PlannedTrajectoryTemplate.xlsx";
private const string usingTemplateFile = "AsbCloudInfrastructure.Services.PlannedTrajectoryService"; private const string usingTemplateFile = "AsbCloudInfrastructure.Services.PlannedTrajectory";
private const string sheetNamePlannedTrajectory = "Плановая траектория"; private const string sheetNamePlannedTrajectory = "Плановая траектория";
private const int headerRowsCount = 2; private const int headerRowsCount = 2;
private const int ColumnWellboreDepth = 1; private const int ColumnWellboreDepth = 1;
@ -123,7 +123,10 @@ namespace AsbCloudInfrastructure.Services.PlannedTrajectory
using var workbook = new XLWorkbook(stream, XLEventTracking.Disabled); using var workbook = new XLWorkbook(stream, XLEventTracking.Disabled);
var trajectoryRows = ParseFileStream(stream); var trajectoryRows = ParseFileStream(stream);
foreach (var row in trajectoryRows) foreach (var row in trajectoryRows)
row.IdWell = idWell; {
row.IdWell = idWell;
row.IdUser = idUser;
}
var rowsCount = await SavePlannedTrajectoryAsync(idWell,trajectoryRows, deletePrevRows, token); var rowsCount = await SavePlannedTrajectoryAsync(idWell,trajectoryRows, deletePrevRows, token);
return rowsCount; return rowsCount;
@ -132,9 +135,7 @@ namespace AsbCloudInfrastructure.Services.PlannedTrajectory
private async Task<int> SavePlannedTrajectoryAsync(int idWell, IEnumerable<PlannedTrajectoryDto> newRows, bool deletePrevRow, CancellationToken token) private async Task<int> SavePlannedTrajectoryAsync(int idWell, IEnumerable<PlannedTrajectoryDto> newRows, bool deletePrevRow, CancellationToken token)
{ {
if (deletePrevRow) if (deletePrevRow)
{
await plannedTrajectoryService.DeleteByIdWellAsync(idWell, token); await plannedTrajectoryService.DeleteByIdWellAsync(idWell, token);
}
var rowsCount = await plannedTrajectoryService.AddRangeAsync(newRows, token); var rowsCount = await plannedTrajectoryService.AddRangeAsync(newRows, token);
return rowsCount; return rowsCount;
} }

View File

@ -29,13 +29,13 @@ namespace AsbCloudInfrastructure.Services.PlannedTrajectory
if (!plannedTrajectoryRows.All(r => r.IdWell == idWell)) if (!plannedTrajectoryRows.All(r => r.IdWell == idWell))
throw new ArgumentInvalidException("Все строки должны относиться к одной скважине", nameof(plannedTrajectoryRows)); throw new ArgumentInvalidException("Все строки должны относиться к одной скважине", nameof(plannedTrajectoryRows));
var offsetHours = wellService.GetTimezone(idWell).Hours; var offsetHours = wellService.GetTimezone(idWell).Hours;
var entitys = plannedTrajectoryRows var entities = plannedTrajectoryRows
.Select(e => Convert(e, offsetHours)); .Select(e => {
foreach(var item in entitys) var entity = Convert(e, offsetHours);
{ entity.Id = 0;
item.Id = 0; return entity;});
}
db.PlannedTrajectories.AddRange(entitys); db.PlannedTrajectories.AddRange(entities);
return await db.SaveChangesAsync(token) return await db.SaveChangesAsync(token)
.ConfigureAwait(false); .ConfigureAwait(false);
} }
@ -82,7 +82,7 @@ namespace AsbCloudInfrastructure.Services.PlannedTrajectory
.AsNoTracking() .AsNoTracking()
.Where(x => x.IdWell == idWell); .Where(x => x.IdWell == idWell);
var entities = await query var entities = await query
.OrderBy(e => e.UpdateDate) .OrderBy(e => e.WellboreDepth)
.ToListAsync(token); .ToListAsync(token);
var result = entities var result = entities
.Select(r => Convert(r, offsetHours)); .Select(r => Convert(r, offsetHours));

View File

@ -61,7 +61,7 @@ namespace AsbCloudWebApi.Controllers
token).ConfigureAwait(false)) token).ConfigureAwait(false))
return Forbid(); return Forbid();
var stream = await plannedTrajectoryImportService.ExportAsync(idWell, token); var stream = await plannedTrajectoryImportService.ExportAsync(idWell, token);
var fileName = plannedTrajectoryImportService.GetFileNameAsync(idWell, token).Result.ToString(); var fileName = await plannedTrajectoryImportService.GetFileNameAsync(idWell, token);
return File(stream, "application/octet-stream", fileName); return File(stream, "application/octet-stream", fileName);
} }
@ -75,12 +75,12 @@ namespace AsbCloudWebApi.Controllers
/// <returns>количество успешно записанных строк в БД</returns> /// <returns>количество успешно записанных строк в БД</returns>
[HttpPost] [HttpPost]
[Permission] [Permission]
[Route("import/{options}")] [Route("import/{deleteBeforeImport}")]
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
public async Task<IActionResult> ImportAsync(int idWell, public async Task<IActionResult> ImportAsync(int idWell,
[FromForm] IFormFileCollection files, [FromForm] IFormFileCollection files,
int options = 0, bool deleteBeforeImport,
CancellationToken token = default) CancellationToken token)
{ {
int? idUser = User.GetUserId(); int? idUser = User.GetUserId();
if (!idUser.HasValue) if (!idUser.HasValue)
@ -97,7 +97,7 @@ namespace AsbCloudWebApi.Controllers
try try
{ {
var result = plannedTrajectoryImportService.ImportAsync(idWell, idUser.Value, stream, (options & 1) > 0, token); var result = await plannedTrajectoryImportService.ImportAsync(idWell, idUser.Value, stream, deleteBeforeImport, token);
return Ok(result); return Ok(result);
} }
catch (FileFormatException ex) catch (FileFormatException ex)
@ -121,7 +121,7 @@ namespace AsbCloudWebApi.Controllers
if (!await CanUserAccessToWellAsync(idWell, if (!await CanUserAccessToWellAsync(idWell,
token).ConfigureAwait(false)) token).ConfigureAwait(false))
return Forbid(); return Forbid();
var result = plannedTrajectoryService.GetAsync(idWell, token); var result = await plannedTrajectoryService.GetAsync(idWell, token);
return Ok(result); return Ok(result);
} }
@ -146,8 +146,7 @@ namespace AsbCloudWebApi.Controllers
return Forbid(); return Forbid();
row.IdUser = idUser.Value; row.IdUser = idUser.Value;
row.IdWell = idWell; row.IdWell = idWell;
var result = await plannedTrajectoryService.AddAsync(row, token) var result = await plannedTrajectoryService.AddAsync(row, token);
.ConfigureAwait(false);
return Ok(result); return Ok(result);
} }
@ -175,8 +174,7 @@ namespace AsbCloudWebApi.Controllers
item.IdUser = idUser.Value; item.IdUser = idUser.Value;
item.IdWell = idWell; item.IdWell = idWell;
} }
var result = await plannedTrajectoryService.AddRangeAsync(rows, token) var result = await plannedTrajectoryService.AddRangeAsync(rows, token);
.ConfigureAwait(false);
return Ok(result); return Ok(result);
} }
@ -184,13 +182,14 @@ namespace AsbCloudWebApi.Controllers
/// Изменить выбранную строку с координатами /// Изменить выбранную строку с координатами
/// </summary> /// </summary>
/// <param name="idWell"></param> /// <param name="idWell"></param>
/// <param name="idRow"></param>
/// <param name="row"></param> /// <param name="row"></param>
/// <param name="token"></param> /// <param name="token"></param>
/// <returns>количество успешно обновленных строк в БД</returns> /// <returns>количество успешно обновленных строк в БД</returns>
[HttpPut("{idRow}")] [HttpPut("{idRow}")]
[Permission] [Permission]
[ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)] [ProducesResponseType(typeof(int), (int)System.Net.HttpStatusCode.OK)]
public async Task<IActionResult> UpdateAsync(int idWell, public async Task<IActionResult> UpdateAsync(int idWell, int idRow,
[FromBody] PlannedTrajectoryDto row, CancellationToken token = default) [FromBody] PlannedTrajectoryDto row, CancellationToken token = default)
{ {
if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false)) if (!await CanUserAccessToWellAsync(idWell, token).ConfigureAwait(false))
@ -198,10 +197,10 @@ namespace AsbCloudWebApi.Controllers
int? idUser = User.GetUserId(); int? idUser = User.GetUserId();
if (!idUser.HasValue) if (!idUser.HasValue)
return Forbid(); return Forbid();
row.Id = idRow;
row.IdUser = idUser.Value; row.IdUser = idUser.Value;
row.IdWell = idWell; row.IdWell = idWell;
var result = await plannedTrajectoryService.UpdateAsync(row, token) var result = await plannedTrajectoryService.UpdateAsync(row, token);
.ConfigureAwait(false);
return Ok(result); return Ok(result);
} }
@ -221,8 +220,7 @@ namespace AsbCloudWebApi.Controllers
token).ConfigureAwait(false)) token).ConfigureAwait(false))
return Forbid(); return Forbid();
var result = await plannedTrajectoryService.DeleteRangeAsync(new int[] { idRow }, token) var result = await plannedTrajectoryService.DeleteRangeAsync(new int[] { idRow }, token);
.ConfigureAwait(false);
return Ok(result); return Ok(result);
} }