fluence-js/src/subscriptions.ts

/*
 * Copyright 2020 Fluence Labs Limited
 *
 * 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.
 */

import {Particle} from "./particle";
import log from "loglevel";

export class Subscriptions {
    private subscriptions: Set<string> = new Set();

    constructor() {}

    /**
     * Subscriptions will be applied by outside message if id will be the same.
     *
     * @param id message identificator
     * @param ttl time to live, subscription will be deleted after this time
     */
    subscribe(id: string, ttl: number) {
        let _this = this;
        setTimeout(() => {
            _this.subscriptions.delete(id)
            log.info(`Particle with id ${id} deleted by timeout`)
        }, ttl)
        this.subscriptions.add(id);
    }

    hasSubscription(particle: Particle): boolean {
        return this.subscriptions.has(particle.id)
    }
}