search_table_data_logic.go 3.7 KB
package table

import (
	"context"
	"github.com/jinzhu/copier"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/interanl/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/interanl/pkg/gateway/bytelib"
	"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 SearchTableDataLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

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

func (l *SearchTableDataLogic) SearchTableData(req *types.SearchTableDataRequest) (resp interface{}, err error) {
	if req.ObjectId == 0 {
		resp = map[string]interface{}{
			"objectId": 0,
			"fields":   nil,
			"total":    0,
			"data":     make([]string, 0),
			"local":    false,
		}
		return
	}
	resp, err = l.getLocal(req)
	if err == nil {
		return resp, nil
	}
	return l.getRemote(req)
}

// getLocal 获取本地数据
func (l *SearchTableDataLogic) getLocal(req *types.SearchTableDataRequest) (interface{}, error) {
	conn := l.svcCtx.DefaultDBConn()
	conditions := make([]*domain.TableDataCondition, 0)
	_ = copier.Copy(&conditions, req.Condition)
	//查询表数据
	objectTable, err := l.svcCtx.ObjectTableRepository.FindOneByTableId(l.ctx, conn, req.ObjectId)
	if err != nil || objectTable.Id <= 0 {
		return nil, xerr.NewErrMsg("表不存在")
	}
	if !objectTable.IsLocal {
		return nil, xerr.NewErrMsg("未保存到本地存储")
	}
	//查询表字段
	objectField, err := l.svcCtx.ObjectFieldRepository.FindOne(l.ctx, conn, int64(req.ObjectId))
	if err != nil || objectField.Id <= 0 {
		return nil, xerr.NewErrMsg("表字段不存在")
	}
	//替换中文字段
	for _, item := range conditions {
		replace := false
		for _, field := range objectField.Fields {
			if field.Name == item.FieldName {
				item.FieldName = field.SQLName
				replace = true
			}
		}
		if !replace {
			return nil, xerr.NewErrMsg("字段" + item.FieldName + "不存在")
		}
	}
	//表数据
	total, list, err := l.svcCtx.ObjectTableDataRepository.Find(l.ctx, conn, req.ObjectId, &domain.ObjectTableDataQuery{
		Page:       req.PageNumber,
		Size:       req.PageSize,
		Conditions: conditions,
	})
	if err != nil {
		return nil, xerr.NewErrMsg("查询表数据失败")
	}
	return map[string]interface{}{
		"objectId": req.ObjectId,
		"fields":   removeIdField(objectField.Fields),
		"total":    total,
		"data":     list,
		"local":    true,
	}, nil
}

// getRemote 获取远程字库数据
func (l *SearchTableDataLogic) getRemote(req *types.SearchTableDataRequest) (resp interface{}, err error) {
	tableDataPreviewRequest := &bytelib.TableDataPreviewRequest{
		Token:      req.Token,
		ObjectType: bytelib.ObjectMetaTable,
		ObjectId:   int64(req.ObjectId),
		Where:      newWhere(req.Condition).WithPageSize(req.PageNumber, req.PageSize),
	}
	var (
		response bytelib.TablePreviewResponse
	)
	response, err = l.svcCtx.ByteMetadataService.TableDataPreview(l.ctx, tableDataPreviewRequest)
	if err != nil {
		return resp, xerr.NewErr(err)
	}
	resp = map[string]interface{}{
		"objectId": response.ObjectId,
		"fields":   removeIdField(response.Fields),
		"total":    response.Grid.Total,
		"data":     response.Grid.List,
		"local":    false,
	}
	return
}

func removeIdField(fields []*bytelib.Field) []*bytelib.Field {
	var result = make([]*bytelib.Field, 0)
	for _, f := range fields {
		if f.SQLName == "id" {
			continue
		}
		result = append(result, f)
	}
	return result
}