db4o java_java com.db4o 类

packageorg.rx.repository.db4o.impl;importcom.db4o.Db4o;importcom.db4o.ObjectContainer;importcom.db4o.ObjectSet;importcom.db4o.config.Configuration;importlombok.SneakyThrows;importorg.rx.api.dto.common.PagedResponse;importorg.rx.api.dto.common.PagingRequest;importorg.rx.beans.BeanMapper;importorg.rx.beans.DateTime;importorg.rx.common.App;importorg.rx.common.InvalidOperationException;importorg.rx.common.NQuery;importorg.rx.repository.db4o.IRepository;importorg.rx.repository.db4o.DataObject;importorg.springframework.stereotype.Component;importjava.util.ArrayList;importjava.util.Comparator;importjava.util.List;importjava.util.UUID;importjava.util.function.Function;importjava.util.function.Predicate;import staticorg.rx.common.Contract.require;

@Componentpublic class Db4oRepository implements IRepository{privateString dbPath;privateConfiguration config;

@SneakyThrowspublicDb4oRepository() {

dbPath= App.readSetting("app.repository.dbFile");if (dbPath == null) {throw new InvalidOperationException("app.repository.dbFile is empty");

}

String dir=dbPath;int i = dir.lastIndexOf("/");if (i != -1) {

dir= dir.substring(0, i);

}

App.createDirectory(dir);

config=Db4o.newConfiguration();

}protected R invoke(Functionfunc) {return NQuery.of(invoke((Function[]) newFunction[]{func})).firstOrDefault();

}protected synchronized List invoke(Function... funcList) {

require(funcList);

List result = new ArrayList<>();//ObjectContainer db = Db4o.openFile(config, dbPath);

ObjectContainer db = App.getOrStore("Db4oRepository-threadDb", k ->Db4o.openFile(config, dbPath));try{for (Functionfunction : funcList) {

result.add(function.apply(db));

}

db.commit();

}catch(Exception e) {

db.rollback();throwe;

}//finally {//db.close();//}

returnresult;

}publicT save(T model) {

require(model);if (!(model instanceofDataObject)) {throw new IllegalArgumentException("model is not a DataObject");

}return invoke(db ->{

T dataObj= single(p ->p.getId().equals(model.getId()));if (dataObj != null) {

dataObj= BeanMapper.getInstance().map(model, dataObj, BeanMapper.Flags.NonCheckMatch |BeanMapper.Flags.SkipNull);

}else{

dataObj=model;

}if (dataObj.getId() == null) {

dataObj.setId(UUID.randomUUID());

}if (dataObj.getCreateTime() == null) {

dataObj.setCreateTime(DateTime.now());

}

dataObj.setModifyTime(DateTime.now());

db.store(dataObj);returndataObj;

});

}

@OverridepublicT delete(UUID id) {

T model=single(id);if (model == null) {return null;

}

model.setDeleted(true);returnsave(model);

}

@OverridepublicT single(UUID id) {return single(p ->p.getId().equals(id));

}

@Overridepublic T single(Predicatecondition) {returnNQuery.of(list(condition)).firstOrDefault();

}

@Overridepublic long count(Predicatecondition) {return executeReader(condition, null, false).count();

}

@Overridepublic List list(Predicatecondition) {return list(condition, null);

}

@Overridepublic List list(Predicate condition, FunctionkeySelector) {return executeReader(condition, keySelector, false).toList();

}

@Overridepublic List listDescending(Predicate condition, FunctionkeySelector) {return executeReader(condition, keySelector, true).toList();

}

@Overridepublic PagedResponse page(Predicate condition, FunctionkeySelector, PagingRequest pagingParam) {

require(pagingParam);

NQuery nQuery = executeReader(condition, keySelector, false);returnpagingParam.page(nQuery);

}

@Overridepublic PagedResponse pageDescending(Predicate condition, FunctionkeySelector, PagingRequest pagingParam) {

require(pagingParam);

NQuery nQuery = executeReader(condition, keySelector, true);returnpagingParam.page(nQuery);

}private NQuery executeReader(Predicate condition, Function keySelector, booleanisDescending) {

require(condition);

com.db4o.query.Predicate predicate = new com.db4o.query.Predicate() {public booleanmatch(T candidate) {return !candidate.isDeleted() &&condition.test(candidate);

}

};return invoke(db ->{

ObjectSetobjectSet;if (keySelector == null) {

objectSet=db.query(predicate);

}else{

Comparator comparator =getComparator(keySelector);if(isDescending) {

comparator=comparator.reversed();

}

objectSet=db.query(predicate, comparator);

}returnNQuery.of(objectSet);

});

}private Comparator getComparator(FunctionkeySelector) {if (keySelector == null) {return(Comparator) Comparator.naturalOrder();

}returnNQuery.getComparator(keySelector);

}

}