1
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
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 }