/*-
 * ========================LICENSE_START=================================
 * O-RAN-SC
 * %%
 * Copyright (C) 2019 Nordix Foundation
 * %%
 * 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
 *
 *      http://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.
 * ========================LICENSE_END===================================
 */

package org.oransc.enrichment.repository;

import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import lombok.Getter;

public class InfoProducer {
    @Getter
    private final String id;

    @Getter
    private final Collection<InfoType> infoTypes;

    @Getter
    private final String jobCallbackUrl;

    @Getter
    private final String producerSupervisionCallbackUrl;

    private final Set<String> enabledJobs = new HashSet<>();

    private int unresponsiveCounter = 0;

    public InfoProducer(String id, Collection<InfoType> infoTypes, String jobCallbackUrl,
        String producerSupervisionCallbackUrl) {
        this.id = id;
        this.infoTypes = infoTypes;
        this.jobCallbackUrl = jobCallbackUrl;
        this.producerSupervisionCallbackUrl = producerSupervisionCallbackUrl;
    }

    public synchronized void setAliveStatus(boolean isAlive) {
        if (isAlive) {
            unresponsiveCounter = 0;
        } else {
            unresponsiveCounter++;
        }
    }

    public synchronized boolean isDead() {
        return this.unresponsiveCounter >= 3;
    }

    public synchronized boolean isAvailable() {
        return this.unresponsiveCounter == 0;
    }

    public synchronized void setJobEnabled(InfoJob job) {
        this.enabledJobs.add(job.getId());
    }

    public synchronized void setJobDisabled(InfoJob job) {
        this.enabledJobs.remove(job.getId());
    }

    /**
     * Is the job enabled for this producer?
     */
    public synchronized boolean isJobEnabled(InfoJob job) {
        return this.enabledJobs.contains(job.getId());
    }

}