sync/src/partition/partitionmodule.js

import { loadFromToml } from 'cytube-common/lib/configuration/configloader';
import { PartitionConfig } from './partitionconfig';
import { PartitionDecider } from './partitiondecider';
import { PartitionClusterClient } from '../io/cluster/partitionclusterclient';
import RedisClientProvider from 'cytube-common/lib/redis/redisclientprovider';
import logger from 'cytube-common/lib/logger';
import LegacyConfig from '../config';
import path from 'path';
import { AnnouncementRefresher } from './announcementrefresher';
import { RedisPartitionMapReloader } from './redispartitionmapreloader';

const PARTITION_CONFIG_PATH = path.resolve(__dirname, '..', '..', 'conf',
                                           'partitions.toml');

class PartitionModule {
    constructor() {
        this.initConfig();
    }

    onReady() {
        this.getAnnouncementRefresher();
    }

    initConfig() {
        logger.initialize(null, null, LegacyConfig.get('debug'));
        try {
            this.partitionConfig = this.loadPartitionConfig();
        } catch (error) {
            process.exit(1);
        }
    }

    loadPartitionConfig() {
        try {
            return loadFromToml(PartitionConfig, PARTITION_CONFIG_PATH);
        } catch (error) {
            if (typeof error.line !== 'undefined') {
                logger.error(`Error in ${PARTITION_CONFIG_PATH}: ${error} ` +
                             `(line ${error.line})`);
            } else {
                logger.error(`Error loading ${PARTITION_CONFIG_PATH}: ` +
                             `${error.stack}`);
            }
            throw error;
        }
    }

    getPartitionMapReloader() {
        if (!this.partitionMapReloader) {
            const redisProvider = this.getRedisClientProvider();
            this.partitionMapReloader = new RedisPartitionMapReloader(
                    redisProvider.get(),  // Client for GET partitionMap
                    redisProvider.get()); // Subscribe client
        }

        return this.partitionMapReloader;
    }

    getPartitionDecider() {
        if (!this.partitionDecider) {
            const reloader = this.getPartitionMapReloader();
            this.partitionDecider = new PartitionDecider(this.partitionConfig,
                    reloader.getPartitionMap());
            reloader.on('partitionMapChange', newMap => {
                this.partitionDecider.setPartitionMap(newMap);
                require('../server').getServer().handlePartitionMapChange();
            });
        }

        return this.partitionDecider;
    }

    getClusterClient() {
        if (!this.partitionClusterClient) {
            this.partitionClusterClient = new PartitionClusterClient(
                    this.getPartitionDecider());
        }

        return this.partitionClusterClient;
    }

    getRedisClientProvider() {
        if (!this.redisClientProvider) {
            this.redisClientProvider = new RedisClientProvider(
                    this.partitionConfig.getRedisConfig()
            );
        }

        return this.redisClientProvider;
    }

    getAnnouncementRefresher() {
        if (!this.announcementRefresher) {
            const provider = this.getRedisClientProvider();
            this.announcementRefresher = new AnnouncementRefresher(
                    provider.get(),
                    provider.get()
            );
        }

        return this.announcementRefresher;
    }
}

export { PartitionModule };