-package fi.vtt.simantics.procore.internal;\r
-\r
-import java.io.File;\r
-import java.io.FileInputStream;\r
-import java.io.FileOutputStream;\r
-import java.io.IOException;\r
-import java.io.InputStream;\r
-import java.io.OutputStream;\r
-import java.nio.ByteBuffer;\r
-\r
-import org.simantics.db.common.utils.Logger;\r
-import org.simantics.db.exception.DatabaseException;\r
-import org.simantics.db.exception.RuntimeDatabaseException;\r
-\r
-\r
-public class ReservedIds {\r
- private File file = null;\r
- private boolean loaded = false;\r
- private Data data = new Data();\r
- public static class Data {\r
- int idCount = 0;\r
- long nextId = 0;\r
- }\r
- ReservedIds(File workDir, String id) {\r
- if (!workDir.isDirectory()) {\r
- if (!workDir.mkdir())\r
- throw new RuntimeDatabaseException("Work directory not correct: " + workDir.getPath());\r
- }\r
- String name = workDir.getPath() + File.separator + "reservedIds." + id;\r
- file = new File(name);\r
- }\r
- public boolean loaded() {\r
- return loaded; \r
- }\r
- public Data load() {\r
- try {\r
- if(file == null || !file.exists())\r
- return data;\r
- loadInternal();\r
- loaded = true;\r
- } catch (IOException e) {\r
- e.printStackTrace();\r
- Logger.defaultLogError("Failed to load reserved ids.", e);\r
- return null;\r
- }\r
- return data;\r
- }\r
- public void create(int idCount, long nextId) {\r
- data.idCount = idCount;\r
- data.nextId = nextId;\r
- saveNoThrow(idCount, nextId);\r
- loaded = true;\r
- }\r
- public void saveNoThrow(int idCount, long nextId) {\r
- try {\r
- save(idCount, nextId);\r
- } catch (IOException e) {\r
- e.printStackTrace();\r
- Logger.defaultLogError("Failed to load reserved ids.", e);\r
- }\r
- }\r
- public void save(int idCount, long nextId)\r
- throws IOException {\r
- data.idCount = idCount;\r
- data.nextId = nextId;\r
- OutputStream stream = new FileOutputStream(file);\r
- try {\r
- byte[] bytes = new byte[12];\r
- ByteBuffer bb = ByteBuffer.wrap(bytes);\r
- bb.putInt(idCount);\r
- bb.putLong(nextId);\r
- stream.write(bb.array());\r
- stream.close();\r
- } finally {\r
- stream.close();\r
- }\r
- loaded = true;\r
- }\r
- public void mergeToFile(File toFile)\r
- throws DatabaseException {\r
- if (!loaded)\r
- load();\r
- long next = data.nextId + data.idCount;\r
- if (toFile.isDirectory()) {\r
- toFile = new File(toFile, file.getName());\r
- }else if (toFile.exists()) {\r
- try {\r
- InputStream is = null;\r
- try {\r
- is = new FileInputStream(toFile);\r
- byte[] bytes = new byte[8];\r
- int n = is.read(bytes);\r
- if (n == 8) {\r
- ByteBuffer bb = ByteBuffer.wrap(bytes);\r
- long oldNext = bb.getLong();\r
- if (oldNext > next)\r
- next = oldNext;\r
- }\r
- } finally {\r
- is.close();\r
- }\r
- } catch (IOException e) {\r
- String msg = "Could not open file " + toFile.getAbsolutePath() + ".";\r
- Logger.defaultLogError(msg, e);\r
- throw new DatabaseException(msg, e);\r
- }\r
- }\r
- try{\r
- OutputStream stream = new FileOutputStream(toFile);\r
- try {\r
- byte[] bytes = new byte[8];\r
- ByteBuffer bb = ByteBuffer.wrap(bytes);\r
- bb.putLong(next);\r
- stream.write(bb.array());\r
- stream.close();\r
- } finally {\r
- stream.close();\r
- }\r
- } catch (IOException e) {\r
- String msg = "Could not save file " + toFile.getAbsolutePath() + ".";\r
- Logger.defaultLogError(msg, e);\r
- throw new DatabaseException(msg, e);\r
- }\r
- }\r
- private void loadInternal()\r
- throws IOException {\r
- InputStream is = new FileInputStream(file);\r
- try {\r
- byte[] bytes = new byte[12];\r
- int n = is.read(bytes);\r
- if (n != 12)\r
- return;\r
- ByteBuffer bb = ByteBuffer.wrap(bytes);\r
- data.idCount = bb.getInt();\r
- data.nextId = bb.getLong();\r
- } finally {\r
- is.close();\r
- }\r
- }\r
-}\r
+package fi.vtt.simantics.procore.internal;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.nio.ByteBuffer;
+
+import org.simantics.db.common.utils.Logger;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.exception.RuntimeDatabaseException;
+
+
+public class ReservedIds {
+ private File file = null;
+ private boolean loaded = false;
+ private Data data = new Data();
+ public static class Data {
+ int idCount = 0;
+ long nextId = 0;
+ }
+ ReservedIds(File workDir, String id) {
+ if (!workDir.isDirectory()) {
+ if (!workDir.mkdir())
+ throw new RuntimeDatabaseException("Work directory not correct: " + workDir.getPath());
+ }
+ String name = workDir.getPath() + File.separator + "reservedIds." + id;
+ file = new File(name);
+ }
+ public boolean loaded() {
+ return loaded;
+ }
+ public Data load() {
+ try {
+ if(file == null || !file.exists())
+ return data;
+ loadInternal();
+ loaded = true;
+ } catch (IOException e) {
+ e.printStackTrace();
+ Logger.defaultLogError("Failed to load reserved ids.", e);
+ return null;
+ }
+ return data;
+ }
+ public void create(int idCount, long nextId) {
+ data.idCount = idCount;
+ data.nextId = nextId;
+ saveNoThrow(idCount, nextId);
+ loaded = true;
+ }
+ public void saveNoThrow(int idCount, long nextId) {
+ try {
+ save(idCount, nextId);
+ } catch (IOException e) {
+ e.printStackTrace();
+ Logger.defaultLogError("Failed to load reserved ids.", e);
+ }
+ }
+ public void save(int idCount, long nextId)
+ throws IOException {
+ data.idCount = idCount;
+ data.nextId = nextId;
+ OutputStream stream = new FileOutputStream(file);
+ try {
+ byte[] bytes = new byte[12];
+ ByteBuffer bb = ByteBuffer.wrap(bytes);
+ bb.putInt(idCount);
+ bb.putLong(nextId);
+ stream.write(bb.array());
+ stream.close();
+ } finally {
+ stream.close();
+ }
+ loaded = true;
+ }
+ public void mergeToFile(File toFile)
+ throws DatabaseException {
+ if (!loaded)
+ load();
+ long next = data.nextId + data.idCount;
+ if (toFile.isDirectory()) {
+ toFile = new File(toFile, file.getName());
+ }else if (toFile.exists()) {
+ try {
+ InputStream is = null;
+ try {
+ is = new FileInputStream(toFile);
+ byte[] bytes = new byte[8];
+ int n = is.read(bytes);
+ if (n == 8) {
+ ByteBuffer bb = ByteBuffer.wrap(bytes);
+ long oldNext = bb.getLong();
+ if (oldNext > next)
+ next = oldNext;
+ }
+ } finally {
+ is.close();
+ }
+ } catch (IOException e) {
+ String msg = "Could not open file " + toFile.getAbsolutePath() + ".";
+ Logger.defaultLogError(msg, e);
+ throw new DatabaseException(msg, e);
+ }
+ }
+ try{
+ OutputStream stream = new FileOutputStream(toFile);
+ try {
+ byte[] bytes = new byte[8];
+ ByteBuffer bb = ByteBuffer.wrap(bytes);
+ bb.putLong(next);
+ stream.write(bb.array());
+ stream.close();
+ } finally {
+ stream.close();
+ }
+ } catch (IOException e) {
+ String msg = "Could not save file " + toFile.getAbsolutePath() + ".";
+ Logger.defaultLogError(msg, e);
+ throw new DatabaseException(msg, e);
+ }
+ }
+ private void loadInternal()
+ throws IOException {
+ InputStream is = new FileInputStream(file);
+ try {
+ byte[] bytes = new byte[12];
+ int n = is.read(bytes);
+ if (n != 12)
+ return;
+ ByteBuffer bb = ByteBuffer.wrap(bytes);
+ data.idCount = bb.getInt();
+ data.nextId = bb.getLong();
+ } finally {
+ is.close();
+ }
+ }
+}
Code Review / simantics / platform.git / blobdiff