forked from ddrilling/AsbCloudServer
122 lines
4.6 KiB
C#
122 lines
4.6 KiB
C#
using AsbCloudApp.Data.GTR;
|
|
using AsbCloudApp.Repositories;
|
|
using AsbCloudApp.Services;
|
|
using AsbCloudWebApi.SignalR;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.AspNetCore.SignalR;
|
|
using System.Collections.Generic;
|
|
using System.ComponentModel.DataAnnotations;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using AsbCloudApp.Requests;
|
|
|
|
namespace AsbCloudWebApi.Controllers.SAUB
|
|
{
|
|
|
|
[Route("api/[controller]")]
|
|
[ApiController]
|
|
public class GtrWitsController : ControllerBase
|
|
|
|
{
|
|
protected readonly IWellService wellService;
|
|
private readonly ITelemetryService telemetryService;
|
|
private readonly IGtrRepository gtrRepository;
|
|
private readonly IHubContext<TelemetryHub> telemetryHubContext;
|
|
|
|
public string SignalRMethodGetDataName { get; protected set; } = "ReceiveData";
|
|
|
|
public GtrWitsController(
|
|
ITelemetryService telemetryService,
|
|
IGtrRepository gtrRepository,
|
|
IWellService wellService,
|
|
IHubContext<TelemetryHub> telemetryHubContext)
|
|
{
|
|
this.telemetryService = telemetryService;
|
|
this.gtrRepository = gtrRepository;
|
|
this.wellService = wellService;
|
|
this.telemetryHubContext = telemetryHubContext;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Получить загруженные данные ГТИ по скважине
|
|
/// </summary>
|
|
/// <param name="idWell">Id скважины</param>
|
|
/// <param name="request">Параметры запроса</param>
|
|
/// <param name = "token" >Токен завершения задачи</param>
|
|
/// <returns></returns>
|
|
[HttpGet("{idWell}")]
|
|
[Permission]
|
|
public async Task<ActionResult<IEnumerable<WitsRecordDto>>> GetDataAsync([Required] int idWell,
|
|
[FromQuery] GtrWithGetDataRequest request,
|
|
CancellationToken token)
|
|
{
|
|
int? idCompany = User.GetCompanyId();
|
|
|
|
if (idCompany is null)
|
|
return Forbid();
|
|
|
|
bool isCompanyOwnsWell = await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
|
|
idWell, token).ConfigureAwait(false);
|
|
|
|
if (!isCompanyOwnsWell)
|
|
return Forbid();
|
|
|
|
var content = await gtrRepository.GetAsync(idWell, request.Begin,
|
|
request.IntervalSec, request.ApproxPointsCount, token).ConfigureAwait(false);
|
|
|
|
return Ok(content);
|
|
}
|
|
|
|
/// <summary>
|
|
/// получение последних данных ГТИ по ключу record
|
|
/// </summary>
|
|
/// <param name="idWell">id скважины</param>
|
|
/// <param name="idRecord">id record</param>
|
|
/// <param name="token"></param>
|
|
/// <returns></returns>
|
|
[HttpGet("{idWell}/{idRecord}")]
|
|
[Permission]
|
|
public async Task<ActionResult<IEnumerable<WitsItemRecordDto>>> GetLastDataByRecordIdAsync(int idWell, int idRecord, CancellationToken token)
|
|
{
|
|
int? idCompany = User.GetCompanyId();
|
|
|
|
if (idCompany is null)
|
|
return Forbid();
|
|
|
|
bool isCompanyOwnsWell = await wellService.IsCompanyInvolvedInWellAsync((int)idCompany,
|
|
idWell, token).ConfigureAwait(false);
|
|
|
|
if (!isCompanyOwnsWell)
|
|
return Forbid();
|
|
|
|
var content = gtrRepository.GetLastDataByRecordId(idWell, idRecord);
|
|
|
|
return Ok(content);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Метод для получения WITS записи от панели оператора.
|
|
/// Сохраняет в БД.
|
|
/// </summary>
|
|
/// <param name="uid">уникальный идентификатор телеметрии</param>
|
|
/// <param name="dtos">WITS запись</param>
|
|
/// <param name="token"></param>
|
|
/// <returns></returns>
|
|
[HttpPost("{uid}")]
|
|
public async Task<IActionResult> PostDataAsync(
|
|
string uid,
|
|
[FromBody] IEnumerable<WitsRecordDto> dtos,
|
|
CancellationToken token)
|
|
{
|
|
var telemetry = telemetryService.GetOrCreateTelemetryByUid(uid);
|
|
await gtrRepository.SaveDataAsync(telemetry.Id, dtos, token).ConfigureAwait(false);
|
|
var idWell = telemetryService.GetIdWellByTelemetryUid(uid);
|
|
if (idWell is not null && dtos is not null)
|
|
_ = Task.Run(() => telemetryHubContext.Clients.Group($"well_{idWell}_gtr")
|
|
.SendAsync(SignalRMethodGetDataName, dtos), CancellationToken.None);
|
|
return Ok();
|
|
}
|
|
}
|
|
|
|
}
|