yacy_search_server/source/de/anomic/kelondro/kelondroRelations.java

package de.anomic.kelondro;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;

public class kelondroRelations {

    private File baseDir;
    private HashMap<String, kelondroIndex> relations;
    
    public kelondroRelations(File location) {
        this.baseDir = location;
    }
    
    private static kelondroRow rowdef(String filename) {
        int p = filename.lastIndexOf('.');
        if (p >= 0) filename = filename.substring(0, p);
        p = filename.lastIndexOf('-');
        assert p >= 0;
        int payloadsize = Integer.parseInt(filename.substring(p + 1));
        filename = filename.substring(0, p);
        p = filename.lastIndexOf('-');
        assert p >= 0;
        int keysize = Integer.parseInt(filename.substring(p + 1));
        return rowdef(keysize, payloadsize);
    }
    
    private static kelondroRow rowdef(int keysize, int payloadsize) {
        return new kelondroRow(
                "byte[] key-" + keysize + ", " +
                "long time-8" + keysize + ", " +
                "int ttl-4" + keysize + ", " +
                "byte[] node-" + payloadsize,
                kelondroNaturalOrder.naturalOrder, 0);
    }
    
    private static String filename(String tablename, int keysize, int payloadsize) {
        return tablename + "-" + keysize + "-" + payloadsize + ".eco";
    }
    
    public void declareRelation(String name, int keysize, int payloadsize) {
        // try to get the relation from the relation-cache
        kelondroIndex relation = relations.get(name);
        if (relation != null) return;
        // try to find the relation as stored on file
        String[] list = baseDir.list();
        String targetfilename = filename(name, keysize, payloadsize);
        for (int i = 0; i < list.length; i++) {
            if (list[i].startsWith(name)) {
                if (!list[i].equals(targetfilename)) continue;
                kelondroRow row = rowdef(list[i]);
                if (row.primaryKeyLength != keysize || row.column(1).cellwidth != payloadsize) continue; // a wrong table
                kelondroIndex table = new kelondroEcoTable(new File(baseDir, list[i]), row, kelondroEcoTable.tailCacheUsageAuto, 1024*1024, 0);
                relations.put(name, table);
                return;
            }
        }
        // the relation does not exist, create it
        kelondroRow row = rowdef(keysize, payloadsize);
        kelondroIndex table = new kelondroEcoTable(new File(baseDir, targetfilename), row, kelondroEcoTable.tailCacheUsageAuto, 1024*1024, 0);
        relations.put(name, table);
    }
    
    public kelondroIndex getRelation(String name) {
        // try to get the relation from the relation-cache
        kelondroIndex relation = relations.get(name);
        if (relation != null) return relation;
        // try to find the relation as stored on file
        String[] list = baseDir.list();
        for (int i = 0; i < list.length; i++) {
            if (list[i].startsWith(name)) {
                kelondroRow row = rowdef(list[i]);
                kelondroIndex table = new kelondroEcoTable(new File(baseDir, list[i]), row, kelondroEcoTable.tailCacheUsageAuto, 1024*1024, 0);
                relations.put(name, table);
                return table;
            }
        }
        // the relation does not exist
        return null;
    }
    
    public String putRelation(String name, String key, String value) throws IOException {
        byte[] r = putRelation(name, key.getBytes(), value.getBytes());
        if (r == null) return null;
        return new String(r);
    }
    
    public byte[] putRelation(String name, byte[] key, byte[] value) throws IOException {
        kelondroIndex table = getRelation(name);
        if (table == null) return null;
        kelondroRow.Entry entry = table.row().newEntry();
        entry.setCol(0, key);
        entry.setCol(1, System.currentTimeMillis());
        entry.setCol(2, 1000000);
        entry.setCol(3, value);
        kelondroRow.Entry oldentry = table.put(entry);
        if (oldentry == null) return null;
        return oldentry.getColBytes(3);
    }
    
    public String getRelation(String name, String key) throws IOException {
        byte[] r = getRelation(name, key.getBytes());
        if (r == null) return null;
        return new String(r);
    }
    
    public byte[] getRelation(String name, byte[] key) throws IOException {
        kelondroIndex table = getRelation(name);
        if (table == null) return null;
        kelondroRow.Entry entry = table.get(key);
        if (entry == null) return null;
        return entry.getColBytes(3);
    }
    
    public boolean hasRelation(String name, byte[] key) throws IOException {
        kelondroIndex table = getRelation(name);
        if (table == null) return false;
        return table.has(key);
    }
    
    public byte[] removeRelation(String name, byte[] key) throws IOException {
        kelondroIndex table = getRelation(name);
        if (table == null) return null;
        kelondroRow.Entry entry = table.remove(key, false);
        if (entry == null) return null;
        return entry.getColBytes(3);
    }
    
    public static void main(String args[]) {
        kelondroRelations r = new kelondroRelations(new File("/Users/admin/"));
        try {
            String table1 = "test1";
            r.declareRelation(table1, 12, 30);
            r.putRelation(table1, "abcdefg", "eineintrag");
            r.putRelation(table1, "abcdefg", "eineintrag");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    
}