com.yahoo.vespa.hosted.provision.applications.Applications Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of node-repository Show documentation
Keeps track of node assignment in a multi-application setup.
The newest version!
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.applications;

import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.ApplicationLockException;
import com.yahoo.config.provision.ApplicationTransaction;
import com.yahoo.config.provision.ApplicationMutex;
import com.yahoo.transaction.Mutex;
import com.yahoo.transaction.NestedTransaction;
import com.yahoo.vespa.hosted.provision.persistence.CuratorDb;

import java.time.Duration;
import java.util.List;
import java.util.Optional;

/**
 * An (in-memory, for now) repository of the node repo's view of applications.
 *
 * This is multithread safe.
 *
 * @author bratseth
 */
public class Applications {

    private final CuratorDb db;

    public Applications(CuratorDb db) {
        this.db = db;
        // read and write all to make sure they are stored in the latest version of the serialized format
        for (ApplicationId id : ids()) {
            try (Mutex lock = db.lock(id)) {
                get(id).ifPresent(application -> put(application, lock));
            } catch (ApplicationLockException e) {
                throw new ApplicationLockException(e);
            }
        }
    }

    /** Returns the ids of all applications */
    public List ids() { return db.readApplicationIds(); }

    /** Returns the application with the given id, or empty if it does not exist */
    public Optional get(ApplicationId id) {
        return db.readApplication(id);
    }

    /** Returns the application with the given id, or throws IllegalArgumentException if it does not exist */
    public Application require(ApplicationId id) {
        return db.readApplication(id).orElseThrow(() -> new IllegalArgumentException("No application '" + id + "' was found"));
    }

    // TODO: Require ProvisionLock instead of Mutex
    public void put(Application application, Mutex applicationLock) {
        NestedTransaction transaction = new NestedTransaction();
        db.writeApplication(application, transaction);
        transaction.commit();
    }

    public void put(Application application, ApplicationTransaction transaction) {
        db.writeApplication(application, transaction.nested());
    }

    public void remove(ApplicationTransaction transaction) {
        db.deleteApplication(transaction);
    }

    /** Create a lock which provides exclusive rights to making changes to the given application */
    public ApplicationMutex lock(ApplicationId application) {
        return new ApplicationMutex(application, db.lock(application));
    }

    /** Create a lock with a timeout which provides exclusive rights to making changes to the given application */
    public ApplicationMutex lock(ApplicationId application, Duration timeout) {
        return new ApplicationMutex(application, db.lock(application, timeout));
    }

    /** Create a lock which provides exclusive rights to perform a maintenance deployment */
    public ApplicationMutex lockMaintenance(ApplicationId application) {
        return new ApplicationMutex(application, db.lockMaintenance(application));
    }

}