102 lines
3.6 KiB
C#
102 lines
3.6 KiB
C#
using dy.net.model.dto;
|
|
using dy.net.model.entity;
|
|
using SqlSugar;
|
|
using System.Linq.Expressions;
|
|
|
|
namespace dy.net.repository
|
|
{
|
|
public class DouyinCookieRepository : BaseRepository<DouyinCookie>
|
|
{
|
|
// 注入SQLSugar客户端
|
|
public DouyinCookieRepository(ISqlSugarClient db) : base(db)
|
|
{
|
|
}
|
|
public async Task<bool> FastResetCookie(string id, string cookie)
|
|
{
|
|
if (!string.IsNullOrWhiteSpace(id))
|
|
{
|
|
var result = await Db.Updateable<DouyinCookie>()
|
|
.SetColumns(it => it.Cookies == cookie)
|
|
.Where(it => it.Id == id)
|
|
.ExecuteCommandAsync();
|
|
return result > 0;
|
|
}
|
|
else
|
|
{
|
|
var result = await Db.Updateable<DouyinCookie>()
|
|
.SetColumns(it => it.Cookies == cookie)
|
|
.ExecuteCommandAsync();
|
|
return result > 0;
|
|
}
|
|
}
|
|
public async Task<List<DouyinCookie>> GetAllCookiesAsync(Expression<Func<DouyinCookie, bool>> whereExpression = null)
|
|
{
|
|
// 1. 初始化查询：先加固定条件 Status == 1
|
|
var query = Db.Queryable<DouyinCookie>()
|
|
.Where(x => x.Status == 1)
|
|
.Where(x => !string.IsNullOrWhiteSpace(x.Cookies)); // 固定条件（必选）
|
|
|
|
// 2. 若传入自定义条件，叠加 Where（自动 AND 组合）
|
|
if (whereExpression != null)
|
|
{
|
|
query = query.Where(whereExpression); // 自定义条件（可选）
|
|
}
|
|
|
|
// 3. 执行查询（SqlSugar 自动合并所有 Where 条件）
|
|
return await query.ToListAsync();
|
|
}
|
|
|
|
public async Task<List<DouyinCookie>> GetAllCookies(Expression<Func<DouyinCookie, bool>> whereExpression = null)
|
|
{
|
|
// 1. 初始化查询：先加固定条件 Status == 1
|
|
var query = Db.Queryable<DouyinCookie>()
|
|
.Where(x => x.Status == 1)
|
|
.Where(x => !string.IsNullOrWhiteSpace(x.Cookies)); // 固定条件（必选）
|
|
|
|
// 2. 若传入自定义条件，叠加 Where（自动 AND 组合）
|
|
if (whereExpression != null)
|
|
{
|
|
query = query.Where(whereExpression); // 自定义条件（可选）
|
|
}
|
|
|
|
// 3. 执行查询（SqlSugar 自动合并所有 Where 条件）
|
|
return await query.ToListAsync();
|
|
}
|
|
|
|
|
|
public async Task<(List<DouyinCookie> list, int totalCount)> GetPagedAsync(int pageIndex, int pageSize)
|
|
{
|
|
var where = this.Db.Queryable<DouyinCookie>();
|
|
|
|
var totalCount = await where.CountAsync();
|
|
var list = await where.Skip((pageIndex - 1) * pageSize).Take(pageSize).ToListAsync();
|
|
return (list, totalCount);
|
|
}
|
|
|
|
public DouyinCookie GetDefault()
|
|
{
|
|
return Db.Queryable<DouyinCookie>()
|
|
.First();
|
|
}
|
|
|
|
|
|
public async Task<bool> SwitchAsync(DouyinCookieSwitchDto dto)
|
|
{
|
|
var res = await Db.Updateable<DouyinCookie>().SetColumns(x => new DouyinCookie { Status = dto.Status }).Where(x => x.Id == dto.Id).ExecuteCommandAsync();
|
|
|
|
return res > 0;
|
|
}
|
|
|
|
public async Task<bool> UpdateCookie(DouyinCookie cookie)
|
|
{
|
|
if (cookie != null)
|
|
{
|
|
var res = await Db.Updateable<DouyinCookie>(cookie).IgnoreColumns(x => new { x.MyUserId }).ExecuteCommandAsync();
|
|
return res > 0;
|
|
}
|
|
return false;
|
|
|
|
}
|
|
}
|
|
}
|