71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"electricity_bill_calc/logger"
|
|
"fmt"
|
|
"github.com/doug-martin/goqu/v9"
|
|
"github.com/jackc/pgx/v5"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type _GMRepository struct {
|
|
log *zap.Logger
|
|
ds goqu.DialectWrapper
|
|
}
|
|
|
|
var GMRepository = &_GMRepository{
|
|
log: logger.Named("Repository", "GM"),
|
|
ds: goqu.Dialect("postgres"),
|
|
}
|
|
|
|
func (gm _GMRepository) DeleteMeterBinding(ctx context.Context, tx pgx.Tx, pid string, tenements []string, meterCodes ...[]string) error {
|
|
DeleteQuery := gm.ds.From(goqu.T("tenement_meter")).
|
|
Where(goqu.I("park_id").Eq(pid)).
|
|
Delete()
|
|
|
|
if len(tenements) > 0 {
|
|
DeleteQuery = DeleteQuery.
|
|
Where(goqu.I("tenement_id").In(tenements))
|
|
}
|
|
|
|
if len(meterCodes) > 0 {
|
|
DeleteQuery = DeleteQuery.
|
|
Where(goqu.I("meter_id").In(meterCodes))
|
|
}
|
|
|
|
DeleteQuerySql, DeleteQueryArgs, _ := DeleteQuery.ToSQL()
|
|
|
|
|
|
_, err := tx.Exec(ctx, DeleteQuerySql, DeleteQueryArgs...)
|
|
if err != nil {
|
|
gm.log.Error("数据库在删除tenement_meter表数据中出错", zap.Error(err))
|
|
tx.Rollback(ctx)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (gm _GMRepository) DeleteTenements(ctx context.Context, tx pgx.Tx, pid string, tenements []string) error {
|
|
DeleteTenements := gm.ds.
|
|
From("tenement").
|
|
Where(goqu.I("park_id").Eq(pid)).
|
|
Delete()
|
|
|
|
fmt.Println(len(tenements))
|
|
if len(tenements) > 0 {
|
|
DeleteTenements = DeleteTenements.
|
|
Where(goqu.I("id").In(tenements))
|
|
}
|
|
|
|
DeleteTenementsSql, DeleteTenementsArgs, _ := DeleteTenements.ToSQL()
|
|
|
|
_, err := tx.Exec(ctx, DeleteTenementsSql, DeleteTenementsArgs...)
|
|
if err != nil {
|
|
tx.Rollback(ctx)
|
|
gm.log.Error("删除商户信息出错",zap.Error(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|