Files
tnb.server/visualdev/Tnb.VisualDev/ScreenDataSourceService.cs
2023-03-13 15:00:34 +08:00

246 lines
8.6 KiB
C#

using JNPF.Common.Enums;
using JNPF.Common.Extension;
using JNPF.DependencyInjection;
using JNPF.DynamicApiController;
using JNPF.FriendlyException;
using JNPF.VisualData.Entity;
using JNPF.VisualData.Entitys.Dto.ScreenDataSource;
using Mapster;
using Microsoft.AspNetCore.Mvc;
using SqlSugar;
using Yitter.IdGenerator;
namespace JNPF.VisualData;
/// <summary>
/// 业务实现:大屏.
/// </summary>
[ApiDescriptionSettings(Tag = "BladeVisual", Name = "db", Order = 160)]
[Route("api/blade-visual/[controller]")]
public class ScreenDataSourceService : IDynamicApiController, ITransient
{
/// <summary>
/// 客户端初始化.
/// </summary>
public ISqlSugarClient _sqlSugarClient;
/// <summary>
/// 服务基础仓储.
/// </summary>
private readonly ISqlSugarRepository<VisualDBEntity> _visualDBRepository;
/// <summary>
/// 初始化一个<see cref="ScreenDataSourceService"/>类型的新实例.
/// </summary>
public ScreenDataSourceService(
ISqlSugarRepository<VisualDBEntity> visualDBRepository,
ISqlSugarClient context)
{
_visualDBRepository = visualDBRepository;
_sqlSugarClient = context;
}
#region Get
/// <summary>
/// 分页.
/// </summary>
/// <returns></returns>
[HttpGet("list")]
public async Task<dynamic> GetList([FromQuery] ScreenDataSourceListQueryInput input)
{
SqlSugarPagedList<ScreenDataSourceListOutput>? data = await _visualDBRepository.AsQueryable().Where(v => v.IsDeleted == 0)
.Select(v => new ScreenDataSourceListOutput { id = v.Id, name = v.Name, driverClass = v.DriverClass, password = v.Password, remark = v.Remark, url = v.Url, username = v.UserName, isDeleted = v.IsDeleted })
.ToPagedListAsync(input.current, input.size);
return new { current = data.pagination.CurrentPage, pages = data.pagination.Total / data.pagination.PageSize, records = data.list, size = data.pagination.PageSize, total = data.pagination.Total };
}
/// <summary>
/// 详情.
/// </summary>
/// <returns></returns>
[HttpGet("detail")]
public async Task<dynamic> GetInfo(string id)
{
VisualDBEntity? entity = await _visualDBRepository.AsQueryable().FirstAsync(v => v.Id == id && v.IsDeleted == 0);
return entity.Adapt<ScreenDataSourceInfoOutput>();
}
/// <summary>
/// 下拉数据源.
/// </summary>
/// <returns></returns>
[HttpGet("db-list")]
public async Task<dynamic> GetDBList()
{
return await _visualDBRepository.AsQueryable().Select(v => new ScreenDataSourceSeleectOutput { id = v.Id, name = v.Name, driverClass = v.DriverClass }).ToListAsync();
}
#endregion
#region Post
/// <summary>
/// 新增与修改.
/// </summary>
/// <returns></returns>
[HttpPost("submit")]
public async Task Submit([FromBody] ScreenDataSourceUpInput input)
{
VisualDBEntity? entity = input.Adapt<VisualDBEntity>();
int isOk = 0;
if (input.id == null)
isOk = await _visualDBRepository.AsInsertable(entity).IgnoreColumns(ignoreNullColumn: true).CallEntityMethod(m => m.Create()).ExecuteCommandAsync();
else
isOk = await _visualDBRepository.AsUpdateable(entity).IgnoreColumns(ignoreAllNullColumns: true).CallEntityMethod(m => m.LastModify()).ExecuteCommandAsync();
if (!(isOk > 0)) throw Oops.Oh(ErrorCode.COM1000);
}
/// <summary>
/// 新增.
/// </summary>
/// <param name="input"></param>
/// <returns></returns>
[HttpPost("save")]
public async Task Create([FromBody] ScreenDataSourceCrInput input)
{
VisualDBEntity? entity = input.Adapt<VisualDBEntity>();
int isOk = await _visualDBRepository.AsInsertable(entity).IgnoreColumns(ignoreNullColumn: true).CallEntityMethod(m => m.Create()).ExecuteCommandAsync();
if (!(isOk > 0)) throw Oops.Oh(ErrorCode.COM1000);
}
/// <summary>
/// 修改.
/// </summary>
/// <param name="input"></param>
/// <returns></returns>
[HttpPost("update")]
public async Task Update([FromBody] ScreenDataSourceUpInput input)
{
VisualDBEntity? entity = input.Adapt<VisualDBEntity>();
int isOk = await _visualDBRepository.AsUpdateable(entity).IgnoreColumns(ignoreAllNullColumns: true).CallEntityMethod(m => m.LastModify()).ExecuteCommandAsync();
if (!(isOk > 0)) throw Oops.Oh(ErrorCode.COM1001);
}
/// <summary>
/// 删除.
/// </summary>
/// <returns></returns>
[HttpPost("remove")]
public async Task Delete(string ids)
{
VisualDBEntity? entity = await _visualDBRepository.AsQueryable().FirstAsync(v => v.Id == ids && v.IsDeleted == 0);
_ = entity ?? throw Oops.Oh(ErrorCode.COM1005);
int isOk = await _visualDBRepository.AsUpdateable(entity).CallEntityMethod(m => m.Delete()).UpdateColumns(it => new { it.IsDeleted }).ExecuteCommandAsync();
if (!(isOk > 0)) throw Oops.Oh(ErrorCode.COM1002);
}
/// <summary>
/// 数据源测试连接.
/// </summary>
/// <returns></returns>
[HttpPost("db-test")]
public dynamic Test([FromBody] ScreenDataSourceUpInput input)
{
if (input.id == null)
input.id = YitIdHelper.NextId().ToString();
_sqlSugarClient.AsTenant().AddConnection(new ConnectionConfig()
{
ConfigId = input.id,
DbType = ToDbTytpe(input.driverClass),
ConnectionString = ToConnectionString(input.driverClass, input.url, input.username, input.password),
InitKeyType = InitKeyType.Attribute,
IsAutoCloseConnection = true
});
_sqlSugarClient = _sqlSugarClient.AsTenant().GetConnectionScope(input.id);
if (_sqlSugarClient.Ado.IsValidConnection())
return Task.FromResult(true);
else
throw Oops.Oh(ErrorCode.D1507);
}
/// <summary>
/// 动态执行SQL.
/// </summary>
/// <returns></returns>
[HttpPost("dynamic-query")]
public async Task<dynamic> Query([FromBody] ScreenDataSourceDynamicQueryInput input)
{
if (input.sql.IsNullOrEmpty()) throw Oops.Oh(ErrorCode.D1513);
if (!string.IsNullOrWhiteSpace(input.id))
{
VisualDBEntity? entity = await _visualDBRepository.AsQueryable().FirstAsync(v => v.Id == input.id && v.IsDeleted == 0);
_sqlSugarClient.AsTenant().AddConnection(new ConnectionConfig()
{
ConfigId = input.id,
DbType = ToDbTytpe(entity.DriverClass),
ConnectionString = ToConnectionString(entity.DriverClass, entity.Url, entity.UserName, entity.Password),
InitKeyType = InitKeyType.Attribute,
IsAutoCloseConnection = true
});
_sqlSugarClient = _sqlSugarClient.AsTenant().GetConnectionScope(input.id);
try
{
return await _sqlSugarClient.Ado.GetDataTableAsync(input.sql);
}
catch (Exception ex)
{
throw Oops.Oh(ErrorCode.D1512, ex.Message);
}
}
return Task.FromResult(true);
}
#endregion
#region PrivateMethod
/// <summary>
/// 转换数据库类型.
/// </summary>
/// <param name="dbType"></param>
/// <returns></returns>
private SqlSugar.DbType ToDbTytpe(string dbType)
{
switch (dbType)
{
case "org.postgresql.Driver":
return SqlSugar.DbType.PostgreSQL;
case "com.mysql.cj.jdbc.Driver":
return SqlSugar.DbType.MySql;
case "oracle.jdbc.OracleDriver":
return SqlSugar.DbType.Oracle;
case "com.microsoft.sqlserver.jdbc.SQLServerDriver":
return SqlSugar.DbType.SqlServer;
default:
throw Oops.Oh(ErrorCode.D1505);
}
}
/// <summary>
/// 转换连接字符串.
/// </summary>
/// <returns></returns>
private string ToConnectionString(string driverClass, string url, string name, string password)
{
switch (driverClass)
{
case "org.postgresql.Driver":
return string.Format(url, name, password);
case "com.mysql.cj.jdbc.Driver":
return string.Format(url, name, password);
case "oracle.jdbc.OracleDriver":
return string.Format(url, name, password);
case "com.microsoft.sqlserver.jdbc.SQLServerDriver":
return string.Format(url, name, password);
default:
throw Oops.Oh(ErrorCode.D1505);
}
}
#endregion
}