001
014
015 package com.liferay.portal.dao.shard;
016
017 import com.liferay.portal.kernel.dao.shard.ShardUtil;
018 import com.liferay.portal.kernel.poller.PollerException;
019 import com.liferay.portal.kernel.poller.PollerProcessor;
020 import com.liferay.portal.kernel.poller.PollerRequest;
021 import com.liferay.portal.kernel.poller.PollerResponse;
022
023
026 public class ShardPollerProcessorWrapper implements PollerProcessor {
027
028 public ShardPollerProcessorWrapper(PollerProcessor pollerProcessor) {
029 _pollerProcessor = pollerProcessor;
030 }
031
032 @Override
033 public void receive(
034 PollerRequest pollerRequest, PollerResponse pollerResponse)
035 throws PollerException {
036
037 try {
038 ShardUtil.pushCompanyService(pollerRequest.getCompanyId());
039
040 _pollerProcessor.receive(pollerRequest, pollerResponse);
041 }
042 finally {
043 ShardUtil.popCompanyService();
044 }
045 }
046
047 @Override
048 public void send(PollerRequest pollerRequest) throws PollerException {
049 try {
050 ShardUtil.pushCompanyService(pollerRequest.getCompanyId());
051
052 _pollerProcessor.send(pollerRequest);
053 }
054 finally {
055 ShardUtil.popCompanyService();
056 }
057 }
058
059 private PollerProcessor _pollerProcessor;
060
061 }