source: java/main/src/main/java/com/framsticks/dumping/SaveStream.java @ 193

package com.framsticks.dumping;

import static com.framsticks.params.AccessOperations.*;
import com.framsticks.params.Access;
import com.framsticks.params.CompositeParam;
import com.framsticks.params.FramsClass;
import com.framsticks.params.ListAccess;
import com.framsticks.params.Sink;
import com.framsticks.parsers.Savers;
import com.framsticks.structure.Node;
import com.framsticks.structure.Path;
import com.framsticks.structure.Tree;
import com.framsticks.util.*;
import com.framsticks.util.dispatching.Dispatching;
import com.framsticks.util.dispatching.FutureHandler;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;
import com.framsticks.util.dispatching.RunAt;

import java.util.HashSet;
import java.util.Set;

import static com.framsticks.structure.TreeOperations.*;
import static com.framsticks.util.lang.Containers.filterInstanceof;

/**
 * @author Piotr Sniegowski
 */
public class SaveStream extends Stream {

        private final static Logger log = LogManager.getLogger(SaveStream.class.getName());

        protected final Sink sink;
        protected final Tree tree;
        protected final FutureHandler<Void> future;
        protected final Stopwatch stopwatch = new Stopwatch();
        protected final Set<FramsClass> storedInfo = new HashSet<FramsClass>();

        private int dispatched = 0;

        public SaveStream(Sink sink, Tree tree, Path root, FutureHandler<Void> future) {
                assert Dispatching.isThreadSafe();
                this.sink = sink;
                this.tree = tree;
                this.future = future;
                dispatchWrite(root);
        }

        protected void dispatchWrite(final Path path) {
                ++dispatched;
                tree.dispatch(new RunAt<Tree>(tree) {
                        @Override
                        protected void runAt() {
                                write(path);
                        }
                });
        }

        protected void finished() {
                assert tree.isActive();
                log.info("stored in {}", stopwatch);
                future.pass(null);
        }

        public void write(final Path path) {
                assert tree.isActive();
                if (!path.isResolved()) {
                        log.debug("path {} is not resolved - skipping", path);
                } else {
                        Access access = bindAccess(path);
                        assert access != null;
                        FramsClass framsClass = access.getFramsClass();
                        assert framsClass != null;
                        if (!storedInfo.contains(framsClass)) {
                                storedInfo.add(framsClass);
                                sink.print("info ").print(path.getTextual()).breakLine();
                                sink.print("file").breakLine();
                                Savers.saveFramsClass(sink, framsClass);
                                sink.print("eof").breakLine();
                                sink.print("ok").breakLine();
                        }
                        if (!(access instanceof ListAccess)) {
                                sink.print("get ").print(path.getTextual()).breakLine();
                                sink.print("file").breakLine();
                                //stream.print("#" + access.getSelected().getClass().getCanonicalName() + "\n");
                                save(access, sink);
                                sink.print("eof").breakLine();
                                sink.print("ok").breakLine();
                        }
                        for (CompositeParam p : filterInstanceof(access.getParams(), CompositeParam.class)) {
                                final Path childPath = path.appendNode(new Node(path.getTree(), p, access.get(p, Object.class)));
                                if (childPath.isResolved() && getInfoFromCache(childPath) != null) {
                                        dispatchWrite(childPath);
                                }
                        }
                }
                --dispatched;
                if (dispatched == 0) {
                        finished();
                }
        }
}