更新
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
<?php
|
||||
|
||||
declare(strict_types=1);
|
||||
|
||||
namespace app\command;
|
||||
|
||||
use app\common\service\prescriptionai\PrescriptionAiStore;
|
||||
use app\common\service\prescriptionai\PrescriptionAiWorker;
|
||||
use think\console\Command;
|
||||
use think\console\Input;
|
||||
use think\console\input\Option;
|
||||
use think\console\Output;
|
||||
|
||||
final class PrescriptionAiBackfill extends Command
|
||||
{
|
||||
protected function configure()
|
||||
{
|
||||
$this->setName('prescription-ai:backfill')->setDescription('预览或分批补登记手工处方AI任务,不调用模型')
|
||||
->addOption('from', null, Option::VALUE_REQUIRED, '开始日期 YYYY-MM-DD')
|
||||
->addOption('to', null, Option::VALUE_REQUIRED, '结束日期 YYYY-MM-DD')
|
||||
->addOption('after-id', null, Option::VALUE_REQUIRED, '上次返回的游标', '0')
|
||||
->addOption('limit', null, Option::VALUE_REQUIRED, '每批最多200', '50')
|
||||
->addOption('apply', null, Option::VALUE_NONE, '确认将本批登记为后台任务');
|
||||
}
|
||||
|
||||
protected function execute(Input $input, Output $output): int
|
||||
{
|
||||
foreach (['from', 'to'] as $key) {
|
||||
$value = (string) $input->getOption($key);
|
||||
if (!preg_match('/^\d{4}-\d{2}-\d{2}$/', $value) || date('Y-m-d', strtotime($value)) !== $value) {
|
||||
$output->writeln('Explicit valid from/to dates are required');
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
if ($input->getOption('apply') && !PrescriptionAiStore::enabled()) {
|
||||
$output->writeln('Enable prescription_analysis before enqueueing');
|
||||
return 1;
|
||||
}
|
||||
$from = strtotime((string) $input->getOption('from'));
|
||||
$to = strtotime((string) $input->getOption('to') . ' 23:59:59');
|
||||
if ($from > $to) {
|
||||
$output->writeln('Invalid date range');
|
||||
return 1;
|
||||
}
|
||||
$result = (new PrescriptionAiWorker())->reconcile((int) $input->getOption('after-id'),
|
||||
(int) $input->getOption('limit'), $from, $to, (bool) $input->getOption('apply'));
|
||||
$output->writeln(json_encode(['dry_run' => !$input->getOption('apply')] + $result));
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user