删除指定的企业用户完成

controller/god_mode.go

@@ -17,7 +17,8 @@ func InitializeGmController(router *fiber.App) {
 	router.Delete("/gm/tenement", security.SingularityAuthorize, deleteTenement)
 	router.Delete("/gm/park", security.SingularityAuthorize, deletePark)
 	router.Delete("/gm/report", security.SingularityAuthorize, deleteReports)
-	router.Delete("/gm/tenement/meter",security.SingularityAuthorize, deleteTenementMeterRelations)
+	router.Delete("/gm/tenement/meter", security.SingularityAuthorize, deleteTenementMeterRelations)
+	router.Delete("/gm/enterprise", security.SingularityAuthorize, deleteEnterprise)
 }
 
 //用于将参数转化为切片
@@ -78,15 +79,22 @@ func deleteReports(c *fiber.Ctx) error {
 }
 
 func deleteEnterprise(c *fiber.Ctx) error {
+	uid := c.Query("uid")
+	result := response.NewResult(c)
+	GmLog.Info("[天神模式]删除指定企业用户", zap.String("uid", uid))
 
-	return nil
+	err := service.GMService.DeleteEnterprises(uid)
+	if err != nil {
+		GmLog.Error("[天神模式]删除指定企业用户失败", zap.Error(err))
+		return result.Error(500, "删除指定企业用户失败。")
+	}
+	return result.Success("指定企业用户已经删除。")
 }
 
-
 func deleteTenementMeterRelations(c *fiber.Ctx) error {
 	result := response.NewResult(c)
 	parkId := c.Query("park")
-	tId := getQueryValues(c,"tenements")
+	tId := getQueryValues(c, "tenements")
 	metersId := getQueryValues(c, "meters")
 	GmLog.Info("删除指定园区中的商户与表计的关联关系", zap.String("park id", parkId))
 	if err := service.GMService.DeleteTenementMeterRelations(parkId, tId, metersId); err != nil {

repository/god_mode.go

@@ -2,9 +2,11 @@ package repository
 
 import (
 	"context"
+	"electricity_bill_calc/global"
 	"electricity_bill_calc/logger"
 	"fmt"
 	"github.com/doug-martin/goqu/v9"
+	"github.com/georgysavva/scany/v2/pgxscan"
 	"github.com/jackc/pgx/v5"
 	"go.uber.org/zap"
 )
@@ -125,7 +127,7 @@ func (gm _GMRepository) DeleteInvoices(ctx context.Context, tx pgx.Tx, parks str
 
 }
 
-func (gm _GMRepository) DeleteMeterPookings(ctx context.Context, tx pgx.Tx, parks string, val ...[]string) error {
+func (gm _GMRepository) DeleteMeterPoolings(ctx context.Context, tx pgx.Tx, parks string, val ...[]string) error {
 	deleteQuery := gm.ds.
 		Delete(goqu.T("meter_relations")).
 		Where(goqu.I("park_id").Eq(parks))
@@ -390,3 +392,58 @@ func (gm _GMRepository) DeleteParks(ctx context.Context, tx pgx.Tx, park []strin
 	}
 	return nil
 }
+
+func (gm _GMRepository) ListAllParkIdsInUser(ctx context.Context, tx pgx.Tx, uid string) ([]string, error) {
+	SearchParkIdsSql, SearchParkIdsArgs, _ := gm.ds.
+		From(goqu.T("park")).
+		Where(goqu.I("user_id").Eq(uid)).
+		Select(goqu.I("id")).ToSQL()
+	var pids []string
+	err := pgxscan.Select(ctx, global.DB, &pids, SearchParkIdsSql, SearchParkIdsArgs...)
+	if err != nil {
+		gm.log.Error("查询["+uid+"]用户下的所有园区失败", zap.Error(err))
+		tx.Rollback(ctx)
+		return nil, err
+	}
+
+	return pids, nil
+}
+
+func (gm _GMRepository) DeleteUsers(ctx context.Context, tx pgx.Tx, uid string) error {
+	var err error
+	//删除用户关联
+	DeleteUserChargeSql, DeleteUserChargeArgs, _ := gm.ds.
+		Delete(goqu.T("user_charge")).
+		Where(goqu.I("id").Eq(uid)).ToSQL()
+
+	_, err = tx.Exec(ctx,DeleteUserChargeSql,DeleteUserChargeArgs...)
+	if err != nil {
+		gm.log.Error("user_charge表关联出错",zap.Error(err))
+		tx.Rollback(ctx)
+		return err
+	}
+
+	//删除用户详细信息
+	DeleteUserDetailSql, DeleteUserDetailArgs,_ := gm.ds.
+		Delete(goqu.T("user_detail")).
+		Where(goqu.I("id").Eq(uid)).ToSQL()
+	_, err = tx.Exec(ctx,DeleteUserDetailSql,DeleteUserDetailArgs...)
+	if err != nil {
+		gm.log.Error("user_detail表详细信息出错",zap.Error(err))
+		tx.Rollback(ctx)
+		return err
+	}
+
+	//删除用户基础信息
+	DeleteUserSql, DeleteUserArgs,_ := gm.ds.
+		Delete(goqu.T("users")).
+		Where(goqu.I("id").Eq(uid)).ToSQL()
+	_, err = tx.Exec(ctx,DeleteUserSql,DeleteUserArgs...)
+	if err != nil {
+		gm.log.Error("user表基础信息出错",zap.Error(err))
+		tx.Rollback(ctx)
+		return err
+	}
+
+	return nil
+}

service/god_mode.go

@@ -61,8 +61,8 @@ func (gm _GMService) DeleteParks(parks []string) error {
 		err = repository.GMRepository.DeleteInvoices(ctx, tx, pid)
 		//删除meter_binding
 		err = repository.GMRepository.DeleteMeterBinding(ctx, tx, pid, []string{})
-		//删除meter_pookings
-		err = repository.GMRepository.DeleteMeterPookings(ctx, tx, pid)
+		//删除meter_poolings
+		err = repository.GMRepository.DeleteMeterPoolings(ctx, tx, pid)
 		//删除tenements
 		err = repository.GMRepository.DeleteTenements(ctx, tx, pid, []string{})
 		//删除meters
@@ -128,3 +128,50 @@ func (gm _GMService) DeleteTenementMeterRelations(pId string, tId []string, mId
 	}
 	return nil
 }
+
+func (gm _GMService) DeleteEnterprises(uid string) error {
+	var err error
+	ctx, cancel := global.TimeoutContext()
+	defer cancel()
+
+	tx, err := global.DB.Begin(ctx)
+	if err != nil {
+		gm.l.Error("未能启动数据库事务", zap.Error(err))
+		return fmt.Errorf("未能启动数据库事务，%w", err)
+	}
+	parks, err := repository.GMRepository.ListAllParkIdsInUser(ctx, tx, uid)
+	if err != nil {
+		gm.l.Error("查询园区错误", zap.Error(err))
+		tx.Rollback(ctx)
+		return err
+	}
+
+	for _, pid := range parks {
+		err = repository.GMRepository.DeleteInvoices(ctx, tx, pid)
+		err = repository.GMRepository.DeleteMeterBinding(ctx, tx, pid, []string{})
+		err = repository.GMRepository.DeleteMeterPookings(ctx, tx, pid)
+		err = repository.GMRepository.DeleteTenements(ctx, tx, pid)
+		err = repository.GMRepository.DeleteMeters(ctx, tx, pid)
+		err = repository.GMRepository.DeleteReports(ctx, tx, pid)
+		err = repository.GMRepository.DeleteBuildings(ctx, tx, pid)
+
+		if err != nil {
+			gm.l.Error("删除用户下关联出错", zap.Error(err))
+			return err
+		}
+	}
+
+	err = repository.GMRepository.DeleteParks(ctx, tx, parks)
+	if err != nil {
+		gm.l.Error("删除用户关联园区错误", zap.Error(err))
+		tx.Rollback(ctx)
+		return err
+	}
+	err = repository.GMRepository.DeleteUsers(ctx, tx, uid)
+	if err != nil {
+		gm.l.Error("删除用户信息出错", zap.Error(err))
+		tx.Rollback(ctx)
+		return err
+	}
+	return nil
+}