AX.FireTrainingSys/FireTrainingSys/Controllers/Template/TemplatePeripheralWaterSourcesController.cs

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

namespace AX.FireTrainingSys.Controllers.V1
{
    /// <summary>
    /// 周边水源控制器。
    /// </summary>
    [Produces("application/json")]
    [Route("api/[controller]")]
    [ApiVersion("1.0")]
    [ApiController]
    public class TemplatePeripheralWaterSourcesController : ControllerBase
    {
        private readonly DriveDbContext dbContext;

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

        /// <summary>
        /// 获得指定单位的周边水源信息。
        /// </summary>
        /// <param name="name">指定的单位名称</param>
        /// <returns></returns>
        [ProducesResponseType(StatusCodes.Status200OK)]
        [ProducesResponseType(StatusCodes.Status400BadRequest)]
        [ProducesResponseType(StatusCodes.Status404NotFound)]
        [HttpGet]
        public async Task<ActionResult<string>> Get([FromQuery, BindRequired]string name)
        {
            if (string.IsNullOrEmpty(name))
                return BadRequest(name);

            var model = await dbContext.TemplatePeripheralWaterSources.FindAsync(name);

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

            return Ok(model.Content);
        }

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

            var model = await dbContext.TemplatePeripheralWaterSources.FindAsync(name);

            if (model == null)
            {
                model = new TemplatePeripheralWaterSource
                {
                    CompanyName = name,
                    Content = content
                };

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

            await dbContext.SaveChangesAsync();

            return Ok();
        }
    }
}