001    /**
002     * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.service.impl;
016    
017    import com.liferay.portal.kernel.exception.PortalException;
018    import com.liferay.portal.kernel.util.Validator;
019    import com.liferay.portal.model.Shard;
020    import com.liferay.portal.service.base.ShardLocalServiceBaseImpl;
021    import com.liferay.portal.util.PropsValues;
022    
023    /**
024     * @author Brian Wing Shun Chan
025     */
026    public class ShardLocalServiceImpl extends ShardLocalServiceBaseImpl {
027    
028            @Override
029            public Shard addShard(String className, long classPK, String name) {
030                    long classNameId = classNameLocalService.getClassNameId(className);
031    
032                    if (Validator.isNull(name)) {
033                            name = PropsValues.SHARD_DEFAULT_NAME;
034                    }
035    
036                    long shardId = counterLocalService.increment();
037    
038                    Shard shard = shardPersistence.create(shardId);
039    
040                    shard.setClassNameId(classNameId);
041                    shard.setClassPK(classPK);
042                    shard.setName(name);
043    
044                    shardPersistence.update(shard);
045    
046                    return shard;
047            }
048    
049            @Override
050            public Shard getShard(String className, long classPK)
051                    throws PortalException {
052    
053                    long classNameId = classNameLocalService.getClassNameId(className);
054    
055                    return shardPersistence.findByC_C(classNameId, classPK);
056            }
057    
058    }