fix

controller/contract/register.go

@@ -275,7 +275,7 @@ func (rg *RegisterController) Add(c *gin.Context) {
 	// 新增合同及套餐
 	if req.Supplement == 1 {
 		//新增合同信息
-		if e = fms.CreateContractRegisterAndServicesAndPayMent(ob, serviceList); e != nil {
+		if e = fms.CreateContractRegisterAndServicesAndPayMent(ob, serviceList, req.PrePayId); e != nil {
 			resp.FailMsg("操作失败", "新增合同及套餐失败, Err: "+e.Error(), c)
 			return
 		}

models/fms/contract_register.go

@@ -429,7 +429,7 @@ func CreateImportContractRegister(item *ContractRegister, serviceList []*Contrac
 }
 
 // CreateContractRegisterAndServices 新增合同登记及套餐
-func CreateContractRegisterAndServicesAndPayMent(item *ContractRegister, serviceDetail []*ContractServiceAndDetail) (err error) {
+func CreateContractRegisterAndServicesAndPayMent(item *ContractRegister, serviceDetail []*ContractServiceAndDetail, prePayId int) (err error) {
 	tx := global.DEFAULT_MYSQL.Begin()
 	defer func() {
 		if err != nil {
@@ -472,7 +472,7 @@ func CreateContractRegisterAndServicesAndPayMent(item *ContractRegister, service
 		//}
 	}
 
-	sql := `UPDATE contract_pre_payment SET contract_register_id = ?, contract_code = ?, modify_time = NOW()`
-	tx.Exec(sql, item.ContractRegisterId, item.ContractCode)
+	sql := `UPDATE contract_pre_payment SET contract_register_id = ?, contract_code = ?, modify_time = NOW() WHERE pre_pay_id = ?`
+	tx.Exec(sql, item.ContractRegisterId, item.ContractCode, prePayId)
 	return
 }