|
1
|
package repository
|
1
|
package repository
|
|
2
|
|
2
|
|
|
3
|
import (
|
3
|
import (
|
|
|
|
4
|
+ "errors"
|
|
4
|
"fmt"
|
5
|
"fmt"
|
|
5
|
|
6
|
|
|
6
|
"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/domain"
|
7
|
"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/domain"
|
|
@@ -86,15 +87,24 @@ func (reponsitory CompanyRepository) Edit(m *domain.Company) error { |
|
@@ -86,15 +87,24 @@ func (reponsitory CompanyRepository) Edit(m *domain.Company) error { |
|
86
|
|
87
|
|
|
87
|
func (reponsitory CompanyRepository) FindOne(queryOptions domain.CompanyFindOneOptions) (domain.Company, error) {
|
88
|
func (reponsitory CompanyRepository) FindOne(queryOptions domain.CompanyFindOneOptions) (domain.Company, error) {
|
|
88
|
var (
|
89
|
var (
|
|
89
|
- err error
|
|
|
|
90
|
- tx = reponsitory.transactionContext.PgTx
|
|
|
|
91
|
- m models.Company
|
90
|
+ err error
|
|
|
|
91
|
+ tx = reponsitory.transactionContext.PgTx
|
|
|
|
92
|
+ m models.Company
|
|
|
|
93
|
+ hasCondition bool
|
|
92
|
)
|
94
|
)
|
|
93
|
|
95
|
|
|
94
|
query := tx.Model(&m)
|
96
|
query := tx.Model(&m)
|
|
95
|
if queryOptions.Id > 0 {
|
97
|
if queryOptions.Id > 0 {
|
|
|
|
98
|
+ hasCondition = true
|
|
96
|
query = query.Where("id=?", queryOptions.Id)
|
99
|
query = query.Where("id=?", queryOptions.Id)
|
|
97
|
}
|
100
|
}
|
|
|
|
101
|
+ if queryOptions.AdminCompanyId > 0 {
|
|
|
|
102
|
+ hasCondition = true
|
|
|
|
103
|
+ query = query.Where("admin_company_id=?", queryOptions.Id)
|
|
|
|
104
|
+ }
|
|
|
|
105
|
+ if !hasCondition {
|
|
|
|
106
|
+ return domain.Company{}, errors.New("findOne 必须要有查询条件")
|
|
|
|
107
|
+ }
|
|
98
|
err = query.First()
|
108
|
err = query.First()
|
|
99
|
if err != nil {
|
109
|
if err != nil {
|
|
100
|
return domain.Company{}, err
|
110
|
return domain.Company{}, err
|