AX.FireTrainingSys/FireTrainingSys/Controllers/Template/TemplateFireWaterMonitorsController.cs

using AX.FireTrainingSys.Models;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.ModelBinding;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text.Json;
using System.Threading.Tasks;

namespace AX.FireTrainingSys.Controllers.V1.Template
{
    /// <summary>
    /// 消防水炮。
    /// </summary>
    [Produces("application/json")]
    [Route("api/[controller]")]
    [ApiVersion("1.0")]
    [ApiController]
    public class TemplateFireWaterMonitorsController : ControllerBase
    {
        private readonly DriveDbContext dbContext;

        public TemplateFireWaterMonitorsController(DriveDbContext dbContext)
        {
            this.dbContext = dbContext;
        }

        /// <summary>
        /// 获得指定单位的所有消防水炮信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <returns></returns>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [HttpGet]
        public async Task<ActionResult<IEnumerable<string>>> GetAll([FromQuery, BindRequired]string name)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            var models = await dbContext.TemplateFireWaterMonitors
                                        .AsNoTracking()
                                        .Where(e => e.CompanyName == name)
                                        .Select(e => e.Content)
                                        .ToListAsync();

            return Ok(models);
        }

        /// <summary>
        /// 获得指定单位的单条消防水炮信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <param name="id">指定的相关编号</param>
        /// <returns></returns>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [ProducesResponseType(StatusCodes.Status404NotFound)]
        [HttpGet("{id}")]
        public async Task<ActionResult<string>> Get([FromQuery, BindRequired]string name, string id)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            var model = await dbContext.TemplateFireWaterMonitors.FindAsync(name, id);

            if (model == null)
                return NotFound(name);

            return Ok(model.Content);
        }

        /// <summary>
        /// 创建或更新指定单位的单条消防水炮信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <param name="id">指定的相关编号</param>
        /// <param name="content">要保存的内容</param>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [HttpPost("{id}")]
        public async Task<Microsoft.AspNetCore.Mvc.ActionResult> PostOrPut([FromQuery, BindRequired]string name, string id, [FromBody]string content)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            var model = await dbContext.TemplateFireWaterMonitors.FindAsync(name, id);

            if (model == null)
            {
                model = new TemplateFireWaterMonitor
                {
                    CompanyName = name,
                    Id = id,
                    Content = content
                };

                dbContext.TemplateFireWaterMonitors.Add(model);
            }
            else
            {
                model.Content = content;
            }

            await dbContext.SaveChangesAsync();

            return Ok();
        }

        /// <summary>
        /// 批量创建或更新指定单位的消防水炮信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <param name="content">要保存的内容</param>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [HttpPost]
        public async Task<Microsoft.AspNetCore.Mvc.ActionResult> PostOrPutAll([FromQuery, BindRequired]string name, [FromBody]string content)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            if (string.IsNullOrEmpty(content))
                return BadRequest(content);

            var data = JsonSerializer.Deserialize<IDictionary<string, string>>(content);

            if (data == null)
                return BadRequest(content);

            using (var transaction = dbContext.Database.BeginTransaction())
            {
                //首先删除所有该单位的相关信息
                var olds = await dbContext
                                .TemplateFireWaterMonitors
                                .Where(e => e.CompanyName == name)
                                .ToListAsync();

                dbContext.TemplateFireWaterMonitors.RemoveRange(olds);

                //依次插入新信息
                foreach (var item in data)
                {
                    var model = new TemplateFireWaterMonitor
                    {
                        CompanyName = name,
                        Id = item.Key,
                        Content = item.Value
                    };

                    dbContext.TemplateFireWaterMonitors.Add(model);
                }

                await dbContext.SaveChangesAsync();

                transaction.Commit();
            }

            return Ok();
        }

        /// <summary>
        /// 批量删除指定单位的消防水炮信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <param name="content">要删除的内容</param>
        /// <returns></returns>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [ProducesResponseType(StatusCodes.Status404NotFound)]
        [HttpDelete]
        public async Task<Microsoft.AspNetCore.Mvc.ActionResult> DeleteAll([FromQuery, BindRequired]string name, [FromBody]string content)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            if (string.IsNullOrEmpty(content))
                return BadRequest(content);

            var data = JsonSerializer.Deserialize<string[]>(content);

            if (data == null)
                return BadRequest(content);

            using (var transaction = dbContext.Database.BeginTransaction())
            {
                await foreach (var (id, model) in AsAsyncEnumerable(name, data))
                {
                    if (model == null)
                        return NotFound(id);

                    dbContext.TemplateFireWaterMonitors.Remove(model);
                }

                await dbContext.SaveChangesAsync();

                transaction.Commit();
            }

            return NoContent();
        }

        /// <summary>
        /// 删除指定单位的单条消防水炮信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <param name="id">指定的相关编号</param>
        /// <returns></returns>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [ProducesResponseType(StatusCodes.Status404NotFound)]
        [HttpDelete("{id}")]
        public async Task<Microsoft.AspNetCore.Mvc.ActionResult> Delete([FromQuery, BindRequired]string name, string id)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            var model = await dbContext.TemplateFireWaterMonitors.FindAsync(name, id);

            if (model == null)
                return NotFound(id);

            dbContext.TemplateFireWaterMonitors.Remove(model);

            await dbContext.SaveChangesAsync();

            return NoContent();
        }

        private async IAsyncEnumerable<(string, TemplateFireWaterMonitor)> AsAsyncEnumerable(string name, string[] data)
        {
            foreach (var item in data)
            {
                var model = await dbContext.TemplateFireWaterMonitors.FindAsync(name, item);
                yield return (item, model);
            }
        }
    }
}