bear.plugins.db.MysqlDbService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bear Show documentation
The newest version!
package bear.plugins.db;

import bear.core.SessionContext;
import bear.main.event.NoticeEventToUI;
import bear.plugins.mysql.MySqlPlugin;
import bear.vcs.CommandLineResult;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import org.joda.time.DateTime;

import java.util.List;
import java.util.concurrent.Callable;

import static bear.core.SessionContext.ui;

/**
* @author Andrey Chaschev [email protected]
*/
public class MysqlDbService extends DbDumpManager.AbstractDbService {
    MySqlPlugin mysql;

    public MysqlDbService(SessionContext $) {
        super($);
    }

    @Override
    public Class getDbDumpInfoClass() {
        return DbDumpInfo.class;
    }

    @Override
    public Class getDumpableEntryClass() {
        return DbDumpManager.SqlDumpableEntry.class;
    }

    @Override
    public DbDumpManager.SqlDumpableEntry list(String dbName) {
        throw new UnsupportedOperationException("todo");
    }

    @Override
    public DbDumpInfo createDump(DbDumpManager.SqlDumpableEntry entry) {
//            ReleasesPlugin.RELEASE_FORMATTER.
        String dbName = entry.database;

        ui.info(new NoticeEventToUI("Mysql Dump", "Started mysql dump at " + $($.bear.sessionHostname)));

        DateTime startedAt = new DateTime();

        CommandLineResult result =
            $.sys.script().line().timeoutMin(24 * 60) //24h
                .addRaw("mysqldump --user=%s -p %s | gzip -c >%s",
                    $(mysql.user), entry.database, $(plugin.dumpArchivePath)).build()
                .callback(MySqlPlugin.passwordCallback($(mysql.password)))
                .run();

        DateTime finishedAt = new DateTime();

        ui.info(new NoticeEventToUI("Mysql Dump", "Mysql dump finished in " + $($.bear.sessionHostname)));

        List dumpInfos = Lists.newArrayList(listDumps());

        DbDumpInfo dbDumpInfo = new DbDumpInfo(
            $(plugin.dumpName),
            $(plugin.dbType),
            dbName,
            "",
            startedAt,
            finishedAt,
            $.sys.fileSizeAsLong($(plugin.dumpArchivePath))
        );

        dumpInfos.add(dbDumpInfo);

        saveDumpList(dumpInfos);

        return dbDumpInfo;
    }

    @Override
    public void restoreDump(final DbDumpInfo dbDumpInfo) {
        $.withMap(
            ImmutableMap.builder()
                .put(plugin.dumpName, dbDumpInfo.name)
                .build(),
            new Callable