重构代码质量和安全性改进

啊鑫

3 天以前 36fdf1c876e5b9daa4446b9c33e9c532d725798b

 MES.Service/service/MessageCenterManager.cs

@@ -8,17 +8,45 @@
{
    //当前类已经继承了 Repository 增、删、查、改的方法

    public List<MessageCenter> GetPushFailedPage(MessageCenter query)
    public (List<MessageCenter> item, int TotalCount) GetPushFailedPage(
        MessageCenter query)
    {
        var totalCount = IsShow(query);

        // 使用SqlSugar的安全查询方法，避免SQL注入风险
        var messageCenters = Db.Queryable<MessageCenter>()
            .Where(it => it.Result == 0 && it.Seq == 1 && it.Title != null)
            .OrderByDescending(it => it.CreateDate)
            .ToPageList(query.PageIndex, query.Limit, ref totalCount);

        return (messageCenters, totalCount);
    }

    //查看是否有未处理的消息
    public int IsShow(MessageCenter query)
    {
        return Context.Queryable<MessageCenter>()
            .Where(it =>
                it.Result == 0 && it.Status == 1 &&
                it.Seq == 1 && it.Title != null)
            .WhereIF(query.isShow == 1, it => it.DealWith == 0)
            .OrderByDescending(it => it.CreateDate)
            .ToPageList(1, 20);
            .Count();
    }


    //将消息设置为已读的
    public int UpdateRead(MessageCenter query)
    {
        return Db.Updateable<MessageCenter>()
            .SetColumns(s => s.DealWith == 1)
            .SetColumns(s => s.UpdateBy == query.UpdateBy)
            .SetColumns(s =>
                s.UpdateDate == DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss"))
            .Where(s => s.Id == query.Id)
            .ExecuteCommand();
    }


    public int save(MessageCenter entity)
    {
        entity.CreateDate = DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss");