search_table_data_logic.go 1.8 KB
package table

import (
	"context"
	"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),
		}
		return
	}
	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,
	}
	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
}