From 1ce011e08b20e3e9391287bf5e7ceceb7f81b017 Mon Sep 17 00:00:00 2001
From: cody <648753004@qq.com>
Date: Thu, 27 Nov 2025 14:03:27 +0800
Subject: [PATCH] update

---
 app/Console/Commands/UpdateCompany.php | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/app/Console/Commands/UpdateCompany.php b/app/Console/Commands/UpdateCompany.php
index d281d5a..b5ada7d 100755
--- a/app/Console/Commands/UpdateCompany.php
+++ b/app/Console/Commands/UpdateCompany.php
@@ -43,8 +43,8 @@ class UpdateCompany extends Command
     public function handle()
     {
         $user_id = $this->option('user_id');
-        $updateLocal = (int) $this->option('address');
-        $updateMarket = (int) $this->option('market');
+        $updateLocal = (int)$this->option('address');
+        $updateMarket = (int)$this->option('market');
         // 更新公司信息
         $this->compnay($user_id);
         // 更新经纬度信息（可选）
@@ -142,6 +142,8 @@ class UpdateCompany extends Command
             // 更新用户关联
             $user->company_id = $company->id;
             $user->save();
+            // 更新上市状态
+            $this->updateMarketStatus($company->id);
             $bar->setMessage($result['enterpriseName'] . ' 更新成功', 'status');
             $bar->advance();
         }
@@ -203,14 +205,18 @@ class UpdateCompany extends Command
      * 根据 company_tag 更新上市状态
      * 判断是否包含上市代码标签，如 688001.SH、000001.SZ、830001.BJ 等
      */
-    public function updateMarketStatus()
+    public function updateMarketStatus($companyId = null)
     {
         $this->info('开始更新上市状态...');
 
         // 获取所有有 company_tag 的公司
-        $companies = Company::whereNotNull('company_tag')
-            ->where('company_tag', '!=', '')
-            ->get();
+        if ($companyId) {
+            $companies = Company::where('id', $companyId)->get();
+        } else {
+            $companies = Company::whereNotNull('company_tag')
+                ->where('company_tag', '!=', '')
+                ->get();
+        }
 
         $total = $companies->count();
         if ($total == 0) {