com.ebay.cloud.cms.dalapi.persistence.impl.embed.EmbedCreateCommand.java Source code

Java tutorial

Introduction

Here is the source code for com.ebay.cloud.cms.dalapi.persistence.impl.embed.EmbedCreateCommand.java

Source

/*
Copyright [2013-2014] eBay Software Foundation
    
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
    
http://www.apache.org/licenses/LICENSE-2.0
    
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

/* 
Copyright 2012 eBay Software Foundation 
    
Licensed under the Apache License, Version 2.0 (the "License"); 
you may not use this file except in compliance with the License. 
You may obtain a copy of the License at 
    
http://www.apache.org/licenses/LICENSE-2.0 
    
Unless required by applicable law or agreed to in writing, software 
distributed under the License is distributed on an "AS IS" BASIS, 
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
See the License for the specific language governing permissions and 
limitations under the License. 
*/

package com.ebay.cloud.cms.dalapi.persistence.impl.embed;

import java.util.BitSet;
import java.util.Date;

import com.ebay.cloud.cms.dal.entity.AbstractEntityIDHelper;
import com.ebay.cloud.cms.dal.exception.CmsDalException;
import com.ebay.cloud.cms.dal.exception.CmsDalException.DalErrCodeEnum;
import com.ebay.cloud.cms.dal.persistence.IPersistenceCommand;
import com.ebay.cloud.cms.dal.persistence.MongoExecutor;
import com.ebay.cloud.cms.dal.persistence.PersistenceContext;
import com.ebay.cloud.cms.dalapi.entity.impl.BsonEntity;
import com.ebay.cloud.cms.metadata.model.InternalFieldEnum;
import com.ebay.cloud.cms.metadata.model.MetaClass;
import com.ebay.cloud.cms.metadata.model.MetaField;
import com.ebay.cloud.cms.metadata.model.MetaField.CardinalityEnum;
import com.ebay.cloud.cms.metadata.model.MetaField.FieldProperty;
import com.ebay.cloud.cms.metadata.model.MetaRelationship;
import com.mongodb.BasicDBList;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.mongodb.WriteResult;

/**
 * create embedded entity
 * 
 * https://jira.mongodb.org/browse/SERVER-831
 * 
 * @author jianxu1, xjiang, zhihzhang
 *
 */
@SuppressWarnings("static-access")
//for inner field, only update root and leaf entity now
public class EmbedCreateCommand extends AbstractEmbedCommand implements IPersistenceCommand {

    private final BsonEntity entity;

    public EmbedCreateCommand(BsonEntity entity, AbstractEntityIDHelper helper) {
        super(entity.getBranchId(), helper);
        this.entity = entity;
    }

    @Override
    public void execute(PersistenceContext context) {
        MetaClass rootMetaClass = getRootEntityMetaClass(entity.getId(), context);
        String parentId = helper.getParentId(entity.getId());
        BitSet parentBits = helper.checkArrayOnPath(parentId, rootMetaClass);

        // query root object with parent id
        DBObject rootQuery = buildGetQuery(parentBits, parentId, rootMetaClass);

        String id = parentId;
        if (parentBits.cardinality() == 0) {
            id = entity.getId();
        }
        DBObject rootFields = buildGetRootFields(id, rootMetaClass);
        DBObject rootObject = MongoExecutor.findOne(context, rootMetaClass, rootQuery, rootFields);
        if (rootObject == null) {
            throw new CmsDalException(DalErrCodeEnum.ENTITY_NOT_FOUND,
                    "Create, parent document doesn't exist!  " + parentId);
        }

        // add embed document into parent document
        DBObject createQuery = buildModifyQuery(rootQuery, entity);
        int newVersion = (Integer) rootObject.get(InternalFieldEnum.VERSION.getDbName()) + 1;
        String parentPath = getUpdatePath(parentBits, parentId, rootObject, rootMetaClass);
        MetaRelationship lastField = helper.getLastMetaField(entity.getId(), rootMetaClass);
        DBObject createBody = buildCreateBody(lastField, parentPath, entity, newVersion, rootMetaClass, rootObject,
                parentId);

        WriteResult result = MongoExecutor.update(context, rootMetaClass, createQuery, createBody);
        if (result.getN() == 0) {
            // something happens between get and replace
            throw new CmsDalException(DalErrCodeEnum.VERSION_CONFLICT, "Version check fails");
        }
    }

    public DBObject buildCreateBody(MetaRelationship lastField, String parentPath, BsonEntity entity,
            int newVersion, MetaClass rootMetaClass, DBObject rootObject, String parentId) {
        BasicDBObject embedParentObject = (BasicDBObject) EmbedDBObjectFilter.filter(parentId, rootObject,
                rootMetaClass, null, helper);
        if (embedParentObject == null) {
            throw new CmsDalException(DalErrCodeEnum.ENTITY_NOT_FOUND,
                    "Create, can not find embed field with Id: " + parentId);
        }

        embedParentObject.remove("_id");

        if (lastField.getCardinality() == CardinalityEnum.Many) {
            BasicDBObject obj = (BasicDBObject) embedParentObject.get(lastField.getDbName());
            BasicDBList valueList = null;
            if (obj != null) {
                valueList = (BasicDBList) obj.get(MetaField.VALUE_KEY);
            }

            if (valueList == null) {
                if (obj == null) {
                    obj = new BasicDBObject();
                    embedParentObject.put(lastField.getDbName(), obj);
                }
                valueList = new BasicDBList();
                valueList.add(entity.getNode());
                obj.put(FieldProperty.LENGTH.getDbName(), 1);
                obj.put(FieldProperty.TIMESTAMP.getDbName(), new Date());
                obj.put(MetaField.VALUE_KEY, valueList);
            } else {
                int size = valueList.size();
                valueList.add(entity.getNode());
                obj.put(FieldProperty.LENGTH.getDbName(), size + 1);
                obj.put(FieldProperty.TIMESTAMP.getDbName(), new Date());
                obj.put(MetaField.VALUE_KEY, valueList);
            }
        } else {
            DBObject vObj = new BasicDBObject();
            vObj.put(MetaField.VALUE_KEY, entity.getNode());
            embedParentObject.put(lastField.getDbName(), vObj);
        }

        BasicDBObject setModifyObject = new BasicDBObject();
        BasicDBObject obj = (BasicDBObject) rootObject.get(embedFieldName);
        if (obj == null) {
            throw new CmsDalException(DalErrCodeEnum.ENTITY_NOT_FOUND,
                    "Create, can not find embed field with Id: " + this.entity.getId());
        }

        setModifyObject.put(embedFieldName, obj);
        BasicDBObject modifyBody = new BasicDBObject();
        modifyBody.put("$set", setModifyObject);

        // increase version on root document
        BasicDBObject versionObject = new BasicDBObject();
        versionObject.put(InternalFieldEnum.VERSION.getDbName(), 1);
        modifyBody.put("$inc", versionObject);

        buildRootUpdateObject(entity, null, modifyBody, rootMetaClass);

        return modifyBody;
    }
}