nstream.adapter.redis.scan.RedisKeyIteratingScanner Maven / Gradle / Ivy

// Copyright 2015-2024 Nstream, inc.
//
// Licensed under the Redis Source Available License 2.0 (RSALv2) Agreement;
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://redis.com/legal/rsalv2-agreement/
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package nstream.adapter.redis.scan;

import java.util.function.Consumer;
import redis.clients.jedis.JedisPooled;
import redis.clients.jedis.params.ScanParams;
import redis.clients.jedis.resps.ScanResult;
import swim.structure.Value;

/**
 * Scanner iterating over a keyspace and processing each key individually.
 */
public class RedisKeyIteratingScanner extends RedisScanner {

  private final RedisKeyScan keyScan;

  public RedisKeyIteratingScanner(final JedisPooled jedis, final RedisValueGetter redisValueGetter, final RedisKeyScan redisKeyScan) {
    super(jedis, redisValueGetter);
    this.keyScan = redisKeyScan;
  }

  @Override
  public void scan(final Consumer valueConsumer) {
    String cursor = ScanParams.SCAN_POINTER_START;
    do {
      final ScanResult result = this.keyScan.scan(this.jedis, cursor);
      cursor = result.getCursor();
      result.getResult().forEach(k -> scanKey(k, valueConsumer));
    } while (!ScanParams.SCAN_POINTER_START.equals(cursor));
  }

}