org.elasticsearch.river.mongodb.StatusChecker Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elasticsearch-river-mongodb Show documentation

MongoDB River for ElasticSearch

The newest version!

package org.elasticsearch.river.mongodb;

import org.elasticsearch.river.mongodb.util.MongoDBRiverHelper;

class StatusChecker implements Runnable {

    private final MongoDBRiver mongoDBRiver;
    private final MongoDBRiverDefinition definition;
    private final SharedContext context;

    public StatusChecker(MongoDBRiver mongoDBRiver, MongoDBRiverDefinition definition, SharedContext context) {
        this.mongoDBRiver = mongoDBRiver;
        this.definition = definition;
        this.context = context;
    }

    @Override
    public void run() {
        while (true) {
            try {
                if (this.mongoDBRiver.startInvoked) {
                    Status status = MongoDBRiverHelper.getRiverStatus(this.mongoDBRiver.client, this.definition.getRiverName());
                    if (status != this.context.getStatus()) {
                        if (status == Status.RUNNING) {
                            MongoDBRiver.logger.trace("About to start river: {}", this.definition.getRiverName());
                            this.mongoDBRiver.start();
                        } else if (status == Status.STOPPED) {
                            MongoDBRiver.logger.info("About to stop river: {}", this.definition.getRiverName());
                            this.mongoDBRiver.close();
                        }
                    }
                }
                Thread.sleep(1000L);
            } catch (InterruptedException e) {
                MongoDBRiver.logger.info("Status thread interrupted", e, (Object) null);
                Thread.currentThread().interrupt();
                break;
            }

        }
    }
}