Examples with Collection com.tvd12.dahlia.core.entity.Collection used on opensource projects

Example 21 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project dahlia by youngmonkeys.

the class DatabaseTest method main.

public static void main(String[] args) {
    // deleteDataDir();
    DahliaCoreLoader loader = new DahliaCoreLoader().storageDirectory("data");
    DahliaCore dahlia = loader.load();
    CommandExecutor commandExecutor = dahlia.getCommandExecutor();
    DatabaseSetting databaseSetting = new DatabaseSetting();
    databaseSetting.setDatabaseName("hello");
    CommandCreateDatabase commandCreateDatabase = new CommandCreateDatabase(databaseSetting);
    Database database = null;
    try {
        database = commandExecutor.execute(commandCreateDatabase);
    } catch (DatabaseExistedException e) {
        database = dahlia.getDatabases().getDatabase("hello");
    }
    CollectionSetting collectionSetting = new CollectionSetting();
    collectionSetting.setCollectionId(1);
    collectionSetting.setCollectionName("test");
    Map<String, FieldSetting> fieldSettings = new HashMap<>();
    FieldLongSetting fieldIdSetting = new FieldLongSetting();
    fieldIdSetting.setNullable(true);
    fieldIdSetting.setDefaultValue(100L);
    fieldSettings.put("_id", fieldIdSetting);
    FieldLongSetting fieldValueSetting = new FieldLongSetting();
    fieldValueSetting.setNullable(true);
    fieldValueSetting.setDefaultValue(300L);
    fieldSettings.put("value", fieldValueSetting);
    FieldTextSetting fieldNameSetting = new FieldTextSetting();
    fieldNameSetting.setNullable(false);
    fieldSettings.put("name", fieldNameSetting);
    collectionSetting.setFields(fieldSettings);
    System.out.println(collectionSetting.toMap());
    CommandCreateCollection commandCreateCollection = new CommandCreateCollection(database.getId(), collectionSetting);
    Collection collection = null;
    try {
        collection = commandExecutor.execute(commandCreateCollection);
    } catch (CollectionExistedException e) {
        collection = database.getCollection("test");
    }
    EzyObject insertOneData = newObjectBuilder().append("_id", 2L).append("value", 323L).append("name", "dungtv").build();
    CommandInsertOne commandInsertOne = new CommandInsertOne(collection.getId(), insertOneData);
    try {
        EzyObject insertOneResult = commandExecutor.execute(commandInsertOne);
        System.out.println("insert one result: " + insertOneResult);
    } catch (DuplicatedIdException e) {
    } catch (Exception e) {
        e.printStackTrace();
    }
    long v1 = 1;
    double v2 = 1.1;
    System.out.println(v2 == v1);
    // EzyObject query1 = newObjectBuilder()
    // .append("_id", newObjectBuilder().append(Keywords.LESS_THAN_EQUAL, 3L))
    // .build();
    // CommandFindOne findOne = new CommandFindOne(collection.getId(), query1);
    // EzyObject findOneResult = commandExecutor.execute(findOne);
    // System.out.println("findOneResult: " + findOneResult);
    // EzyObject query2 = newObjectBuilder()
    // .append(Keywords.OR, newArrayBuilder()
    // .append(newObjectBuilder().append("_id", newObjectBuilder().append(Keywords.LESS_THAN_EQUAL, 3L)))
    // .append(newObjectBuilder().append("value", 223))
    // )
    // .build();
    EzyObject query3 = newObjectBuilder().append(Keywords.OR, newArrayBuilder().append(newObjectBuilder().append(Keywords.LESS_THAN_EQUAL, newObjectBuilder().append("_id", 3L))).append(newObjectBuilder().append("value", 223))).build();
    FindOptions options = new FindOptions().setSkip(0).setLimit(10);
    CommandFind commandFind = new CommandFind(collection.getId(), query3, options.toObject());
    EzyArray findResult = commandExecutor.execute(commandFind);
    System.out.println("findResult = " + findResult);
    Long size = dahlia.execute(new CommandCount(collection.getId()));
    System.out.println("size: " + size);
}

Example 22 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project dahlia by youngmonkeys.

the class CommandCreateCollectionHandler method handle.

@Override
public Object handle(CommandCreateCollection command) {
    int databaseId = command.getDatabaseId();
    CollectionSetting setting = command.getSetting();
    String collectionName = setting.getCollectionName();
    Database database = databases.getDatabase(databaseId);
    if (database == null)
        throw new DatabaseNotFoundException(databaseId);
    Collection existedCollection = database.getCollection(collectionName);
    if (existedCollection != null)
        throw new CollectionExistedException(collectionName);
    int recordSize = recordSizeReader.read(setting.getAllFields());
    setting.setRecordSize(recordSize);
    Collection collection = collectionFactory.newCollection(setting);
    synchronized (runtimeSetting) {
        runtimeSetting.setMaxCollectionId(collection.getId());
        storage.storeRuntimeSetting(runtimeSetting);
    }
    database.addCollection(collection);
    databases.addCollection(collection);
    CollectionStorage collectionStorage = collectionStorageFactory.newCollectionStorage(collectionName, database.getName());
    storage.addCollectionStorage(collection.getId(), collectionStorage);
    collectionStorage.storeSetting(setting);
    return collection;
}

Example 23 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project dahlia by youngmonkeys.

the class CommandSaveHandler method handle.

@Override
public Object handle(CommandSave command) {
    int collectionId = command.getCollectionId();
    EzyArray data = command.getData();
    Collection collection = databases.getCollection(collectionId);
    CollectionSetting setting = collection.getSetting();
    FieldSetting sId = setting.getId();
    Map<String, FieldSetting> sFields = setting.getFields();
    long dataSize = collection.getDataSize();
    CollectionStorage collectionStorage = storage.getCollectionStorage(collectionId);
    EzyArray answerItems = EzyEntityFactory.newArray();
    synchronized (collection) {
        for (int i = 0; i < data.size(); ++i) {
            EzyObject answerItem = EzyEntityFactory.newObject();
            EzyObject item = data.get(i);
            Comparable id = item.get(Constants.FIELD_ID);
            if (id != null) {
                Record existed = collection.findById(id);
                if (existed != null)
                    answerItem.put(Constants.RESULT_FIELD_EXISTED, true);
            } else {
                while (true) {
                    id = UUID.randomUUID();
                    Record existed = collection.findById(id);
                    if (existed == null)
                        break;
                }
            }
            Record record = new Record(id, dataSize);
            collection.insert(record);
            collection.increaseDataSize();
            collectionStorage.storeRecord(record, sId, sFields, item);
            answerItem.put(Constants.FIELD_ID, id);
            answerItems.add(id);
        }
    }
    return answerItems;
}

Example 24 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project dahlia by youngmonkeys.

the class CommandUpdateHandler method handle.

@Override
public Object handle(CommandUpdate command) {
    int collectionId = command.getCollectionId();
    Collection collection = databases.getCollection(collectionId);
    if (collection == null)
        throw new CollectionNotFoundException(collectionId);
    EzyObject query = command.getQuery();
    Predicate<EzyObject> predicate = queryToPredicate.toPredicate(query);
    CollectionSetting setting = collection.getSetting();
    CollectionStorage collectionStorage = storage.getCollectionStorage(collectionId);
    FieldSetting sId = setting.getId();
    Map<String, FieldSetting> sFields = setting.getFields();
    EzyObject update = command.getUpdate();
    List<EzyPair<Record, EzyObject>> updateItems = new ArrayList<>();
    synchronized (collection) {
        collection.forEach(new RecordConsumer() {

            @Override
            public void accept(Record r) {
                EzyObject value = collectionStorage.readRecord(r, sId, sFields);
                boolean accepted = predicate.test(value);
                if (accepted)
                    updateItems.add(new EzyPair<>(r, value));
            }
        });
        for (EzyPair<Record, EzyObject> pair : updateItems) {
            Record record = pair.getKey();
            EzyObject updateItem = pair.getValue();
            updateItem(updateItem, update);
            collectionStorage.storeRecord(record, sId, sFields, updateItem);
        }
    }
    EzyArray answer = EzyEntityFactory.newArray();
    for (EzyPair<Record, EzyObject> pair : updateItems) {
        EzyObject answerItem = EzyEntityFactory.newObject();
        EzyObject updateItem = pair.getValue();
        answerItem.put(Constants.FIELD_ID, updateItem.get(Constants.FIELD_ID));
    }
    return answer;
}

Example 25 with Collection

use of com.tvd12.dahlia.core.entity.Collection in project dahlia by youngmonkeys.

the class DahliaCoreLoader method newDatabases.

protected Databases newDatabases(DatabaseFactory databaseFactory, CollectionFactory collectionFactory, Map<String, DatabaseSetting> databaseSettings, Map<String, List<CollectionSetting>> collectionSettingsMap) {
    Databases databases = new Databases();
    for (String databaseName : databaseSettings.keySet()) {
        DatabaseSetting databaseSetting = databaseSettings.get(databaseName);
        Database database = databaseFactory.createDatabase(databaseSetting);
        List<CollectionSetting> collectionSettings = collectionSettingsMap.get(databaseName);
        for (CollectionSetting collectionSetting : collectionSettings) {
            Collection collection = collectionFactory.createCollection(collectionSetting);
            database.addCollection(collection);
            databases.addCollection(collection);
        }
        databases.addDatabase(database);
    }
    return databases;
}