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 PollerResponse receive(PollerRequest pollerRequest)
034 throws PollerException {
035
036 try {
037 ShardUtil.pushCompanyService(pollerRequest.getCompanyId());
038
039 return _pollerProcessor.receive(pollerRequest);
040 }
041 finally {
042 ShardUtil.popCompanyService();
043 }
044 }
045
046 @Override
047 public void send(PollerRequest pollerRequest) throws PollerException {
048 try {
049 ShardUtil.pushCompanyService(pollerRequest.getCompanyId());
050
051 _pollerProcessor.send(pollerRequest);
052 }
053 finally {
054 ShardUtil.popCompanyService();
055 }
056 }
057
058 private final PollerProcessor _pollerProcessor;
059
060 }