hongze 1 年之前
父節點
當前提交
4b9a14dfb3
共有 1 個文件被更改,包括 11 次插入1 次删除
  1. 11 1
      models/overseas_custom/overseas_custom_seller.go

+ 11 - 1
models/overseas_custom/overseas_custom_seller.go

@@ -2,8 +2,10 @@ package overseas_custom
 
 import (
 	"context"
+	"fmt"
 	"github.com/beego/beego/v2/client/orm"
 	"hongze/hz_crm_api/models/company"
+	"hongze/hz_crm_api/utils"
 	"time"
 )
 
@@ -99,6 +101,13 @@ type EnglishCompany struct {
 }
 
 func (m *OverseasCustomerSale) HandelCompanyStatus(sellerId int) (err error) {
+	var databaseName string
+	if utils.RunMode == "debug" {
+		databaseName = "test_v2_hongze_rddp"
+	} else {
+		databaseName = "hongze_rddp"
+	}
+
 	o := orm.NewOrm()
 	tx, err := o.BeginWithCtx(context.TODO())
 	if err != nil {
@@ -122,7 +131,8 @@ func (m *OverseasCustomerSale) HandelCompanyStatus(sellerId int) (err error) {
 				overseasStatus = "试用"
 				resetBtn = 1
 			}
-			sql = ` UPDATE test_v2_hongze_rddp.english_company SET overseas_status=?,reset_btn=? WHERE company_id=? `
+			sql = ` UPDATE %s.english_company SET overseas_status=?,reset_btn=? WHERE company_id=? `
+			sql = fmt.Sprintf(sql, databaseName)
 			_, err = tx.Raw(sql, overseasStatus, resetBtn, v.CompanyId).Exec()
 			if err != nil {
 				return err