frostfs-sdk-csharp/src/FrostFS.SDK.ClientV2/Tools/SearchReader.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;

using Grpc.Core;

using FrostFS.Object;
using FrostFS.Refs;
using System.Threading;

namespace FrostFS.SDK.ClientV2;

internal class SearchReader(AsyncServerStreamingCall<SearchResponse> call) : IDisposable
{
    public AsyncServerStreamingCall<SearchResponse> Call { get; private set; } = call;

    public async Task<List<ObjectID>?> Read(CancellationToken cancellationToken)
    {
        if (!await Call.ResponseStream.MoveNext(cancellationToken))
            return null;
       
        var response = Call.ResponseStream.Current;
        
        Verifier.CheckResponse(response);

        return response.Body?.IdList.ToList();
    }

    public void Dispose()
    {
        Call.Dispose();
    }
}