Дмитрий Морозов, 396, JUnit

src/ru/fizteh/fivt/students/dmitry_morozov/junit/BadDBFileException.java

@@ -0,0 +1,9 @@
+package ru.fizteh.fivt.students.dmitry_morozov.junit;
+
+public class BadDBFileException extends RuntimeException {
+
+    public BadDBFileException(String msg) {
+        super(msg);
+    }
+
+}

src/ru/fizteh/fivt/students/dmitry_morozov/junit/ChangingCommand.java

@@ -0,0 +1,24 @@
+package ru.fizteh.fivt.students.dmitry_morozov.junit;
+
+enum CommandName {
+    PUT, REMOVE
+};
+
+public class ChangingCommand {
+    public CommandName cName;
+    public String[] args;
+
+    public ChangingCommand(CommandName c, String arg1, String arg2) {
+        cName = c;
+        if (cName == CommandName.PUT) {
+            args = new String[2];
+            args[0] = arg1;
+            args[1] = arg2;
+        }
+        if (cName == CommandName.REMOVE) {
+            args = new String[1];
+            args[0] = arg1;
+        }
+    }
+
+}

src/ru/fizteh/fivt/students/dmitry_morozov/junit/DBCollection.java

@@ -0,0 +1,89 @@
+package ru.fizteh.fivt.students.dmitry_morozov.junit;
+
+import java.util.List;
+import java.io.File;
+import java.io.IOException;
+
+import ru.fizteh.fivt.storage.strings.Table;
+import ru.fizteh.fivt.storage.strings.TableProvider;
+
+public class DBCollection implements TableProvider {
+    private String dirPath;
+    private FileMap maps;
+    private static final String INFO_FILE_NAME = "tables_info.dat";
+
+    public DBCollection(String dirPath) throws IllegalArgumentException {
+        this.dirPath = dirPath;
+        maps = null;
+        if (dirPath == null) {
+            throw new IllegalArgumentException("path is null");
+        }
+        if (dirPath.endsWith(File.separator) && !dirPath.equals(File.separator)) {
+            this.dirPath = dirPath.substring(0, dirPath.length() - 1);
+        }
+        try {
+            maps = new FileMap(dirPath + File.separator + INFO_FILE_NAME);
+        } catch (BadDBFileException | IOException e) {
+            throw new IllegalArgumentException(e.getMessage());
+        }
+    }
+
+    public List<String> showTables() { // Actually doesn't
+                                       // throw anything.
+        return maps.list();
+    }
+
+    @Override
+    public Table getTable(String name) throws IllegalArgumentException {
+        String fullPath = maps.get(name);
+        if (fullPath == null) {
+            return null;
+        }
+        Table res;
+        try {
+            res = new TableWithTransactions(name);
+        } catch (BadDBFileException e) {
+            throw new IllegalArgumentException(e.getMessage());
+        }
+        return res;
+    }
+
+    @Override
+    public Table createTable(String name) {
+        // TODO Auto-generated method stub
+        if (maps.get(name) != null) {
+            return null;
+        }
+        maps.put(name, dirPath + File.separator + name);
+        File dir = new File(dirPath + File.separator + name);
+        if (!dir.mkdirs()) {
+            throw new IllegalArgumentException();
+        }
+        TableWithTransactions res;
+        try {
+            res = new TableWithTransactions(dirPath + File.separator + name);
+        } catch (BadDBFileException e) {
+            maps.remove(name);
+            throw new IllegalStateException();
+        }
+        return res;
+    }
+
+    @Override
+    public void removeTable(String name) {
+        String tname = maps.get(name);
+        if (tname != null) { // Database found.
+            if (!Utils.removeDirectory(tname)) {
+                throw new IllegalStateException("Couldn't remove directory");
+            } else {
+                maps.remove(name);
+            }
+        } else { // Database not found.
+            throw new IllegalStateException();
+        }
+    }
+
+    public void close() throws IOException {
+        maps.close();
+    }
+}

src/ru/fizteh/fivt/students/dmitry_morozov/junit/FileMap.java

@@ -0,0 +1,158 @@
+package ru.fizteh.fivt.students.dmitry_morozov.junit;
+
+import java.io.DataInputStream;
+import java.io.DataOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+
+import ru.fizteh.fivt.storage.strings.Table;
+
+public class FileMap implements Table {
+    private Map<String, String> table;
+    private File dbFile;
+
+    public FileMap(String path) throws BadDBFileException, IOException {
+        table = new HashMap<>();
+        dbFile = new File(path);
+        if (!dbFile.exists()) {
+            if (!dbFile.createNewFile()) {
+                throw new BadDBFileException("Couldn't create db file");
+            }
+        } else {
+            if (!dbFile.isFile()) {
+                throw new BadDBFileException("Is not a file");
+            }
+        }
+        if (!(dbFile.setReadable(true)) && dbFile.setWritable(true)) {
+            throw new BadDBFileException("Couldn't set rw options");
+        }
+        DataInputStream in = new DataInputStream(new FileInputStream(dbFile));
+
+        while (true) {
+            String key = readString(in);
+            if (key == null) {
+                break;
+            }
+            String value = readString(in);
+            if (value == null) {
+                in.close();
+                throw new BadDBFileException("Couldn't set rw options");
+            }
+            table.put(key, value);
+        }
+        in.close();
+
+    }
+
+    /**
+     * @return String read from file. If meets end of file, returns null.
+     * @throws BadDBFileException
+     * @throws IOException
+     * */
+
+    private String readString(DataInputStream in) throws IOException,
+            BadDBFileException {
+        final int sizeOfInt = 4;
+        int len;        
+        StringBuilder res = new StringBuilder();
+        if (in.available() >= sizeOfInt) {
+            len = in.readInt();
+            if (0 != len % 2 || in.available() < len) {
+                in.close();
+                throw new BadDBFileException("File was damaged");
+            }
+            len /= 2;
+            while (len > 0) {
+                char curChar = in.readChar();
+                res.append(curChar);
+                len--;
+            }
+        } else {
+            return null;
+        }
+        return res.toString();
+    }
+
+    public String put(String key, String value) throws IllegalArgumentException {
+        if (key == null) {
+            throw new IllegalArgumentException();
+        }
+        String res = table.get(key);
+        table.put(key, value);
+        return res;
+    }
+
+    public String get(String key) throws IllegalArgumentException {
+        if (key == null) {
+            throw new IllegalArgumentException();
+        }
+        String val = table.get(key);
+        return val;
+    }
+
+    public List<String> list() {
+        List<String> res = new ArrayList<>();
+        Set<Entry<String, String>> tableSet = table.entrySet();
+        for (Entry<String, String> i : tableSet) {
+            res.add(i.getKey());
+        }
+        return res;
+    }
+
+    public String remove(String key) throws IllegalArgumentException {
+        if (key == null) {
+            throw new IllegalArgumentException();
+        }
+        return table.remove(key);
+    }
+
+    public void close() throws IOException {
+        DataOutputStream out = new DataOutputStream(
+                new FileOutputStream(dbFile));
+        Set<Entry<String, String>> tableSet = table.entrySet();
+        for (Entry<String, String> it : tableSet) {
+            writeData(out, it.getKey());
+            writeData(out, it.getValue());
+        }
+        out.flush();
+        out.close();
+    }
+
+    private void writeData(DataOutputStream out, String toWrite)
+            throws IOException {
+        int len = toWrite.length();
+        out.writeInt(len * 2);
+        out.writeChars(toWrite);
+    }
+
+    public boolean isEmpty() {
+        return table.isEmpty();
+    }
+
+    public int size() {
+        return table.size();
+    }
+
+    @Override
+    public String getName() {
+        return dbFile.getName();
+    }
+
+    @Override
+    public int commit() {
+        return 0;
+    }
+
+    @Override
+    public int rollback() {
+        return 0;
+    }
+}