com.github.cherimojava.orchidae.config.cfgMongo.java Source code

Introduction

Here is the source code for com.github.cherimojava.orchidae.config.cfgMongo.java

Source

/**
 * Copyright (C) 2014 cherimojava (http://github.com/cherimojava/orchidae)
 *
 * 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.github.cherimojava.orchidae.config;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.UnknownHostException;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import javax.inject.Named;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;

import com.github.cherimojava.data.mongo.entity.EntityFactory;
import com.github.cherimojava.orchidae.util.FileUtil;
import com.github.cherimojava.orchidae.util.UserUtil;
import com.google.common.base.Throwables;
import com.mongodb.MongoClient;
import com.mongodb.ServerAddress;
import com.mongodb.client.MongoDatabase;

import de.flapdoodle.embed.mongo.Command;
import de.flapdoodle.embed.mongo.MongodExecutable;
import de.flapdoodle.embed.mongo.MongodStarter;
import de.flapdoodle.embed.mongo.config.ArtifactStoreBuilder;
import de.flapdoodle.embed.mongo.config.DownloadConfigBuilder;
import de.flapdoodle.embed.mongo.config.IMongodConfig;
import de.flapdoodle.embed.mongo.config.MongodConfigBuilder;
import de.flapdoodle.embed.mongo.config.Net;
import de.flapdoodle.embed.mongo.config.RuntimeConfigBuilder;
import de.flapdoodle.embed.mongo.config.Storage;
import de.flapdoodle.embed.mongo.config.Timeout;
import de.flapdoodle.embed.mongo.distribution.Version;
import de.flapdoodle.embed.process.config.IRuntimeConfig;
import de.flapdoodle.embed.process.config.io.ProcessOutput;
import de.flapdoodle.embed.process.extract.UserTempNaming;
import de.flapdoodle.embed.process.io.IStreamProcessor;
import de.flapdoodle.embed.process.io.Processors;
import de.flapdoodle.embed.process.io.directories.FixedPath;
import de.flapdoodle.embed.process.runtime.Network;
import de.flapdoodle.embed.process.store.Downloader;

/**
 * MongoDB configuration
 */
@Configuration
@Import({ cfgProduction.class, cfgTest.class })
public class cfgMongo {

    @Autowired
    @Named("dbName")
    private String mongoDBName;

    @Autowired
    @Named("dbPath")
    private String dbpath;

    @Value("${log.path:./log}")
    String logPath;

    private MongodExecutable exe;

    File mongoStoragePath() {
        return new File(dbpath);
    }

    @Bean
    public File logPath() {
        return new File(logPath);
    }

    @Bean
    public FileUtil fileSeparator() {
        return new FileUtil();
    }

    @Bean
    public EntityFactory entityFactory() throws UnknownHostException {
        return new EntityFactory(mongoDatabase());
    }

    @Value("${cache.user.size:20}")
    int userCacheSize;

    @Value("${concurrencyLevel:1}")
    int concurrencyLevel;

    @Bean
    public UserUtil userUtil() {
        return new UserUtil(userCacheSize, concurrencyLevel);
    }

    @Bean
    Integer mongoPort() {
        // try {
        // return Network.getFreeServerPort();
        // } catch (IOException e) {
        return 27017;
        // }
    }

    @Bean
    Version version() {
        return Version.V2_6_1;
    }

    @Bean
    MongoDatabase mongoDatabase() throws UnknownHostException {
        return new MongoClient(new ServerAddress("localhost", mongoPort())).getDatabase(mongoDBName);
    }

    private IMongodConfig mongodConfig() throws IOException {
        return new MongodConfigBuilder().version(version()).net(new Net(mongoPort(), Network.localhostIsIPv6()))
                .replication(new Storage(mongoStoragePath().toString(), null, 0)).timeout(new Timeout()).build();
    }

    private IRuntimeConfig runtimeConfig() {
        FixedPath path = new FixedPath("bin/");
        IStreamProcessor mongodOutput;
        IStreamProcessor mongodError;
        IStreamProcessor commandsOutput;
        try {
            mongodOutput = Processors.named("[mongod>]", new FileStreamProcessor(new File(logPath, "mongo.log")));

            mongodError = new FileStreamProcessor(new File(logPath, "mongo-err.log"));
            commandsOutput = Processors.named("[mongod>]", new FileStreamProcessor(new File(logPath, "mongo.log")));
        } catch (FileNotFoundException e) {
            throw Throwables.propagate(e);
        }
        return new RuntimeConfigBuilder().defaults(Command.MongoD)
                .processOutput(new ProcessOutput(mongodOutput, mongodError, commandsOutput))
                .artifactStore(new ArtifactStoreBuilder().downloader(new Downloader())
                        .executableNaming(new UserTempNaming()).tempDir(path).download(new DownloadConfigBuilder()
                                .defaultsForCommand(Command.MongoD).artifactStorePath(path)))
                .build();
    }

    @PostConstruct
    public void setup() {
        try {
            MongodStarter runtime = MongodStarter.getInstance(runtimeConfig());
            exe = runtime.prepare(mongodConfig());
            exe.start();
        } catch (Exception e) {
            throw Throwables.propagate(e);
        }
    }

    @PreDestroy
    public void teardown() {
        exe.stop();
    }

    public class FileStreamProcessor implements IStreamProcessor {

        private FileOutputStream outputStream;

        public FileStreamProcessor(File file) throws FileNotFoundException {
            file.getParentFile().mkdirs();
            outputStream = new FileOutputStream(file);
        }

        @Override
        public void process(String block) {
            try {
                outputStream.write(block.getBytes());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        @Override
        public void onProcessed() {
            try {
                outputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}