org.springframework.data.aerospike.query.cache.IndexRefresher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-data-aerospike Show documentation

There is a newer version: 4.8.0

Show newest version

/*
 * Copyright 2020 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * 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 org.springframework.data.aerospike.query.cache;

import com.aerospike.client.IAerospikeClient;
import com.aerospike.client.Info;
import com.aerospike.client.cluster.Node;
import com.aerospike.client.policy.InfoPolicy;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.aerospike.query.model.IndexesInfo;

import java.util.Arrays;

/**
 * @author Anastasiia Smirnova
 */
public class IndexRefresher {

    private final Logger log = LoggerFactory.getLogger(IndexRefresher.class);

    private final IAerospikeClient client;
    private final InfoPolicy infoPolicy;
    private final InternalIndexOperations indexOperations;
    private final IndexesCacheUpdater indexesCacheUpdater;

    public IndexRefresher(IAerospikeClient client, InfoPolicy infoPolicy,
                          InternalIndexOperations indexOperations, IndexesCacheUpdater indexesCacheUpdater) {
        this.client = client;
        this.infoPolicy = infoPolicy;
        this.indexOperations = indexOperations;
        this.indexesCacheUpdater = indexesCacheUpdater;
    }

    public void refreshIndexes() {
        log.trace("Loading indexes");
        IndexesInfo cache = Arrays.stream(client.getNodes())
            .filter(Node::isActive)
            .findAny() // we do want to send info request to the random node (sending request to the first node may
            // lead to uneven request distribution)
            .map(node -> Info.request(infoPolicy, node, indexOperations.buildGetIndexesCommand()))
            .map(indexOperations::parseIndexesInfo)
            .orElse(IndexesInfo.empty());
        log.debug("Loaded indexes: {}", cache.indexes);
        this.indexesCacheUpdater.update(cache);
    }
}