import { Message } from '../model/message';
import { HasSubscribers, Publisher, Subscriber, Subscription } from './publisher';


/**
 * A Publisher that handle all the data in memory. It is a very simple implementation that should be used
 * only for development and test purposes.
 */
export class InMemoryPublisher implements Publisher, HasSubscribers {

    private listeners: Map<string, Array<Subscriber>> = new Map();

    public async publish(message: Message) {
        const aggregationListeners = this.listeners.get(message.stream.aggregation);

        if (aggregationListeners != null && aggregationListeners.length) {
            aggregationListeners.forEach(subscriber => subscriber(message));
            return new Date().getTime().toString();
        }
        return null;
    }

    public async subscribe(aggregation: string, subscriber: Subscriber): Promise<Subscription> {
        let aggregateListeners = this.listeners.get(aggregation);
        if (!aggregateListeners) {
            aggregateListeners = new Array<Subscriber>();
            this.listeners.set(aggregation, aggregateListeners);
        }
        aggregateListeners.push(subscriber);
        return {
            remove: async () => {
                const index = aggregateListeners.indexOf(subscriber);
                aggregateListeners.splice(index, 1);
            }
        };
    }
}