io.castled.commands.CastledServerCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of container Show documentation
There is a newer version: 1.0.0
package io.castled.commands;

import com.mysql.cj.jdbc.MysqlDataSource;
import io.castled.CastledApplication;
import io.castled.CastledConfiguration;
import io.castled.CastledStateStore;
import io.castled.ObjectRegistry;
import io.castled.daos.InstallationDAO;
import io.castled.events.CastledEventsClient;
import io.castled.events.NewInstallationEvent;
import io.castled.exceptions.CastledRuntimeException;
import io.castled.migrations.MigrationType;
import io.castled.migrations.DataMigratorFactory;
import io.castled.models.RedisConfig;
import io.castled.services.UsersService;
import io.castled.utils.AsciiArtUtils;
import io.castled.utils.FileUtils;
import io.dropwizard.cli.ServerCommand;
import io.dropwizard.setup.Environment;
import lombok.extern.slf4j.Slf4j;
import net.sourceforge.argparse4j.inf.Namespace;
import org.flywaydb.core.Flyway;
import org.jdbi.v3.core.Jdbi;
import org.redisson.Redisson;
import org.redisson.api.RedissonClient;
import org.redisson.config.Config;

import java.io.IOException;
import java.util.List;
import java.util.Objects;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.stream.Collectors;

@Slf4j
public class CastledServerCommand extends ServerCommand {

    public CastledServerCommand(CastledApplication castledApplication) {
        super(castledApplication, "castled-server", "Runs the castled server");
    }

    protected void run(Environment environment, Namespace namespace, CastledConfiguration configuration) throws Exception {
        runMigrations(configuration);
        super.run(environment, namespace, configuration);
        AsciiArtUtils.drawCastled();
    }

    private void runMigrations(CastledConfiguration configuration) {

        RedisConfig redisConfig = configuration.getRedisConfig();
        Config config = new Config();
        config.useSingleServer().setAddress(String.format("redis://%s:%s",
                redisConfig.getHost(), redisConfig.getPort()));
        RedissonClient redisson = Redisson.create(config);
        Lock lock = null;
        try {
            lock = redisson.getLock("data_migrations");
            tryLock(lock);
            runSQLMigrations(configuration);
            runCodeLevelMigrations();
        } finally {
            if (lock != null) {
                lock.unlock();
            }
            redisson.shutdown();
        }
    }

    private void tryLock(Lock lock) {
        try {
            boolean tryLock = lock.tryLock(30, TimeUnit.MINUTES);
            if (!tryLock) {
                throw new CastledRuntimeException("Timeout waiting to acquire lock for data migration");
            }
        } catch (InterruptedException e) {
            throw new CastledRuntimeException("Interrupted waiting to acquire lock for data migration");
        }
    }

    private void runSQLMigrations(CastledConfiguration configuration) {
        Flyway flyway = new Flyway();
        MysqlDataSource mysqlDataSource = new MysqlDataSource();
        mysqlDataSource.setURL(configuration.getDatabase().getUrl());
        flyway.setDataSource(mysqlDataSource);
        flyway.setLocations("migration");
        flyway.migrate();

    }

    private void runCodeLevelMigrations() {
        try {
            DataMigratorFactory dataMigratorFactory = ObjectRegistry.getInstance(DataMigratorFactory.class);
            List allMigrations = Objects.requireNonNull(FileUtils.getResourceFileLines("java_migrations"))
                    .stream().map(MigrationType::valueOf).collect(Collectors.toList());
            for (MigrationType migrationType : allMigrations) {
                dataMigratorFactory.getDataMigrator(migrationType).migrateData();
            }
        } catch (IOException e) {
            log.error("Code level data migration failed", e);
            throw new CastledRuntimeException(e.getMessage());
        }
    }
}