1   /**
2    * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
3    *
4    * This library is free software; you can redistribute it and/or modify it under
5    * the terms of the GNU Lesser General Public License as published by the Free
6    * Software Foundation; either version 2.1 of the License, or (at your option)
7    * any later version.
8    *
9    * This library is distributed in the hope that it will be useful, but WITHOUT
10   * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
11   * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
12   * details.
13   */
14  
15  package com.liferay.portal.dao.shard;
16  
17  import com.liferay.portal.kernel.log.Log;
18  import com.liferay.portal.kernel.log.LogFactoryUtil;
19  import com.liferay.portal.util.PropsValues;
20  
21  import javax.sql.DataSource;
22  
23  /**
24   * <a href="ShardUtil.java.html"><b><i>View Source</i></b></a>
25   *
26   * @author Alexander Chow
27   */
28  public class ShardUtil {
29  
30      public static String COMPANY_SCOPE = "COMPANY_SCOPE";
31  
32      public static DataSource getDataSource() {
33          return _shardAdvice.getDataSource();
34      }
35  
36      public static ShardSelector getShardSelector() {
37          return _shardSelector;
38      }
39  
40      public static boolean isEnabled() {
41          if (_shardAdvice != null) {
42              return true;
43          }
44          else {
45              return false;
46          }
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  }