update_chart_logic.go 4.4 KB
package chart

import (
	"context"
	"encoding/json"
	"github.com/samber/lo"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/interanl/pkg/db/transaction"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/interanl/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/pkg/contextdata"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/pkg/xerr"

	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/api/internal/svc"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateChartLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewUpdateChartLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateChartLogic {
	return &UpdateChartLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UpdateChartLogic) UpdateChart(req *types.ChartUpdateRequest) (resp *types.ChartUpdateResponse, err error) {
	var (
		conn         = l.svcCtx.DefaultDBConn()
		chart        *domain.Chart
		chartSetting *domain.ChartSetting
		tenantId     = contextdata.GetTenantFromCtx(l.ctx)
	)

	if chart, err = l.svcCtx.ChartRepository.FindOne(l.ctx, conn, req.Id); err != nil || chart.TenantId != tenantId {
		return nil, xerr.NewErrMsgErr("图表不存在", err)
	}
	if chartSetting, err = l.svcCtx.ChartSettingRepository.FindOne(l.ctx, conn, req.Id); err != nil {
		return nil, xerr.NewErrMsgErr("图表配置不存在", err)
	}
	oldDataSource := chartSetting.DataSourceIds

	if err = transaction.UseTrans(l.ctx, l.svcCtx.DB, func(ctx context.Context, conn transaction.Conn) error {
		if len(req.Cover) > 0 && chart.Cover != req.Cover {
			chart.Cover = req.Cover
		}
		if chart, err = l.svcCtx.ChartRepository.UpdateWithVersion(l.ctx, conn, chart); err != nil {
			return err
		}
		chartProperty := types.NewProperty(req.ChartProperty)
		chartSetting.Title = chartProperty.Title
		chartSetting.TableAbility = chartProperty.TableAbility
		chartSetting.Series = chartProperty.Series
		chartSetting.Other = chartProperty.Other
		chartSetting.DataSourceIds = chartProperty.GetAllDataSourceId()
		if chartSetting, err = l.svcCtx.ChartSettingRepository.UpdateWithVersion(l.ctx, conn, chartSetting); err != nil {
			return err
		}
		err = l.SyncTableData(tenantId, conn, chartSetting.DataSourceIds, oldDataSource)
		return err
	}, true); err != nil {
		return nil, xerr.NewErrMsgErr("创建失败", err)
	}
	return
}

func (l *UpdateChartLogic) SyncTableData(tenantId int64, conn transaction.Conn, newDataSourceIds, oldDataSourceIds []int64) error {
	if len(newDataSourceIds) > 0 {
		for _, sourceId := range newDataSourceIds {
			pusher := &types.SyncTableDataPusher{
				CompanyId: tenantId,
				ObjectId:  int(sourceId),
			}
			mBytes, _ := json.Marshal(pusher)
			_, _ = l.svcCtx.Redis.Lpush(l.svcCtx.Config.Name+":table_data", string(mBytes))
		}
	}
	//对比更新前后数据源
	left, right := lo.Difference(newDataSourceIds, oldDataSourceIds)
	//同步
	if len(left) > 0 {
		for _, sourceId := range left {
			pusher := &types.SyncTableDataPusher{
				CompanyId: tenantId,
				ObjectId:  int(sourceId),
			}
			mBytes, _ := json.Marshal(pusher)
			_, _ = l.svcCtx.Redis.Lpush(l.svcCtx.Config.Name+":table_data", string(mBytes))
		}
	}
	//删除
	if len(right) > 0 {
		for _, sourceId := range right {
			err := DeleteObjectTable(l.ctx, l.svcCtx, conn, int(sourceId))
			if err != nil {
				return err
			}
		}
	}
	return nil
}

func DeleteObjectTable(ctx context.Context, svcCtx *svc.ServiceContext, conn transaction.Conn, sourceId int) error {
	//验证其他图表是否使用
	used, err := svcCtx.ChartSettingRepository.CheckUseDataSource(ctx, conn, sourceId)
	if err == nil && !used { //未使用,删除
		err = svcCtx.ObjectTableDataRepository.DropTable(ctx, conn, sourceId)
		if err != nil {
			return err
		}
		//是否远程已删除
		objectTable, err := svcCtx.ObjectTableRepository.FindOne(ctx, conn, sourceId)
		if err == nil && objectTable.Id > 0 && objectTable.RemoteDeleted == 1 {
			//删除表
			_, err = svcCtx.ObjectTableRepository.Delete(ctx, conn, &domain.ObjectTable{Id: sourceId})
			if err != nil {
				return err
			}
			//删除字段
			_, err = svcCtx.ObjectFieldRepository.Delete(ctx, conn, &domain.ObjectField{Id: int64(sourceId)})
			if err != nil {
				return err
			}
		}
	}
	return nil
}