WorkOrder.php 51 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494
  1. <?php
  2. namespace app\api\controller;
  3. use app\common\controller\Api;
  4. use app\job\ImageJob;
  5. use app\service\AIGatewayService;
  6. use app\service\ImageService;
  7. use think\App;
  8. use think\Db;
  9. use think\Exception;
  10. use think\Log;
  11. use think\Queue;
  12. use think\queue\job\Redis;
  13. use think\Request;
  14. class WorkOrder extends Api{
  15. protected $noNeedLogin = ['*'];
  16. protected $noNeedRight = ['*'];
  17. public function index(){echo '访问成功';}
  18. /**
  19. * 获取图片生成状态
  20. * @return json 任务状态和图片路径
  21. */
  22. public function GetImageStatus(){
  23. $params = $this->request->param();
  24. $taskId = $params['task_id'];
  25. if (empty($taskId)) {
  26. $res = [
  27. 'code' => 1,
  28. 'msg' => '任务ID不能为空'
  29. ];
  30. return json($res);
  31. }
  32. // 从Redis中获取任务状态(支持文生图 text_to_image_task 和图生图 img_to_img_task)
  33. $redis = new \Redis();
  34. $redis->connect('127.0.0.1', 6379);
  35. $redis->auth('123456');
  36. $redis->select(15);
  37. $taskData = $redis->get("img_to_img_task:{$taskId}");
  38. if (!$taskData) {
  39. $taskData = $redis->get("text_to_image_task:{$taskId}");
  40. }
  41. if (!$taskData) {
  42. $res = [
  43. 'code' => 1,
  44. 'msg' => '任务不存在或已过期',
  45. ];
  46. return json($res);
  47. }
  48. $taskInfo = json_decode($taskData, true);
  49. $res = [
  50. 'code' => 0,
  51. 'msg' => '查询成功',
  52. 'data' => $taskInfo
  53. ];
  54. return json($res);
  55. }
  56. /**
  57. * AI队列入口处理 出图接口
  58. * 此方法处理图像转换为文本的请求,将图像信息存入队列以供后续处理。
  59. */
  60. public function imageToText()
  61. {
  62. $params = $this->request->param();
  63. $service = new ImageService();
  64. $service->handleImage($params);
  65. $this->success('任务成功提交至队列');
  66. }
  67. /**
  68. * product 产品专用统一api调用接口
  69. * AI模型调用接口
  70. */
  71. public function GetTxtToTxt(){
  72. $params = $this->request->param();
  73. $prompt = $params['prompt'];//提示词
  74. $model = $params['model'];//模型
  75. $old_path = $params['path'] ?? '';//原图路径
  76. if(empty($model)) {
  77. return json(['code' => 1, 'msg' => '模型请求失败']);
  78. }
  79. $aiGateway = new AIGatewayService();
  80. $service = new ImageService();
  81. //判断模型入口:status_val
  82. if($params['status_val'] == '图生文'){
  83. $service->handleImgToText($params);
  84. $res = [
  85. 'code' => 0,
  86. 'msg' => '正在优化提示词,请稍等.....'
  87. ];
  88. return json($res);
  89. }else if($params['status_val'] == '文生文'){
  90. $fullPrompt = $prompt;
  91. $fullPrompt .= "
  92. 请根据上述内容生成一段完整的话术,要求:
  93. 1. 内容必须是连贯的一段话,不要使用列表、分隔线或其他结构化格式
  94. 2. 不要包含非文本元素的描述
  95. 3. 不要添加任何额外的引导语、解释或开场白
  96. 4. 语言流畅自然";
  97. $result = $service->handleTextToText($fullPrompt, $model);
  98. if(!empty($params['id'])){
  99. Db::name('product')->where('id', $params['id'])->update(['content' => $result['data']]);
  100. }
  101. $res = [
  102. 'code' => 0,
  103. 'msg' => '优化成功',
  104. 'content' => $result['data']
  105. ];
  106. return json($res);
  107. }elseif($params['status_val'] == '文生图'){
  108. $result = $service->handleTextToImg($params);
  109. if ($result['success']) {
  110. $res = [
  111. 'code' => 0,
  112. 'msg' => $result['message'],
  113. 'data' => [
  114. 'task_id' => $result['task_id']
  115. ]
  116. ];
  117. } else {
  118. $res = [
  119. 'code' => 1,
  120. 'msg' => $result['message']
  121. ];
  122. }
  123. return json($res);
  124. }elseif($params['status_val'] == '图生图'){
  125. $result = $service->handleImgToImg($params);
  126. if (isset($result['success']) && $result['success']) {
  127. return json([
  128. 'code' => 0,
  129. 'time' => date('Y-m-d H:i:s'),
  130. 'msg' => $result['message'],
  131. 'data' => ['task_id' => $result['task_id'] ?? '']
  132. ]);
  133. }
  134. return json([
  135. 'code' => 1,
  136. 'msg' => $result['message'] ?? '图生图任务提交失败',
  137. 'data' => null
  138. ]);
  139. }else{
  140. $this->error('请求失败');
  141. }
  142. }
  143. /**
  144. * 即梦AI--创建视频任务接口
  145. * 首帧图 + 尾帧图 = 新效果视频
  146. */
  147. public function Create_ImgToVideo()
  148. {
  149. $aiGateway = new AIGatewayService();
  150. $apiUrl = $aiGateway->config['Create_ImgToVideo']['api_url'];
  151. $apiKey = $aiGateway->config['Create_ImgToVideo']['api_key'];
  152. $params = $this->request->param();
  153. $modelParams = [
  154. 'resolution' => $params['resolution'] ?? '720p', // 分辨率:480p, 720p, 1080p
  155. 'duration' => $params['duration'] ?? 5, // 视频时长(秒)
  156. 'camerafixed' => $params['camerafixed'] ?? false, // 相机固定
  157. 'watermark' => $params['watermark'] ?? true, // 水印
  158. 'aspect_ratio' => $params['aspect_ratio'] ?? '16:9' // 视频比例:16:9, 4:3, 1:1等
  159. ];
  160. // 构建提示词
  161. $prompt = $params['prompt'] ?? '';
  162. $prompt .= ' --resolution ' . $modelParams['resolution'];
  163. $prompt .= ' --duration ' . $modelParams['duration'];
  164. $prompt .= ' --camerafixed ' . ($modelParams['camerafixed'] ? 'true' : 'false');
  165. $prompt .= ' --watermark ' . ($modelParams['watermark'] ? 'true' : 'false');
  166. $prompt .= ' --aspect_ratio ' . $modelParams['aspect_ratio'];
  167. // 构建请求数据
  168. $data = [
  169. 'model' => 'doubao-seedance-1-0-pro-250528',//模型
  170. 'content' => [
  171. [
  172. 'type' => 'text',
  173. 'text' => $prompt
  174. ],
  175. [
  176. 'type' => 'image_url',
  177. 'image_url' => [
  178. 'url' => $params['first_image_url']// 首帧图片URL(参数)
  179. ],
  180. 'role' => 'first_image'
  181. ],
  182. [
  183. 'type' => 'image_url',
  184. 'image_url' => [
  185. 'url' => $params['last_image_url'] // 尾帧图片URL(参数)
  186. ],
  187. 'role' => 'last_image'
  188. ]
  189. ]
  190. ];
  191. // 转换为 JSON 字符串
  192. $jsonData = json_encode($data);
  193. // 初始化 cURL
  194. $ch = curl_init();
  195. curl_setopt($ch, CURLOPT_URL, $apiUrl);
  196. curl_setopt($ch, CURLOPT_POST, true);
  197. curl_setopt($ch, CURLOPT_POSTFIELDS, $jsonData);
  198. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  199. 'Content-Type: application/json',
  200. 'Authorization: Bearer ' . $apiKey
  201. ]);
  202. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  203. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false); // 开发环境临时关闭SSL验证
  204. curl_setopt($ch, CURLOPT_TIMEOUT, 60); // 超时时间
  205. // 执行请求
  206. $response = curl_exec($ch);
  207. // 检查 cURL 错误
  208. if (curl_errno($ch)) {
  209. $error = curl_error($ch);
  210. curl_close($ch);
  211. return json(['code' => 0, 'msg' => 'Curl 错误: ' . $error]);
  212. }
  213. curl_close($ch);
  214. // 解析响应
  215. $responseData = json_decode($response, true);
  216. // 检查 API 错误
  217. if (isset($responseData['error'])) {
  218. return json(['code' => 0, 'msg' => 'API 错误: ' . $responseData['error']['message']]);
  219. }
  220. // 获取任务 ID
  221. $taskId = $responseData['id'] ?? '';
  222. if (empty($taskId)) {
  223. return json(['code' => 0, 'msg' => '获取任务 ID 失败']);
  224. }
  225. // 返回结果
  226. return json([
  227. 'code' => 1,
  228. 'data' => [
  229. 'task_id' => $taskId,
  230. 'status' => $responseData['status'] ?? '',
  231. 'created_at' => $responseData['created_at'] ?? ''
  232. ]
  233. ]);
  234. }
  235. /**
  236. * 即梦AI--获取视频接口
  237. * 首帧图 + 尾帧图 = 新效果视频
  238. */
  239. public function Get_ImgToVideo()
  240. {
  241. $aiGateway = new AIGatewayService();
  242. $apiUrl = $aiGateway->config['Create_ImgToVideo']['api_url'];
  243. $apiKey = $aiGateway->config['Create_ImgToVideo']['api_key'];
  244. $params = $this->request->param();
  245. $taskId = $params['task_id'] ?? '';
  246. if (empty($taskId)) {
  247. return json(['code' => 0, 'msg' => '任务 ID 不能为空']);
  248. }
  249. // 查询任务状态
  250. $queryUrl = $apiUrl . '/' . $taskId;
  251. $ch2 = curl_init();
  252. curl_setopt($ch2, CURLOPT_URL, $queryUrl);
  253. curl_setopt($ch2, CURLOPT_HTTPHEADER, [
  254. 'Content-Type: application/json',
  255. 'Authorization: Bearer ' . $apiKey
  256. ]);
  257. curl_setopt($ch2, CURLOPT_RETURNTRANSFER, true);
  258. curl_setopt($ch2, CURLOPT_SSL_VERIFYPEER, false); // 开发环境临时关闭SSL验证
  259. curl_setopt($ch2, CURLOPT_TIMEOUT, 60); // 超时时间
  260. $queryResponse = curl_exec($ch2);
  261. // 检查 cURL 错误
  262. if (curl_errno($ch2)) {
  263. $error = curl_error($ch2);
  264. curl_close($ch2);
  265. return json(['code' => 0, 'msg' => 'Curl 错误: ' . $error]);
  266. }
  267. curl_close($ch2);
  268. // 解析查询响应
  269. $queryData = json_decode($queryResponse, true);
  270. // print_r($queryData);die;
  271. // 轮询任务状态,直到完成
  272. $maxPolls = 30; // 最大轮询次数
  273. $pollCount = 0;
  274. $taskStatus = $queryData['status'] ?? '';
  275. while (!in_array($taskStatus, ['completed', 'succeeded']) && $pollCount < $maxPolls) {
  276. sleep(5); // 每5秒轮询一次
  277. $pollCount++;
  278. // 再次查询任务状态
  279. $ch3 = curl_init();
  280. curl_setopt($ch3, CURLOPT_URL, $queryUrl);
  281. curl_setopt($ch3, CURLOPT_HTTPHEADER, [
  282. 'Content-Type: application/json',
  283. 'Authorization: Bearer ' . $apiKey
  284. ]);
  285. curl_setopt($ch3, CURLOPT_RETURNTRANSFER, true);
  286. curl_setopt($ch3, CURLOPT_SSL_VERIFYPEER, false);
  287. curl_setopt($ch3, CURLOPT_TIMEOUT, 60);
  288. $pollResponse = curl_exec($ch3);
  289. curl_close($ch3);
  290. $pollData = json_decode($pollResponse, true);
  291. $taskStatus = $pollData['status'] ?? '';
  292. // 检查任务是否失败
  293. if ($taskStatus === 'failed') {
  294. return json(['code' => 0, 'msg' => '任务执行失败']);
  295. }
  296. }
  297. // 如果任务已经成功,直接使用 $queryData
  298. if (empty($pollData) && isset($queryData['status']) && $queryData['status'] === 'succeeded') {
  299. $pollData = $queryData;
  300. }
  301. // 检查轮询是否超时
  302. if (!in_array($taskStatus, ['completed', 'succeeded'])) {
  303. return json(['code' => 0, 'msg' => '任务执行超时']);
  304. }
  305. // 获取视频 URL
  306. $videoUrl = $pollData['content']['video_url'] ?? '';
  307. if (empty($videoUrl)) {
  308. return json(['code' => 0, 'msg' => '获取视频 URL 失败', 'data' => ['pollData' => $pollData]]);
  309. }
  310. // 确保保存目录存在
  311. $saveDir = ROOT_PATH . 'public/uploads/ceshi/';
  312. if (!is_dir($saveDir)) {
  313. mkdir($saveDir, 0755, true);
  314. }
  315. // 生成保存文件名
  316. $fileName = $taskId . '.mp4';
  317. $savePath = $saveDir . '/' . $fileName;
  318. // 下载视频
  319. // 设置超时时间为300秒(5分钟)
  320. $context = stream_context_create(['http' => ['timeout' => 300]]);
  321. $videoContent = file_get_contents($videoUrl, false, $context);
  322. if ($videoContent === false) {
  323. return json(['code' => 0, 'msg' => '下载视频失败', 'data' => ['videoUrl' => $videoUrl]]);
  324. }
  325. // 保存视频到文件
  326. $saveResult = file_put_contents($savePath, $videoContent);
  327. if ($saveResult === false) {
  328. return json(['code' => 0, 'msg' => '保存视频失败', 'data' => ['savePath' => $savePath, 'dirWritable' => is_writable($saveDir)]]);
  329. }
  330. // 生成视频的访问 URL
  331. $videoAccessUrl = '/uploads/ceshi/' . $fileName;
  332. // 返回结果
  333. return json([
  334. 'code' => 1,
  335. 'data' => [
  336. 'task_id' => $taskId,
  337. 'status' => $taskStatus,
  338. 'video_url' => $videoAccessUrl,
  339. 'save_path' => $savePath
  340. ]
  341. ]);
  342. }
  343. /**
  344. * 九个分镜头生成流程
  345. * 模型:gemini-3-pro-image-preview
  346. * 说明:
  347. * 第一步:(提示词 + 原始图片 = 九个分镜头图片) 或 (提示词 = 九个分镜头图片)
  348. * 第二步:使用九个分镜头图进行裁剪单图生成连贯视频
  349. * 第三步:在通过分镜头视频拼接成一个完整的视频(列如每个分镜头视频为8秒,九个为72秒形成完整视频)
  350. */
  351. public function Get_txttonineimg()
  352. {
  353. // 发起接口请求
  354. // $apiUrl = 'https://chatapi.onechats.ai/v1beta/models/gemini-3-pro-image-preview:generateContent';
  355. $apiUrl = 'https://chatapi.onechats.ai/v1beta/models/gemini-3-pro-image-preview:streamGenerateContent';
  356. $apiKey = 'sk-9aIV9nx7pJxJFMrB8REtNbhjYuNBxCcnEOwiJDHd6UwmN2eJ';
  357. // $params = $this->request->param();
  358. $prompt = '生成一个苹果(九个分镜头图片)';
  359. $requestData = [
  360. "contents" => [
  361. [
  362. "role" => "user",
  363. "parts" => [
  364. ["text" => $prompt]
  365. ]
  366. ]
  367. ],
  368. "generationConfig" => [
  369. "responseModalities" => ["TEXT", "IMAGE"],
  370. "imageConfig" => [
  371. "aspectRatio" => "1:1"
  372. ]
  373. ]
  374. ];
  375. $ch = curl_init();
  376. curl_setopt($ch, CURLOPT_URL, $apiUrl);
  377. curl_setopt($ch, CURLOPT_POST, true);
  378. curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($requestData, JSON_UNESCAPED_UNICODE));
  379. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  380. 'Content-Type: application/json',
  381. 'Authorization: Bearer ' . $apiKey
  382. ]);
  383. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  384. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false); // 开发环境临时关闭SSL验证
  385. curl_setopt($ch, CURLOPT_TIMEOUT, 60); // 生成图片超时时间(建议60秒)
  386. $response = curl_exec($ch);
  387. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  388. $error = curl_error($ch);
  389. curl_close($ch);
  390. $res = json_decode($response, true);
  391. // 构建URL路径(使用正斜杠)
  392. $url_path = '/uploads/txtnewimg/';
  393. // 构建物理路径(使用正斜杠确保统一格式)
  394. $save_path = ROOT_PATH . 'public' . '/' . 'uploads' . '/' . 'txtnewimg' . '/';
  395. // 移除ROOT_PATH中可能存在的反斜杠,确保统一使用正斜杠
  396. $save_path = str_replace('\\', '/', $save_path);
  397. // 自动创建文件夹(如果不存在)
  398. if (!is_dir($save_path)) {
  399. mkdir($save_path, 0755, true);
  400. }
  401. // 提取base64图片数据
  402. $text_content = $res['candidates'][0]['content']['parts'][0]['inlineData']['data'];
  403. $str = 'data:image/jpeg;base64,';
  404. $text_content = $str. $text_content;
  405. // 匹配base64图片数据
  406. preg_match('/data:image\/(png|jpg|jpeg);base64,([^"]+)/', $text_content, $matches);
  407. if (empty($matches)) {
  408. return '未找到图片数据';
  409. }
  410. $image_type = $matches[1];
  411. $base64_data = $matches[2];
  412. // 解码base64数据
  413. $image_data = base64_decode($base64_data);
  414. if ($image_data === false) {
  415. return '图片解码失败';
  416. }
  417. // 生成唯一文件名(包含扩展名)
  418. $file_name = uniqid() . '.' . $image_type;
  419. $full_file_path = $save_path . $file_name;
  420. // 保存图片到文件系统
  421. if (!file_put_contents($full_file_path, $image_data)) {
  422. return '图片保存失败';
  423. }
  424. // 生成数据库存储路径(使用正斜杠格式)
  425. $db_img_path = $url_path . $file_name;
  426. return $db_img_path;
  427. }
  428. public function GetTxtToImg(){
  429. $params = $this->request->param();
  430. $prompt = $params['prompt'];//提示词
  431. $model = $params['model'];//模型
  432. $size = $params['size'];//尺寸
  433. // 调用AI生成图片
  434. $aiGateway = new AIGatewayService();
  435. $res = $aiGateway->callDalleApi($prompt, $model, $size);
  436. // 提取base64图片数据
  437. $imageData = '';
  438. $imageType = 'png'; // 默认图片类型
  439. if (isset($res['candidates'][0]['content']['parts'][0]['text'])) {
  440. $text_content = $res['candidates'][0]['content']['parts'][0]['text'];
  441. // 匹配base64图片数据和类型
  442. if (preg_match('/data:image\/([a-zA-Z0-9]+);base64,([^\s]+)/', $text_content, $matches)) {
  443. $imageType = strtolower($matches[1]);
  444. $base64Data = $matches[2];
  445. // 解码base64数据
  446. $imageData = base64_decode($base64Data);
  447. }
  448. }
  449. if (!$imageData) {
  450. return json(['code' => 1, 'msg' => '图片生成失败,未找到有效图片数据']);
  451. }
  452. // 创建保存目录(public/uploads/log/YYYY-MM/)
  453. $yearMonth = date('Ym');
  454. $saveDir = ROOT_PATH . 'public/uploads/ceshi/' . $yearMonth . '/';
  455. if (!is_dir($saveDir)) {
  456. mkdir($saveDir, 0755, true);
  457. }
  458. // 生成唯一文件名
  459. $fileName = uniqid() . '.' . $imageType;
  460. $filePath = $saveDir . $fileName;
  461. // 保存图片到本地文件
  462. if (file_put_contents($filePath, $imageData) === false) {
  463. return json(['code' => 1, 'msg' => '图片保存失败']);
  464. }
  465. // 生成前端可访问的URL
  466. $imageUrl = '/uploads/ceshi/' . $yearMonth . '/' . $fileName;
  467. // 返回标准JSON响应
  468. return json([
  469. 'code' => 0,
  470. 'msg' => '图片生成成功',
  471. 'data' => [
  472. 'url' => $imageUrl
  473. ]
  474. ]);
  475. }
  476. /**
  477. * 学生端文生视频接口 - 用于生成视频
  478. */
  479. public function GetTxtToVideo(){
  480. $aiGateway = new AIGatewayService();
  481. $apiUrl = $aiGateway->config['videos']['api_url'];
  482. $apiKey = $aiGateway->config['videos']['api_key'];
  483. // 获取并验证参数
  484. $params = $this->request->param();
  485. // echo "<pre>";
  486. // print_r($params);
  487. // echo "<pre>";die;
  488. $postData = [
  489. 'prompt' => $params['prompt'],
  490. 'model' => $params['model'],
  491. 'seconds' => $params['duration'],
  492. 'size' => $params['size'],
  493. ];
  494. // 初始化CURL
  495. $ch = curl_init();
  496. curl_setopt($ch, CURLOPT_URL, $apiUrl);
  497. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  498. curl_setopt($ch, CURLOPT_POST, true);
  499. curl_setopt($ch, CURLOPT_POSTFIELDS, $postData);
  500. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  501. 'Authorization: Bearer ' . $apiKey
  502. ]);
  503. curl_setopt($ch, CURLOPT_TIMEOUT, 300);
  504. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  505. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  506. curl_setopt($ch, CURLOPT_HEADER, true); // 获取响应头
  507. curl_setopt($ch, CURLOPT_VERBOSE, true); // 启用详细输出以进行调试
  508. // 创建临时文件来捕获详细的cURL输出
  509. $verbose = fopen('php://temp', 'w+');
  510. curl_setopt($ch, CURLOPT_STDERR, $verbose);
  511. // 执行请求
  512. $response = curl_exec($ch);
  513. //HTTP状态码
  514. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  515. // 获取详细的cURL调试信息
  516. rewind($verbose);
  517. //CURL调试信息
  518. $verboseLog = stream_get_contents($verbose);
  519. fclose($verbose);
  520. // 分离头部和主体
  521. $header_size = curl_getinfo($ch, CURLINFO_HEADER_SIZE);
  522. //响应头部
  523. $header = substr($response, 0, $header_size);
  524. //响应主体
  525. $body = substr($response, $header_size);
  526. // 检查CURL错误
  527. $curlError = curl_error($ch);
  528. curl_close($ch);
  529. $responseData = json_decode($body, true);
  530. // 检查API是否返回了错误信息
  531. if (isset($responseData['error'])) {
  532. $errorMessage = isset($responseData['error']['message']) ? $responseData['error']['message'] : 'API请求失败';
  533. return json([
  534. 'code' => 1,
  535. 'msg' => '视频生成请求失败',
  536. 'data' => [
  537. 'error_type' => isset($responseData['error']['type']) ? $responseData['error']['type'] : 'unknown',
  538. 'error_code' => isset($responseData['error']['code']) ? $responseData['error']['code'] : 'unknown',
  539. 'error_message' => $errorMessage
  540. ]
  541. ]);
  542. }
  543. // 检查是否有自定义错误格式
  544. if (isset($responseData['code']) && $responseData['code'] === 'fail_to_fetch_task' && isset($responseData['message'])) {
  545. return json([
  546. 'code' => 1,
  547. 'msg' => '视频生成请求失败',
  548. 'data' => [
  549. 'error_code' => $responseData['code'],
  550. 'error_message' => $responseData['message']
  551. ]
  552. ]);
  553. }
  554. // 检查是否存在id字段
  555. if (!isset($responseData['id'])) {
  556. return json([
  557. 'code' => 1,
  558. 'msg' => '无法获取视频ID',
  559. 'data' => [
  560. 'response_data' => $responseData,
  561. 'http_code' => $httpCode
  562. ]
  563. ]);
  564. }
  565. // 1. 先检查视频状态
  566. $statusUrl = 'https://chatapi.onechats.ai/v1/videos/' . $responseData['id'];
  567. $statusData = $this->fetchVideoStatus($statusUrl, $apiKey);
  568. // 检查视频状态
  569. if ($statusData['status'] !== 'completed') {
  570. return json([
  571. 'code' => 202,
  572. 'msg' => '视频尚未生成完成',
  573. 'data' => [
  574. 'video_id' => $responseData['id'],
  575. 'status' => $statusData['status'],
  576. 'progress' => $statusData['progress'],
  577. 'created_at' => $statusData['created_at'],
  578. 'message' => '请稍后再试,视频仍在' . ($statusData['status'] === 'queued' ? '排队中' : '处理中')
  579. ]
  580. ]);
  581. }
  582. // 2. 视频生成完成,准备下载
  583. $apiUrl = 'https://chatapi.onechats.ai/v1/videos/' . $responseData['id'] . '/content';
  584. // 获取可选的variant参数
  585. $variant = $this->request->get('variant', '');
  586. if (!empty($variant)) {
  587. $apiUrl .= '?variant=' . urlencode($variant);
  588. }
  589. // 创建保存目录
  590. $saveDir = ROOT_PATH . 'public' . DS . 'uploads' . DS . 'videos' . DS . date('Ymd');
  591. if (!is_dir($saveDir)) {
  592. mkdir($saveDir, 0755, true);
  593. }
  594. // 生成唯一文件名
  595. $filename = $responseData['id'] . '.mp4';
  596. $localPath = DS . 'uploads' . DS . 'videos' . DS . date('Ymd') . DS . $filename;
  597. $fullPath = $saveDir . DS . $filename;
  598. // 3. 下载视频
  599. $videoData = $this->downloadVideo($apiUrl, $apiKey);
  600. // 4. 保存视频文件
  601. if (file_put_contents($fullPath, $videoData) === false) {
  602. throw new Exception('视频保存失败');
  603. }
  604. // 确保路径使用正斜杠,并只保存相对路径部分
  605. $localPath = str_replace('\\', '/', $localPath);
  606. // 移除开头的斜杠,确保路径格式为uploads/videos/...
  607. $savePath = ltrim($localPath, '/');
  608. // 返回成功响应
  609. return json([
  610. 'code' => 0,
  611. 'msg' => '视频下载成功',
  612. 'data' => [
  613. 'video_id' => $responseData['id'],
  614. 'web_url' => $savePath
  615. ]
  616. ]);
  617. }
  618. /**
  619. * 学生端视频状态查询接口 - 用于通过video_id查询视频状态
  620. */
  621. public function Getvideo_id(){
  622. // 获取并验证参数
  623. $params = $this->request->param();
  624. $videoId = $params['video_id'] ?? '';
  625. // 验证video_id参数
  626. if (empty($videoId)) {
  627. return json([
  628. 'code' => 1,
  629. 'msg' => '缺少必要参数:video_id',
  630. 'data' => []
  631. ]);
  632. }
  633. $aiGateway = new AIGatewayService();
  634. // $apiUrl = $aiGateway->config['videos']['api_url'];
  635. $apiKey = $aiGateway->config['videos']['api_key'];
  636. $apiUrl = 'https://chatapi.onechats.ai/v1/videos/' . $videoId;
  637. // 先检查本地是否已经有该视频
  638. $localVideoPath = ROOT_PATH . 'public' . DS . 'uploads' . DS . 'videos' . DS . date('Ymd') . DS . $videoId . '.mp4';
  639. if (file_exists($localVideoPath)) {
  640. // 视频已存在本地,直接返回
  641. $webPath = '/uploads/videos/' . date('Ymd') . '/' . $videoId . '.mp4';
  642. return json([
  643. 'code' => 0,
  644. 'msg' => '视频下载成功',
  645. 'data' => [
  646. 'video_id' => $videoId,
  647. 'web_url' => substr($webPath, 1) // 移除开头的斜杠
  648. ]
  649. ]);
  650. }
  651. // 检查视频状态
  652. $statusData = $this->fetchVideoStatus($apiUrl, $apiKey);
  653. // 检查视频状态
  654. if ($statusData['status'] !== 'completed') {
  655. return json([
  656. 'code' => 202,
  657. 'msg' => '视频尚未生成完成',
  658. 'data' => [
  659. 'video_id' => $videoId,
  660. 'status' => $statusData['status'],
  661. 'progress' => isset($statusData['progress']) ? $statusData['progress'] : 0,
  662. 'created_at' => $statusData['created_at'],
  663. 'message' => '请稍后再试,视频仍在' . ($statusData['status'] === 'queued' ? '排队中' : '处理中')
  664. ]
  665. ]);
  666. }
  667. // 视频生成完成,下载视频
  668. $downloadUrl = 'https://chatapi.onechats.ai/v1/videos/' . $videoId . '/content';
  669. // 获取可选的variant参数
  670. $variant = $this->request->get('variant', '');
  671. if (!empty($variant)) {
  672. $downloadUrl .= '?variant=' . urlencode($variant);
  673. }
  674. // 创建保存目录
  675. $saveDir = ROOT_PATH . 'public' . DS . 'uploads' . DS . 'videos' . DS . date('Ymd');
  676. if (!is_dir($saveDir)) {
  677. mkdir($saveDir, 0755, true);
  678. }
  679. // 生成唯一文件名
  680. $filename = $videoId . '.mp4';
  681. $localPath = DS . 'uploads' . DS . 'videos' . DS . date('Ymd') . DS . $filename;
  682. $fullPath = $saveDir . DS . $filename;
  683. // 下载视频
  684. $videoData = $this->downloadVideo($downloadUrl, $apiKey);
  685. // 保存视频文件
  686. if (file_put_contents($fullPath, $videoData) === false) {
  687. throw new Exception('视频保存失败');
  688. }
  689. // 确保路径使用正斜杠,并只保存相对路径部分
  690. $localPath = str_replace(DIRECTORY_SEPARATOR, '/', $localPath);
  691. // 移除开头的斜杠,确保路径格式为uploads/videos/...
  692. $savePath = ltrim($localPath, '/');
  693. // 返回成功响应
  694. return json([
  695. 'code' => 0,
  696. 'msg' => '视频下载成功',
  697. 'data' => [
  698. 'video_id' => $videoId,
  699. 'web_url' => $savePath
  700. ]
  701. ]);
  702. }
  703. //获取视频列表
  704. public function Getvideolist(){
  705. if (!$this->request->isGet()) {
  706. $this->error('请求方式错误');
  707. }
  708. $params = $this->request->param();
  709. $search = input('search', '');
  710. $page = isset($params['page']) ? (int)$params['page'] : 1;
  711. $limit = isset($params['limit']) ? (int)$params['limit'] : 50;
  712. $where = [];
  713. if (!empty($search)) {
  714. $where['prompt'] = ['like', '%' . $search . '%'];
  715. }
  716. $list = Db::name('video')->where('mod_rq', null)
  717. ->where($where)
  718. ->order('id desc')
  719. ->limit(($page - 1) * $limit, $limit)
  720. ->select();
  721. $total = Db::name('video')->where('mod_rq', null)
  722. ->where($where)
  723. ->count();
  724. $res['code'] = 0;
  725. $res['msg'] = '成功';
  726. $res['count'] = $total;
  727. $res['data'] = $list;
  728. return json($res);
  729. }
  730. /**
  731. * 文生视频/图生视频接口
  732. * @return \think\response\Json
  733. * @throws \Exception
  734. */
  735. //文生视频
  736. public function video(){
  737. $aiGateway = new AIGatewayService();
  738. $apiUrl = $aiGateway->config['videos']['api_url'];
  739. $apiKey = $aiGateway->config['videos']['api_key'];
  740. $params = $this->request->param();
  741. $postData = [
  742. 'prompt' => $params['prompt'],
  743. 'model' => $params['model'],
  744. 'seconds' => $params['seconds'],
  745. 'size' => $params['size'],
  746. ];
  747. // 初始化CURL
  748. $ch = curl_init();
  749. curl_setopt($ch, CURLOPT_URL, $apiUrl);
  750. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  751. curl_setopt($ch, CURLOPT_POST, true);
  752. curl_setopt($ch, CURLOPT_POSTFIELDS, $postData);
  753. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  754. 'Authorization: Bearer ' . $apiKey
  755. ]);
  756. curl_setopt($ch, CURLOPT_TIMEOUT, 300);
  757. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  758. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  759. curl_setopt($ch, CURLOPT_HEADER, true); // 获取响应头
  760. curl_setopt($ch, CURLOPT_VERBOSE, true); // 启用详细输出以进行调试
  761. // 创建临时文件来捕获详细的cURL输出
  762. $verbose = fopen('php://temp', 'w+');
  763. curl_setopt($ch, CURLOPT_STDERR, $verbose);
  764. // 执行请求
  765. $response = curl_exec($ch);
  766. //HTTP状态码
  767. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  768. // 获取详细的cURL调试信息
  769. rewind($verbose);
  770. //CURL调试信息
  771. $verboseLog = stream_get_contents($verbose);
  772. fclose($verbose);
  773. // 分离头部和主体
  774. $header_size = curl_getinfo($ch, CURLINFO_HEADER_SIZE);
  775. //响应头部
  776. $header = substr($response, 0, $header_size);
  777. //响应主体
  778. $body = substr($response, $header_size);
  779. // 检查CURL错误
  780. $curlError = curl_error($ch);
  781. curl_close($ch);
  782. $responseData = json_decode($body, true);
  783. // 检查API是否返回了错误信息
  784. if (isset($responseData['error'])) {
  785. $errorMessage = isset($responseData['error']['message']) ? $responseData['error']['message'] : 'API请求失败';
  786. return json([
  787. 'code' => 1,
  788. 'msg' => '视频生成请求失败',
  789. 'data' => [
  790. 'error_type' => isset($responseData['error']['type']) ? $responseData['error']['type'] : 'unknown',
  791. 'error_code' => isset($responseData['error']['code']) ? $responseData['error']['code'] : 'unknown',
  792. 'error_message' => $errorMessage
  793. ]
  794. ]);
  795. }
  796. // 检查是否有自定义错误格式
  797. if (isset($responseData['code']) && $responseData['code'] === 'fail_to_fetch_task' && isset($responseData['message'])) {
  798. return json([
  799. 'code' => 1,
  800. 'msg' => '视频生成请求失败',
  801. 'data' => [
  802. 'error_code' => $responseData['code'],
  803. 'error_message' => $responseData['message']
  804. ]
  805. ]);
  806. }
  807. // 检查是否存在id字段
  808. if (!isset($responseData['id'])) {
  809. return json([
  810. 'code' => 1,
  811. 'msg' => '无法获取视频ID',
  812. 'data' => [
  813. 'response_data' => $responseData,
  814. 'http_code' => $httpCode
  815. ]
  816. ]);
  817. }
  818. $videoData = [
  819. 'video_id' => $responseData['id'],
  820. 'prompt' => $postData['prompt'],
  821. 'model' => $postData['model'],
  822. 'seconds' => $postData['seconds'],
  823. 'size' => $postData['size'],
  824. 'sys_rq' => date("Y-m-d H:i:s")
  825. ];
  826. // 尝试插入数据
  827. try {
  828. $res = Db::name('video')->insert($videoData);
  829. return json([
  830. 'code' => 0,
  831. 'msg' => '视频正在生成中',
  832. 'data ' => [
  833. 'video_id' => $responseData['id'],
  834. 'insert_result' => $res
  835. ]
  836. ]);
  837. } catch (Exception $e) {
  838. return json([
  839. 'code' => 1,
  840. 'msg' => '数据库操作失败',
  841. 'data' => [
  842. 'error_message' => $e->getMessage()
  843. ]
  844. ]);
  845. }
  846. }
  847. /**
  848. * 获取视频内容
  849. * 下载已完成的视频内容
  850. */
  851. public function videoContent(){
  852. // 从请求参数获取video_id,如果没有则使用默认值
  853. $video_id = input('get.video_id');
  854. $apiKey = 'sk-sWW1GFlnjbrDRb1DkMEzePIxgdvLK6cZt0Qg93yDMVP2z1yN';
  855. // 1. 先检查视频状态
  856. $statusUrl = 'https://chatapi.onechats.ai/v1/videos/' . $video_id;
  857. $statusData = $this->fetchVideoStatus($statusUrl, $apiKey);
  858. // 检查视频状态
  859. if ($statusData['status'] !== 'completed') {
  860. return json([
  861. 'code' => 202,
  862. 'msg' => '视频尚未生成完成',
  863. 'data' => [
  864. 'video_id' => $video_id,
  865. 'status' => $statusData['status'],
  866. 'progress' => $statusData['progress'],
  867. 'created_at' => $statusData['created_at'],
  868. 'message' => '请稍后再试,视频仍在' . ($statusData['status'] === 'queued' ? '排队中' : '处理中')
  869. ]
  870. ]);
  871. }
  872. // 2. 视频生成完成,准备下载
  873. $apiUrl = 'https://chatapi.onechats.ai/v1/videos/' . $video_id . '/content';
  874. // 获取可选的variant参数
  875. $variant = $this->request->get('variant', '');
  876. if (!empty($variant)) {
  877. $apiUrl .= '?variant=' . urlencode($variant);
  878. }
  879. // 创建保存目录
  880. $saveDir = ROOT_PATH . 'public' . DS . 'uploads' . DS . 'videos' . DS . date('Ymd');
  881. if (!is_dir($saveDir)) {
  882. mkdir($saveDir, 0755, true);
  883. }
  884. // 生成唯一文件名
  885. $filename = $video_id . '.mp4';
  886. $localPath = DS . 'uploads' . DS . 'videos' . DS . date('Ymd') . DS . $filename;
  887. $fullPath = $saveDir . DS . $filename;
  888. // 3. 下载视频
  889. $videoData = $this->downloadVideo($apiUrl, $apiKey);
  890. // 4. 保存视频文件
  891. if (file_put_contents($fullPath, $videoData) === false) {
  892. throw new Exception('视频保存失败');
  893. }
  894. // 确保路径使用正斜杠,并只保存相对路径部分
  895. $localPath = str_replace('\\', '/', $localPath);
  896. // 移除开头的斜杠,确保路径格式为uploads/videos/...
  897. $savePath = ltrim($localPath, '/');
  898. // 将正确格式的文件路径存入数据库
  899. Db::name('video')->where('video_id', $video_id)->update([
  900. 'web_url' => $savePath
  901. ]);
  902. // 返回成功响应
  903. return json([
  904. 'code' => 0,
  905. 'msg' => '视频下载成功',
  906. 'data' => [
  907. 'video_id' => $video_id,
  908. 'local_path' => $localPath,
  909. 'web_url' => $savePath,
  910. 'file_size' => filesize($fullPath)
  911. ]
  912. ]);
  913. }
  914. /**
  915. * 获取视频状态
  916. */
  917. private function fetchVideoStatus($url, $apiKey) {
  918. $ch = curl_init();
  919. curl_setopt($ch, CURLOPT_URL, $url);
  920. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  921. curl_setopt($ch, CURLOPT_HTTPGET, true);
  922. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  923. 'Authorization: Bearer ' . $apiKey,
  924. 'Accept: application/json'
  925. ]);
  926. curl_setopt($ch, CURLOPT_TIMEOUT, 30);
  927. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  928. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  929. $response = curl_exec($ch);
  930. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  931. $error = curl_error($ch);
  932. curl_close($ch);
  933. if ($error) {
  934. throw new Exception('获取视频状态失败: ' . $error);
  935. }
  936. if ($httpCode < 200 || $httpCode >= 300) {
  937. throw new Exception('获取视频状态失败,HTTP状态码: ' . $httpCode);
  938. }
  939. $data = json_decode($response, true);
  940. if (!is_array($data)) {
  941. throw new Exception('视频状态数据格式错误');
  942. }
  943. return $data;
  944. }
  945. /**
  946. * 下载视频文件
  947. */
  948. private function downloadVideo($url, $apiKey) {
  949. $ch = curl_init();
  950. curl_setopt($ch, CURLOPT_URL, $url);
  951. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  952. curl_setopt($ch, CURLOPT_HTTPGET, true);
  953. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  954. 'Authorization: Bearer ' . $apiKey
  955. ]);
  956. curl_setopt($ch, CURLOPT_TIMEOUT, 300);
  957. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  958. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, 0);
  959. $response = curl_exec($ch);
  960. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  961. $error = curl_error($ch);
  962. curl_close($ch);
  963. if ($error) {
  964. throw new Exception('视频下载失败: ' . $error);
  965. }
  966. if ($httpCode < 200 || $httpCode >= 300) {
  967. throw new Exception('视频下载失败,HTTP状态码: ' . $httpCode);
  968. }
  969. return $response;
  970. }
  971. private function sendPostRequest($url, $data, $apiKey)
  972. {
  973. $ch = curl_init();
  974. curl_setopt($ch, CURLOPT_URL, $url);
  975. curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
  976. curl_setopt($ch, CURLOPT_POST, true);
  977. curl_setopt($ch, CURLOPT_HTTPHEADER, [
  978. 'Authorization: Bearer ' . $apiKey,
  979. 'Accept: application/json',
  980. 'Content-Type: application/json'
  981. ]);
  982. curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($data));
  983. curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
  984. curl_setopt($ch, CURLOPT_SSL_VERIFYHOST, false);
  985. curl_setopt($ch, CURLOPT_TIMEOUT, 60); // 延长超时时间
  986. $response = curl_exec($ch);
  987. $httpCode = curl_getinfo($ch, CURLINFO_HTTP_CODE);
  988. $error = curl_error($ch);
  989. curl_close($ch);
  990. return [
  991. 'response' => $response,
  992. 'http_code' => $httpCode,
  993. 'error' => $error
  994. ];
  995. }
  996. /**
  997. * 查询队列列表
  998. * 统计文件对应的队列情况
  999. */
  1000. public function get_queue_logs()
  1001. {
  1002. $params = $this->request->param('old_image_file', '');
  1003. $queue_logs = Db::name('queue_logs')
  1004. ->where('old_image_file', $params)
  1005. ->order('id desc')
  1006. ->select();
  1007. $result = []; //初始化变量,避免未定义错误
  1008. foreach ($queue_logs as &$log) {
  1009. $taskId = $log['id'];
  1010. $statusCount = Db::name('image_task_log')
  1011. ->field('status, COUNT(*) as count')
  1012. ->where('task_id', $taskId)
  1013. ->where('mod_rq', null)
  1014. ->group('status')
  1015. ->select();
  1016. $log['已完成数量'] = 0;
  1017. $log['处理中数量'] = 0;
  1018. $log['排队中的数量'] = 0;
  1019. $log['失败数量'] = 0;
  1020. foreach ($statusCount as $item) {
  1021. switch ($item['status']) {
  1022. case 0:
  1023. $log['排队中的数量'] = $item['count'];
  1024. break;
  1025. case 1:
  1026. $log['处理中数量'] = $item['count'];
  1027. break;
  1028. case 2:
  1029. $log['已完成数量'] = $item['count'];
  1030. break;
  1031. case -1:
  1032. $log['失败数量'] = $item['count'];
  1033. break;
  1034. }
  1035. }
  1036. // if ($log['排队中的数量'] >$log['已完成数量']) {
  1037. // $result[] = $log;
  1038. // }
  1039. if ($log['排队中的数量']) {
  1040. $result[] = $log;
  1041. }
  1042. // if ($log['处理中数量'] >= 0) {
  1043. // $result[] = $log;
  1044. // }
  1045. }
  1046. return json([
  1047. 'code' => 0,
  1048. 'msg' => '查询成功',
  1049. 'data' => $result,
  1050. 'count' => count($result)
  1051. ]);
  1052. }
  1053. /**
  1054. * 查询总队列状态(统计当前处理的数据量)
  1055. */
  1056. public function queueStats()
  1057. {
  1058. $statusList = Db::name('image_task_log')
  1059. ->field('status, COUNT(*) as total')
  1060. ->where('mod_rq', null)
  1061. ->where('create_time', '>=', date('Y-m-d 00:00:00'))
  1062. ->group('status')
  1063. ->select();
  1064. $statusCount = [];
  1065. foreach ($statusList as $item) {
  1066. $statusCount[$item['status']] = $item['total'];
  1067. }
  1068. // 总数为所有状态和
  1069. $total = array_sum($statusCount);
  1070. //获取队列当前状态
  1071. $statusText = Db::name('queue_logs')->order('id desc')->value('status');
  1072. return json([
  1073. 'code' => 0,
  1074. 'msg' => '获取成功',
  1075. 'data' => [
  1076. '总任务数' => $total,
  1077. '待处理' => $statusCount[0] ?? 0,
  1078. '处理中' => $statusCount[1] ?? 0,
  1079. '成功' => $statusCount[2] ?? 0,
  1080. '失败' => $statusCount[-1] ?? 0,
  1081. '当前状态' => $statusText
  1082. ]
  1083. ]);
  1084. }
  1085. /**
  1086. * 获取 Redis 连接实例
  1087. * @return \Redis|null Redis 实例或 null(如果连接失败)
  1088. */
  1089. private function getRedisConnection()
  1090. {
  1091. if (!class_exists('\Redis')) {
  1092. return null;
  1093. }
  1094. $redis = new \Redis();
  1095. $redis->connect('127.0.0.1', 6379);
  1096. $redis->auth('');
  1097. $redis->select(15);
  1098. return $redis;
  1099. }
  1100. /**
  1101. * 显示当前运行中的队列监听进程
  1102. */
  1103. public function viewQueueStatus()
  1104. {
  1105. $redis = $this->getRedisConnection();
  1106. if (!$redis) {
  1107. return json([
  1108. 'code' => 1,
  1109. 'msg' => 'Redis扩展未安装或未启用',
  1110. 'data' => null
  1111. ]);
  1112. }
  1113. $key = 'queues:imgtotxt';
  1114. // 判断 key 是否存在,避免报错
  1115. if (!$redis->exists($key)) {
  1116. return json([
  1117. 'code' => 0,
  1118. 'msg' => '查询成功,队列为空',
  1119. 'count' => 0,
  1120. 'tasks_preview' => []
  1121. ]);
  1122. }
  1123. $count = $redis->lLen($key);
  1124. $list = $redis->lRange($key, 0, 9);
  1125. // 解码 JSON 内容,确保每一项都有效
  1126. $parsed = array_filter(array_map(function ($item) {
  1127. return json_decode($item, true);
  1128. }, $list), function ($item) {
  1129. return !is_null($item);
  1130. });
  1131. return json([
  1132. 'code' => 0,
  1133. 'msg' => '查询成功',
  1134. 'count' => $count,
  1135. 'tasks_preview' => $parsed
  1136. ]);
  1137. }
  1138. /**
  1139. * 清空队列并删除队列日志记录
  1140. */
  1141. public function stopQueueProcesses()
  1142. {
  1143. Db::name('image_task_log')
  1144. ->where('log', '队列中')
  1145. ->whereOr('status', 1)
  1146. ->where('create_time', '>=', date('Y-m-d 00:00:00'))
  1147. ->update([
  1148. 'status' => "-1",
  1149. 'log' => '清空取消队列',
  1150. 'mod_rq' => date('Y-m-d H:i:s')
  1151. ]);
  1152. Db::name('image_task_log')
  1153. ->whereLike('log', '%处理中%')
  1154. ->where('create_time', '>=', date('Y-m-d 00:00:00'))
  1155. ->update([
  1156. 'status' => "-1",
  1157. 'log' => '清空取消队列',
  1158. 'mod_rq' => date('Y-m-d H:i:s')
  1159. ]);
  1160. $redis = $this->getRedisConnection();
  1161. if (!$redis) {
  1162. return json([
  1163. 'code' => 1,
  1164. 'msg' => 'Redis扩展未安装或未启用',
  1165. 'data' => null
  1166. ]);
  1167. }
  1168. $key_txttoimg = 'queues:txttoimg:reserved';
  1169. $key_txttotxt = 'queues:txttotxt:reserved';
  1170. $key_imgtotxt = 'queues:imgtotxt:reserved';
  1171. $key_imgtoimg = 'queues:imgtoimg:reserved';
  1172. // 清空 Redis 队列
  1173. $redis->del($key_txttoimg);
  1174. $redis->del($key_txttotxt);
  1175. $redis->del($key_imgtotxt);
  1176. $redis->del($key_imgtoimg);
  1177. $count = $redis->lLen($key_txttoimg) + $redis->lLen($key_txttotxt) + $redis->lLen($key_imgtotxt) + $redis->lLen($key_imgtoimg);
  1178. // if ($count === 0) {
  1179. // return json([
  1180. // 'code' => 1,
  1181. // 'msg' => '暂无队列需要停止'
  1182. // ]);
  1183. // }
  1184. return json([
  1185. 'code' => 0,
  1186. 'msg' => '成功停止队列任务'
  1187. ]);
  1188. }
  1189. /**
  1190. * 开启队列任务
  1191. * 暂时用不到、服务器已开启自动开启队列模式
  1192. */
  1193. // public function kaiStats()
  1194. // {
  1195. // // 判断是否已有监听进程在运行
  1196. // $check = shell_exec("ps aux | grep 'queue:listen' | grep -v grep");
  1197. // if ($check) {
  1198. // return json([
  1199. // 'code' => 1,
  1200. // 'msg' => '监听进程已存在,请勿重复启动'
  1201. // ]);
  1202. // }
  1203. // // 启动监听
  1204. // $command = 'nohup php think queue:listen --queue --timeout=300 --sleep=3 --memory=256 > /var/log/job_queue.log 2>&1 &';
  1205. // exec($command, $output, $status);
  1206. // if ($status === 0) {
  1207. // return json([
  1208. // 'code' => 0,
  1209. // 'msg' => '队列监听已启动'
  1210. // ]);
  1211. // } else {
  1212. // return json([
  1213. // 'code' => 1,
  1214. // 'msg' => '队列启动失败',
  1215. // 'output' => $output
  1216. // ]);
  1217. // }
  1218. // }
  1219. /**
  1220. * 新增模版
  1221. */
  1222. public function Add_Product_Template(){
  1223. try {
  1224. $params = $this->request->param();
  1225. $chinese_description = $params['chinese_description'] ?? '';
  1226. $template_image_url = $params['template_image_url'] ?? '';
  1227. $template_name = $params['template_name'] ?? '';
  1228. $size = $params['size'] ?? '';
  1229. $style = $params['style'] ?? '';
  1230. $type = $params['type'] ?? '';
  1231. $user_id = $params['user_id'] ?? '';
  1232. $seconds = $params['seconds'] ?? '';
  1233. $video_id = $params['video_id'] ?? '';
  1234. if (empty($template_name)) {
  1235. return json(['code' => 1, 'msg' => '模板名称不能为空']);
  1236. }
  1237. $data = [
  1238. 'toexamine' => "未审核",
  1239. 'chinese_description' => $chinese_description,
  1240. 'template_image_url' => $template_image_url,
  1241. 'template_name' => $template_name,
  1242. 'type' => $type,
  1243. 'style' => $style,
  1244. 'seconds' => $seconds,
  1245. 'size' => $size,
  1246. 'video_id' => $video_id,
  1247. 'user_id' => $user_id,
  1248. 'sys_rq' => date('Y-m-d'),
  1249. 'create_time' => date('Y-m-d H:i:s')
  1250. ];
  1251. // echo "<pre>";
  1252. // print_r($data);
  1253. // echo "<pre>";die;
  1254. $result = Db::name('product_template')->insert($data);
  1255. if ($result) {
  1256. return json(['code' => 0, 'msg' => '模板保存成功']);
  1257. } else {
  1258. return json(['code' => 1, 'msg' => '模板保存失败: 数据库操作未影响任何行']);
  1259. }
  1260. } catch (\Exception $e) {
  1261. Log::record('模板保存异常: ' . $e->getMessage(), 'error');
  1262. Log::record('异常堆栈: ' . $e->getTraceAsString(), 'error');
  1263. return json(['code' => 1, 'msg' => '模板保存失败: ' . $e->getMessage()]);
  1264. }
  1265. }
  1266. /**
  1267. * 查询模版
  1268. */
  1269. public function product_template()
  1270. {
  1271. $params = $this->request->param();
  1272. // 构建查询条件
  1273. $where = [];
  1274. if (!empty($params['search'])) {
  1275. $where['chinese_description|template_name|style'] = ['like', '%' . $params['search'] . '%'];
  1276. }
  1277. if (!empty($params['sys_id'])) {
  1278. $where['sys_id'] = ['like', '%' . $params['sys_id'] . '%'];
  1279. $products = Db::name('product_template')->order('id desc')->where($where)
  1280. ->whereNull('mod_rq')
  1281. ->select();
  1282. }else{
  1283. $products = Db::name('product_template')->order('id desc')->where($where)
  1284. ->where('release','1')
  1285. ->whereNull('mod_rq')
  1286. ->select();
  1287. }
  1288. // $http_url = Db::name('http_url')->field('baseUrl,port')->find();
  1289. // if ($products && $http_url) {
  1290. // $base_url = !empty($http_url['baseUrl']) && !empty($http_url['port'])
  1291. // ? 'http://' . $http_url['baseUrl'] . ':' . $http_url['port'] : '';
  1292. // if ($base_url) {
  1293. // foreach ($products as &$val) {
  1294. // $val['template_image_url'] = $base_url . $val['template_image_url'];
  1295. // }
  1296. // }
  1297. // }
  1298. return json([
  1299. 'code' => 0,
  1300. 'msg' => '请求成功',
  1301. 'data' => $products
  1302. ]);
  1303. }
  1304. /**
  1305. *获取服务器URL地址和端口 IP地址:端口
  1306. * 用于获取图片路径拼接时
  1307. **/
  1308. public function GetHttpUrl(){
  1309. $data = Db::name('http_url')->find();
  1310. $fullUrl = "http://" . $data['baseUrl'] . ":" . $data['port'];
  1311. $res = [
  1312. 'code' => 0,
  1313. 'msg' => '成功',
  1314. 'data' => [
  1315. 'id' => $data['id'],
  1316. 'full_url' => $fullUrl,
  1317. 'baseUrl' => $data['baseUrl'],
  1318. 'port' => $data['port']
  1319. ]
  1320. ];
  1321. return json($res);
  1322. }
  1323. }