1   /**
2    * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
3    *
4    * The contents of this file are subject to the terms of the Liferay Enterprise
5    * Subscription License ("License"). You may not use this file except in
6    * compliance with the License. You can obtain a copy of the License by
7    * contacting Liferay, Inc. See the License for the specific language governing
8    * permissions and limitations under the License, including but not limited to
9    * distribution rights of the Software.
10   *
11   *
12   *
13   */
14  
15  package com.liferay.portal.dao.shard;
16  
17  import com.liferay.portal.kernel.dao.shard.ShardUtil;
18  import com.liferay.portal.kernel.poller.PollerException;
19  import com.liferay.portal.kernel.poller.PollerProcessor;
20  import com.liferay.portal.kernel.poller.PollerRequest;
21  import com.liferay.portal.kernel.poller.PollerResponse;
22  
23  /**
24   * <a href="ShardPollerProcessorWrapper.java.html"><b><i>View Source</i></b></a>
25   *
26   * @author Alexander Chow
27   */
28  public class ShardPollerProcessorWrapper implements PollerProcessor {
29  
30      public ShardPollerProcessorWrapper(PollerProcessor pollerProcessor) {
31          _pollerProcessor = pollerProcessor;
32      }
33  
34      public void receive(
35              PollerRequest pollerRequest, PollerResponse pollerResponse)
36          throws PollerException {
37  
38          try {
39              ShardUtil.pushCompanyService(pollerRequest.getCompanyId());
40  
41              _pollerProcessor.receive(pollerRequest, pollerResponse);
42          }
43          finally {
44              ShardUtil.popCompanyService();
45          }
46      }
47  
48      public void send(PollerRequest pollerRequest) throws PollerException {
49          try {
50              ShardUtil.pushCompanyService(pollerRequest.getCompanyId());
51  
52              _pollerProcessor.send(pollerRequest);
53          }
54          finally {
55              ShardUtil.popCompanyService();
56          }
57      }
58  
59      private PollerProcessor _pollerProcessor;
60  
61  }