public async Task <IActionResult> Add(string repository)
{
var viewModel = new AddSpiderViewModel();
var dockerRepository = await _dbContext.DockerRepositories.FirstOrDefaultAsync(x =>
$"{x.Registry}{x.Repository}".Replace("http://", "").Replace("https://", "") == repository);
if (dockerRepository == null)
{
throw new Exception("镜像仓库不存在");
}
if (!string.IsNullOrWhiteSpace(dockerRepository.Registry))
{
var httpClient = Common.HttpClientFactory.GetHttpClient(dockerRepository.Registry,
dockerRepository.UserName, dockerRepository.Password);
var json = await httpClient.GetStringAsync(
$"{dockerRepository.Registry}v2/{dockerRepository.Repository}/tags/list");
var repositoryTags = JsonConvert.DeserializeObject <RepositoryTags>(json);
viewModel.Tags = repositoryTags.Tags;
}
return(View(viewModel));
}
public async Task <IActionResult> Add(AddSpiderViewModel dto)
{
if (!ModelState.IsValid)
{
return(await Add(dto.Repository));
}
var exists = await _dbContext.Spiders.AnyAsync(x =>
x.Name == dto.Name);
if (exists)
{
ModelState.AddModelError("Name", "名称已经存在");
}
var dockerRepository = await _dbContext.DockerRepositories.FirstOrDefaultAsync(x =>
$"{x.Registry}{x.Repository}".Replace("http://", "").Replace("https://", "") == dto.Repository);
if (dockerRepository == null)
{
ModelState.AddModelError("Repository", "镜像仓库不存在");
return(await Add(dto.Repository));
}
try
{
TriggerBuilder.Create().WithCronSchedule(dto.Cron).Build();
}
catch
{
ModelState.AddModelError("Cron", "CRON 表达式不正确");
}
if (ModelState.IsValid)
{
// var transaction = await _dbContext.Database.BeginTransactionAsync();
try
{
var spider = new Portal.Entity.Spider
{
Name = dto.Name,
Cron = dto.Cron,
Type = dto.Type,
Registry = dockerRepository.Registry,
Repository = dto.Repository,
Environment = dto.Environment,
Tag = dto.Tag,
Arguments = dto.Arguments,
CreationTime = DateTime.Now
};
_dbContext.Spiders.Add(spider);
await _dbContext.SaveChangesAsync();
var id = spider.Id.ToString();
var trigger = TriggerBuilder.Create().WithCronSchedule(dto.Cron).WithIdentity(id).Build();
var qzJob = JobBuilder.Create <TriggerJob>().WithIdentity(id).WithDescription(spider.Name)
.RequestRecovery(true).StoreDurably(true)
.Build();
await _sched.ScheduleJob(qzJob, trigger);
//transaction.Commit();
}
catch (Exception e)
{
_logger.LogError($"添加任务失败: {e}");
// try
// {
// transaction.Rollback();
// }
// catch (Exception re)
// {
// _logger.LogError($"回滚添加任务失败: {re}");
// }
ModelState.AddModelError(string.Empty, "添加任务失败");
return(View("Add", dto));
}
return(Redirect("/spider"));
}
else
{
return(View("Add", dto));
}
}
public async Task <IActionResult> Add(AddSpiderViewModel dto)
{
if (!ModelState.IsValid)
{
return(View("Add", dto));
}
var exists = await _dbContext.Spiders.AnyAsync(x =>
x.Name == dto.Name);
if (exists)
{
ModelState.AddModelError("Name", "名称已经存在");
}
var imageExists = await _dbContext.DockerImages.AnyAsync(x => x.Image == dto.Image);
if (!imageExists)
{
ModelState.AddModelError("Image", "镜像不存在");
}
try
{
TriggerBuilder.Create().WithCronSchedule(dto.Cron).Build();
}
catch
{
ModelState.AddModelError("Cron", "CRON 表达式不正确");
}
if (ModelState.IsValid)
{
var transaction = await _dbContext.Database.BeginTransactionAsync();
try
{
var spider = new Portal.Entity.Spider
{
Name = dto.Name,
Cron = dto.Cron,
Image = dto.Image,
Environment = dto.Environment,
CreationTime = DateTime.Now,
LastModificationTime = DateTime.Now
};
_dbContext.Spiders.Add(spider);
await _dbContext.SaveChangesAsync();
var id = spider.Id.ToString();
var trigger = TriggerBuilder.Create().WithCronSchedule(dto.Cron).WithIdentity(id).Build();
var qzJob = JobBuilder.Create <TriggerJob>().WithIdentity(id).WithDescription(spider.Name)
.RequestRecovery(true)
.Build();
await _sched.ScheduleJob(qzJob, trigger);
transaction.Commit();
}
catch (Exception e)
{
_logger.LogError($"添加任务失败: {e}");
try
{
transaction.Rollback();
}
catch (Exception re)
{
_logger.LogError($"回滚添加任务失败: {re}");
}
ModelState.AddModelError(string.Empty, "添加任务失败");
return(View("Add", dto));
}
return(Redirect("/spider"));
}
else
{
return(View("Add", dto));
}
}
