Java tutorial
/* 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.entmgr.branch.impl; import java.util.ArrayList; import java.util.LinkedList; import java.util.List; import com.ebay.cloud.cms.consts.CMSConsts; import com.ebay.cloud.cms.dal.entity.IEntity; import com.ebay.cloud.cms.dal.persistence.ConsistentPolicy; import com.ebay.cloud.cms.dal.persistence.IPersistenceService; import com.ebay.cloud.cms.dal.persistence.PersistenceContext; import com.ebay.cloud.cms.dal.persistence.PersistenceContext.DBCollectionPolicy; import com.ebay.cloud.cms.dal.search.ISearchQuery; import com.ebay.cloud.cms.dal.search.ISearchService; import com.ebay.cloud.cms.dal.search.SearchOption; import com.ebay.cloud.cms.dal.search.SearchProjection; import com.ebay.cloud.cms.dal.search.SearchResult; import com.ebay.cloud.cms.dal.search.impl.criteria.FieldSearchCriteria; import com.ebay.cloud.cms.dal.search.impl.field.ProjectionField; import com.ebay.cloud.cms.dal.search.impl.query.SearchQuery; import com.ebay.cloud.cms.entmgr.branch.IBranch; import com.ebay.cloud.cms.entmgr.branch.IBranchService; import com.ebay.cloud.cms.entmgr.entity.EntityContext; import com.ebay.cloud.cms.metadata.model.MetaClass; import com.ebay.cloud.cms.metadata.model.MetaRelationship; import com.ebay.cloud.cms.metadata.model.MetaRelationship.RelationTypeEnum; import com.ebay.cloud.cms.metadata.model.Repository; import com.ebay.cloud.cms.metadata.model.internal.BranchMetaClass; import com.ebay.cloud.cms.metadata.service.IMetadataService; import com.ebay.cloud.cms.metadata.service.IRepositoryService; import com.ebay.cloud.cms.mongo.MongoDataSource; import com.ebay.cloud.cms.utils.CheckConditions; import com.mongodb.DBCollection; /** * * @author jianxu1 * @date 2012/7/10 * */ public class BranchServiceImpl extends AbstractBranchCommand implements IBranchService { // private static final Logger log = // LoggerFactory.getLogger(BranchServiceImpl.class); private final IRepositoryService repositoryService; private final ISearchService searchService; private final IPersistenceService persistenceService; private BranchCache branchCache; private MongoDataSource ds; public BranchServiceImpl(MongoDataSource ds, IRepositoryService repositoryService, IPersistenceService persistenceService, ISearchService searchService) { this.repositoryService = repositoryService; this.persistenceService = persistenceService; this.searchService = searchService; this.ds = ds; branchCache = new BranchCache(); } private PersistenceContext getBranchPersistenceContext(String repoName, EntityContext entityContext) { Repository repo = repositoryService.getRepository(repoName); IMetadataService metaService = repo.getMetadataService(); PersistenceContext pc = new PersistenceContext(metaService, DBCollectionPolicy.Merged, entityContext.getConsistentPolicy(), CMSConsts.BRANCH_DB_COLL_NAME, entityContext.getRegistration()); pc.setDbConfig(entityContext.getDbConfig()); pc.setAdditionalCriteria(entityContext.getAdditionalCriteria()); pc.setFetchFieldProperties(entityContext.isFetchFieldProperty()); return pc; } /** * POST repository/{repoId}/branch/ */ @Override public IBranch createBranch(IBranch branch, EntityContext context) { CheckConditions.checkNotNull(branch); String repoName = branch.getRepositoryName(); PersistenceContext persistenceContext = getBranchPersistenceContext(repoName, context); BranchCreateCommand createCommand = new BranchCreateCommand(branch, repositoryService, persistenceService, context); createCommand.execute(persistenceContext); String newBranchId = createCommand.getCreatedBranchId(); // getBranch will lookup this branch in db and put it into cache return this.getBranch(repoName, newBranchId, context); } Branch getBranchFromDB(String repoName, String branchId, EntityContext context) { Branch branch; PersistenceContext persistenceContext = getBranchPersistenceContext(repoName, context); BranchGetCommand getCommand = new BranchGetCommand(persistenceService, branchId); ConsistentPolicy oldConsistency = persistenceContext.getConsistentPolicy(); ConsistentPolicy cp = ConsistentPolicy.PRIMARY; persistenceContext.setConsistentPolicy(cp); getCommand.execute(persistenceContext); persistenceContext.setConsistentPolicy(oldConsistency); branch = getCommand.getBranch(); return branch; } @Override public IBranch getBranch(String repoName, String branchId, EntityContext context) { IBranch branch = branchCache.getBranch(repoName, branchId); if (branch == null) { Branch b = getBranchFromDB(repoName, branchId, context); if (b != null) { branchCache.putBranch(b); branch = b; } } return branch; } @Override public List<IBranch> getMainBranches(String repoName, EntityContext context) { Repository repo = this.repositoryService.getRepository(repoName); MetaClass brancMetaData = BranchMetaClass.getMetaClass(repo); FieldSearchCriteria criteria = new FieldSearchCriteria(brancMetaData.getFieldByName(BranchMetaClass.IsMain), context.getRegistration().searchStrategy, FieldSearchCriteria.FieldOperatorEnum.EQ, Boolean.TRUE); SearchProjection searchProject = new SearchProjection(); searchProject.addField(ProjectionField.STAR); ISearchQuery query = new SearchQuery(brancMetaData, criteria, searchProject, context.getRegistration().searchStrategy); SearchOption option = new SearchOption(); PersistenceContext persistenceContext = getBranchPersistenceContext(repoName, context); SearchResult result = searchService.search(query, option, persistenceContext); List<IEntity> bsonList = result.getResultSet(); List<IBranch> branches = new ArrayList<IBranch>(bsonList.size()); for (IEntity e : bsonList) { branches.add(toBranch(e)); } branchCache.clearCache(); return branches; } public void clearBranchCache() { branchCache.clearCache(); } @Override public void ensureIndex(String repoName, List<MetaClass> metadatas, EntityContext context) { CheckConditions.checkNotNull(repoName, "Repository name can not be null"); Repository repo = this.repositoryService.getRepository(repoName); List<IBranch> branches = getMainBranches(repoName, context); List<MetaClass> allMetadatas = new LinkedList<MetaClass>(); getMetadata(metadatas, allMetadatas); for (IBranch b : branches) { // create index on main branch collection PersistenceContext persistenceContext = PersistenceContextFactory.createEntityPersistenceConext( repo.getMetadataService(), b.getId(), context.getConsistentPolicy(), context.getRegistration(), context.isFetchFieldProperty(), context.getDbConfig(), context.getAdditionalCriteria()); persistenceService.ensureIndex(allMetadatas, persistenceContext, true); } } /** * @param metadatas */ private void getMetadata(List<MetaClass> baseMetadatas, List<MetaClass> allMetadatas) { for (MetaClass baseMeta : baseMetadatas) { allMetadatas.add(baseMeta); List<MetaRelationship> baseRelationships = baseMeta.getFromReference(); if (!baseRelationships.isEmpty()) { List<MetaClass> outerMetadatas = new LinkedList<MetaClass>(); for (MetaRelationship baseRelationship : baseRelationships) { if (baseRelationship.getRelationType().equals(RelationTypeEnum.Embedded)) { MetaClass fromMeta = baseRelationship.getSourceMetaClass(); outerMetadatas.add(fromMeta); } } getMetadata(outerMetadatas, allMetadatas); } } } @Override public void deleteMetadata(String repoName, MetaClass metadata, EntityContext context) { // drop all metadata related collection for each branch Repository repo = this.repositoryService.getRepository(repoName); List<IBranch> branches = getMainBranches(repoName, context); for (IBranch b : branches) { PersistenceContext persistenceContext = PersistenceContextFactory.createEntityPersistenceConext( repo.getMetadataService(), b.getId(), context.getConsistentPolicy(), context.getRegistration(), context.isFetchFieldProperty(), context.getDbConfig(), context.getAdditionalCriteria()); persistenceContext.setMongoDataSource(ds); DBCollection coll = persistenceContext.getDBCollection(metadata); coll.drop(); } } }