/* This file is part of the db4o object database http://www.db4o.com
Copyright (C) 2004 - 2011 Versant Corporation http://www.versant.com
db4o is free software; you can redistribute it and/or modify it under
the terms of version 3 of the GNU General Public License as published
by the Free Software Foundation.
db4o is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License along
with this program. If not, see http://www.gnu.org/licenses/. */
package com.db4o.internal.ids;
import com.db4o.ext.*;
import com.db4o.foundation.*;
import com.db4o.internal.*;
/**
* @exclude
*/
public class SequentialIdGenerator {
private final int _minValidId;
private final int _maxValidId;
private int _idGenerator;
private boolean _overflow;
private int _lastIdGenerator;
private final Function4<Integer, Integer> _findFreeId;
public SequentialIdGenerator(Function4<Integer, Integer> findFreeId, int initialValue, int minValidId, int maxValidId) {
_findFreeId = findFreeId;
_minValidId = minValidId;
_maxValidId = maxValidId;
initializeGenerator(initialValue);
}
public SequentialIdGenerator(Function4<Integer, Integer> findFreeId, int minValidId, int maxValidId) {
this(findFreeId, minValidId - 1, minValidId, maxValidId);
}
public void read(ByteArrayBuffer buffer) {
initializeGenerator(buffer.readInt());
}
private void initializeGenerator(int val){
if(val < 0){
_overflow = true;
_idGenerator = - val;
}else {
_idGenerator = val;
}
_lastIdGenerator = _idGenerator;
}
public void write(ByteArrayBuffer buffer) {
buffer.writeInt(persistentGeneratorValue());
}
public int persistentGeneratorValue() {
return _overflow ? -_idGenerator : _idGenerator;
}
public int newId() {
adjustIdGenerator(_idGenerator);
if(! _overflow){
return _idGenerator;
}
int id = _findFreeId.apply(_idGenerator);
if(id > 0){
adjustIdGenerator(id - 1);
return id;
}
id = _findFreeId.apply(_minValidId);
if(id > 0){
adjustIdGenerator(id - 1);
return id;
}
throw new Db4oFatalException("Out of IDs");
}
private void adjustIdGenerator(int id) {
if(id == _maxValidId){
_idGenerator = _minValidId;
_overflow = true;
return;
}
_idGenerator = id + 1;
}
public int marshalledLength() {
return Const4.INT_LENGTH;
}
public boolean isDirty(){
return _idGenerator != _lastIdGenerator;
}
public void setClean() {
_lastIdGenerator = _idGenerator;
}
}