LinkIndexSetup.java

/*
 * Copyright 2012 NGDATA nv
 *
 * 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.
 */
package org.lilyproject.server.modules.general;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.io.IOException;

import com.ngdata.sep.SepModel;
import com.ngdata.sep.impl.SepConsumer;
import org.apache.hadoop.conf.Configuration;
import org.apache.zookeeper.KeeperException;
import org.lilyproject.hbaseindex.IndexManager;
import org.lilyproject.hbaseindex.IndexNotFoundException;
import org.lilyproject.linkindex.LinkIndex;
import org.lilyproject.linkindex.LinkIndexUpdater;
import org.lilyproject.repository.api.RepositoryException;
import org.lilyproject.repository.api.RepositoryManager;
import org.lilyproject.sep.LilyPayloadExtractor;
import org.lilyproject.sep.ZooKeeperItfAdapter;
import org.lilyproject.util.hbase.HBaseTableFactory;
import org.lilyproject.util.io.Closer;
import org.lilyproject.util.zookeeper.ZooKeeperItf;

public class LinkIndexSetup {
    private final SepModel sepModel;
    private final boolean linkIndexEnabled;
    private final int threads;
    private final RepositoryManager repositoryManager;
    private final Configuration hbaseConf;
    private final HBaseTableFactory tableFactory;
    private final ZooKeeperItf zk;
    private final String hostName;
    private SepConsumer sepConsumer;

    public LinkIndexSetup(SepModel sepModel, boolean linkIndexEnabled, int threads, RepositoryManager repositoryManager,
            Configuration hbaseConf, HBaseTableFactory tableFactory, ZooKeeperItf zk, String hostName) {
        this.sepModel = sepModel;
        this.linkIndexEnabled = linkIndexEnabled;
        this.threads = threads;
        this.repositoryManager = repositoryManager;
        this.hbaseConf = hbaseConf;
        this.tableFactory = tableFactory;
        this.zk = zk;
        this.hostName = hostName;
    }

    @PostConstruct
    public void start() throws InterruptedException, KeeperException, IOException,
            IndexNotFoundException, RepositoryException {

        if (linkIndexEnabled) {
            // assure the subscription exists
            sepModel.addSubscriptionSilent("LinkIndexUpdater");
        } else {
            // assure the subscription doesn't exist
            sepModel.removeSubscriptionSilent("LinkIndexUpdater");
        }

        if (linkIndexEnabled) {
            IndexManager indexManager = new IndexManager(hbaseConf, tableFactory);

            LinkIndex linkIndex = new LinkIndex(indexManager, /* TODO multiple repositories */ repositoryManager);

            LinkIndexUpdater linkIndexUpdater = new LinkIndexUpdater(repositoryManager, linkIndex);

            sepConsumer = new SepConsumer("LinkIndexUpdater", 0L, linkIndexUpdater, threads, hostName,
                    new ZooKeeperItfAdapter(zk), hbaseConf, new LilyPayloadExtractor());
            sepConsumer.start();
        }
    }

    @PreDestroy
    public void stop() {
        Closer.close(sepConsumer);
    }
}