com.jivesoftware.os.miru.service.locator.MiruResourceLocatorInitializer.java Source code

Introduction

Here is the source code for com.jivesoftware.os.miru.service.locator.MiruResourceLocatorInitializer.java

Source

/*
 * Copyright 2014 jonathan.
 *
 * 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 com.jivesoftware.os.miru.service.locator;

import com.jivesoftware.os.miru.service.MiruServiceConfig;
import com.jivesoftware.os.routing.bird.health.api.HealthChecker;
import com.jivesoftware.os.routing.bird.health.api.HealthFactory;
import com.jivesoftware.os.routing.bird.health.api.ScheduledMinMaxHealthCheckConfig;
import com.jivesoftware.os.routing.bird.health.checkers.DiskFreeHealthChecker;
import java.io.File;
import java.io.IOException;
import org.apache.commons.io.FileUtils;
import org.merlin.config.defaults.LongDefault;
import org.merlin.config.defaults.StringDefault;

/**
 * @author jonathan
 */
public class MiruResourceLocatorInitializer {

    static interface ResidentResourceDiskCheck extends ScheduledMinMaxHealthCheckConfig {

        @StringDefault("resident>resource>disk")
        @Override
        public String getName();

        @LongDefault(80)
        @Override
        public Long getMax();

    }

    public MiruResourceLocator initialize(final MiruServiceConfig config) throws IOException {

        File[] residentDiskPaths = pathToFile(config.getDiskResourceLocatorPaths().split(","));
        for (File residentDiskPath : residentDiskPaths) {
            FileUtils.forceMkdir(residentDiskPath);
        }
        final File[] checkThese = new File[residentDiskPaths.length + 1];
        System.arraycopy(residentDiskPaths, 0, checkThese, 0, residentDiskPaths.length);
        checkThese[checkThese.length - 1] = new File(System.getProperty("user.dir"));

        HealthFactory.scheduleHealthChecker(ResidentResourceDiskCheck.class,
                config1 -> (HealthChecker) new DiskFreeHealthChecker(config1, checkThese));

        final MiruResourceLocator residentDiskResourceLocator = new DiskIdentifierPartResourceLocator(
                residentDiskPaths, config.getOnDiskInitialChunkSize(), config.getInMemoryChunkSize());

        final File[] transientPaths = pathToFile(config.getTransientResourceLocatorPaths().split(","));
        for (File transientPath : transientPaths) {
            FileUtils.forceMkdir(transientPath);
        }

        return residentDiskResourceLocator;
    }

    private File[] pathToFile(String[] paths) {
        File[] files = new File[paths.length];
        for (int i = 0; i < files.length; i++) {
            files[i] = new File(paths[i].trim());
        }
        return files;
    }
}