io.permazen.cli.cmd.KVLoadCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of permazen-cli Show documentation
Permazen classes supporting command line interfaces.
The newest version!

/*
 * Copyright (C) 2015 Archie L. Cobbs. All rights reserved.
 */

package io.permazen.cli.cmd;

import io.permazen.cli.Session;
import io.permazen.cli.SessionMode;
import io.permazen.cli.parse.Parser;
import io.permazen.kv.KVTransaction;
import io.permazen.kv.util.XMLSerializer;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.util.EnumSet;
import java.util.Map;

public class KVLoadCommand extends AbstractKVCommand {

    public KVLoadCommand() {
        super("kvload -R:reset file.xml:file");
    }

    @Override
    public String getHelpSummary() {
        return "Load key/value pairs from an XML file";
    }

    @Override
    public String getHelpDetail() {
        return "Imports key/value pairs from an XML file created previously via \"kvsave\". Does NOT remove any key/value pairs"
          + "already in the database unless the \"-R\" flag is given, in which case the database is completely wiped first."
          + "\n\nWARNING: this command can corrupt a Permazen database.";
    }

    @Override
    protected Parser getParser(String typeName) {
        return "file".equals(typeName) ? new InputFileParser() : super.getParser(typeName);
    }

    @Override
    public EnumSet getSessionModes() {
        return EnumSet.of(SessionMode.KEY_VALUE);
    }

    @Override
    public Session.Action getAction(Session session, Map params) {
        return new LoadAction(params.containsKey("reset"), (File)params.get("file.xml"));
    }

    private static class LoadAction implements KVAction {

        private final boolean reset;
        private final File file;

        LoadAction(boolean reset, File file) {
            this.reset = reset;
            this.file = file;
        }

        @Override
        public void run(Session session) throws Exception {
            final KVTransaction kvt = session.getKVTransaction();
            if (this.reset)
                kvt.removeRange(null, null);
            final int count;
            try (BufferedInputStream input = new BufferedInputStream(new FileInputStream(this.file))) {
                count = new XMLSerializer(kvt).read(input);
            }
            session.getOutput().println("Read " + count + " key/value pairs from \"" + this.file + "\"");
        }
    }
}