Method QueryAttributes

internal/storage/memory/data_reader.go:201–271

QueryAttributes queries the database for attributes based on the provided filter.

(_ context.Context, tenantID string, filter *base.AttributeFilter, _ string, pagination database.CursorPagination)

// QueryAttributes queries the database for attributes based on the provided filter.
func (r *DataReader) QueryAttributes(_ context.Context, tenantID string, filter *base.AttributeFilter, _ string, pagination database.CursorPagination) (iterator *database.AttributeIterator, err error) {
	txn := r.database.DB.Txn(false)
	defer txn.Abort()

	var lowerBound string

	if pagination.Cursor() != "" {
		var t database.ContinuousToken
		t, err = utils.EncodedContinuousToken{Value: pagination.Cursor()}.Decode()
		if err != nil {
			return nil, err
		}
		lowerBound = t.(utils.ContinuousToken).Value
	}

	// Get the index and arguments based on the filter.
	index, args := utils.GetAttributesIndexNameAndArgsByFilters(tenantID, filter)

	// Get the result iterator based on the index and arguments.
	var result memdb.ResultIterator
	result, err = txn.Get(constants.AttributesTable, index, args...) // Query attributes table
	if err != nil {
		return nil, errors.New(base.ErrorCode_ERROR_CODE_EXECUTION.String())
	}

	// Filter the result iterator and add the attributes to the collection.
	attr := make([]storage.Attribute, 0, 10)
	fit := memdb.NewFilterIterator(result, utils.FilterAttributesQuery(tenantID, filter))
	for obj := fit.Next(); obj != nil; obj = fit.Next() {
		t, ok := obj.(storage.Attribute)
		if !ok {
			return nil, errors.New(base.ErrorCode_ERROR_CODE_TYPE_CONVERSATION.String())
		}
		attr = append(attr, t)
	}

	// Sort attributes based on the provided order field
	sort.Slice(attr, func(i, j int) bool {
		switch pagination.Sort() {
		case "entity_id":
			return attr[i].EntityID < attr[j].EntityID
		default:
			return false // No sorting if order field is invalid
		}
	})

	var attrs []*base.Attribute
	count := uint32(0)
	limit := pagination.Limit()

	for _, t := range attr {
		// Skip attributes below the lower bound
		switch pagination.Sort() {
		case "entity_id":
			if t.EntityID < lowerBound {
				continue
			}
		}