1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * Permission is hereby granted, free of charge, to any person obtaining a copy
5    * of this software and associated documentation files (the "Software"), to deal
6    * in the Software without restriction, including without limitation the rights
7    * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8    * copies of the Software, and to permit persons to whom the Software is
9    * furnished to do so, subject to the following conditions:
10   *
11   * The above copyright notice and this permission notice shall be included in
12   * all copies or substantial portions of the Software.
13   *
14   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20   * SOFTWARE.
21   */
22  
23  package com.liferay.portal.dao.shard;
24  
25  import com.liferay.portal.kernel.log.Log;
26  import com.liferay.portal.kernel.log.LogFactoryUtil;
27  import com.liferay.portal.util.PropsValues;
28  
29  import javax.sql.DataSource;
30  
31  /**
32   * <a href="ShardUtil.java.html"><b><i>View Source</i></b></a>
33   *
34   * @author Alexander Chow
35   *
36   */
37  public class ShardUtil {
38  
39      public static String COMPANY_SCOPE = "COMPANY_SCOPE";
40  
41      public static DataSource getDataSource() {
42          return _shardAdvice.getDataSource();
43      }
44  
45      public static ShardSelector getShardSelector() {
46          return _shardSelector;
47      }
48  
49      public static String popCompanyService() {
50          String value = null;
51  
52          if (_shardAdvice != null) {
53              value = _shardAdvice.popCompanyService();
54          }
55  
56          return value;
57      }
58  
59      public static void pushCompanyService(long companyId) {
60          if (_shardAdvice != null) {
61              _shardAdvice.pushCompanyService(companyId);
62          }
63      }
64  
65      public static void pushCompanyService(String shardName) {
66          if (_shardAdvice != null) {
67              _shardAdvice.pushCompanyService(shardName);
68          }
69      }
70  
71      public void setShardAdvice(ShardAdvice shardAdvice) {
72          _shardAdvice = shardAdvice;
73      }
74  
75      private static Log _log = LogFactoryUtil.getLog(ShardUtil.class);
76  
77      private static ShardAdvice _shardAdvice;
78      private static ShardSelector _shardSelector;
79  
80      static {
81          try {
82              _shardSelector = (ShardSelector)Class.forName(
83                  PropsValues.SHARD_SELECTOR).newInstance();
84          }
85          catch (Exception e) {
86              _log.error(e, e);
87          }
88      }
89  
90  }