io.bdeploy.bhive.util.SimpleTreeMapper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of api Show documentation

Public API including dependencies, ready to be used for integrations and plugins.

There is a newer version: 7.3.6

package io.bdeploy.bhive.util;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Map.Entry;

import com.google.common.base.Splitter;

import io.bdeploy.bhive.model.ObjectId;
import io.bdeploy.bhive.model.Tree;
import io.bdeploy.bhive.model.Tree.EntryType;
import io.bdeploy.bhive.model.Tree.Key;
import io.bdeploy.bhive.util.StorageHelper.CustomMapper;

/**
 * Custom tree serialization/de-serialization. JSON/YAML is simply too slow for the amount of objects.
 */
public class SimpleTreeMapper implements CustomMapper {

    @Override
    public byte[] write(Object obj) {
        Tree t = (Tree) obj;
        StringBuilder builder = new StringBuilder();

        for (Entry entry : t.getChildren().entrySet()) {
            builder.append(entry.getKey().getType().name()).append('|');
            builder.append(entry.getValue().toString()).append('|');
            builder.append("").append('|'); // field reserved for future use
            builder.append(entry.getKey().getName()).append('\n');
        }

        return builder.toString().getBytes(StandardCharsets.UTF_8);
    }

    @Override
    public Object read(InputStream is) {
        Tree.Builder builder = new Tree.Builder();
        try (BufferedReader r = new BufferedReader(new InputStreamReader(is, StandardCharsets.UTF_8))) {
            String line;
            while ((line = r.readLine()) != null) {
                List parts = Splitter.on('|').splitToList(line);
                builder.add(new Tree.Key(parts.get(3), EntryType.valueOf(parts.get(0))), ObjectId.parse(parts.get(1)));
            }
        } catch (IOException e) {
            throw new IllegalStateException("Cannot read tree", e);
        }
        return builder.build();
    }

}