de.cosmocode.palava.media.expire.AssetExpirationService.java Source code

Introduction

Here is the source code for de.cosmocode.palava.media.expire.AssetExpirationService.java

Source

package de.cosmocode.palava.media.expire;
/**
 * Copyright 2010 CosmoCode GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import java.util.List;

import javax.persistence.EntityManager;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Preconditions;
import com.google.common.collect.Iterables;
import com.google.inject.Inject;
import com.google.inject.Provider;

import de.cosmocode.palava.core.Registry;
import de.cosmocode.palava.core.lifecycle.Executable;
import de.cosmocode.palava.core.lifecycle.Initializable;
import de.cosmocode.palava.core.lifecycle.LifecycleException;
import de.cosmocode.palava.entity.EntityService;
import de.cosmocode.palava.jpa.Transactional;
import de.cosmocode.palava.media.asset.AssetBase;
import de.cosmocode.palava.media.asset.AssetExpiredEvent;
import de.cosmocode.palava.media.asset.AssetUnexpiredEvent;
import de.cosmocode.palava.scope.UnitOfWork;

/**
 * A service which checks periodically for expired assets.
 *
 * @since 2.0
 * @author Willi Schoenborn
 */
final class AssetExpirationService implements Initializable, Executable, Runnable {

    private static final Logger LOG = LoggerFactory.getLogger(AssetExpirationService.class);

    private final EntityService<AssetBase> service;

    private final AssetExpiredEvent expiredEvent;

    private final AssetUnexpiredEvent unexpiredEvent;

    private final Provider<EntityManager> provider;

    @Inject
    public AssetExpirationService(EntityService<AssetBase> service, Registry registry,
            Provider<EntityManager> provider) {
        this.service = Preconditions.checkNotNull(service, "Service");
        Preconditions.checkNotNull(registry, "Registry");
        this.expiredEvent = registry.proxy(AssetExpiredEvent.class);
        this.unexpiredEvent = registry.proxy(AssetUnexpiredEvent.class);
        this.provider = Preconditions.checkNotNull(provider, "Provider");
    }

    @UnitOfWork
    @Override
    public void initialize() throws LifecycleException {
        try {
            checkQuery(AssetBase.EXPIRING);
            checkQuery(AssetBase.UNEXPIRING);
        } catch (IllegalArgumentException e) {
            throw new LifecycleException(e);
        }
    }

    private void checkQuery(String name) {
        LOG.debug("Checking for presence of named query {}", name);
        provider.get().createNamedQuery(name);
    }

    @Override
    public void execute() throws LifecycleException {
        run();
    }

    @UnitOfWork
    @Transactional
    @Override
    public void run() {
        checkExpiring();
        checkUnexpiring();
    }

    private void checkExpiring() {
        final List<AssetBase> assets = service.list(AssetBase.EXPIRING);
        assert Iterables.all(assets, AssetBase.IS_EXPIRING) : "Expected all assets to be expiring";

        LOG.info("Found {} expiring assets", assets.size());

        for (AssetBase asset : assets) {
            expiredEvent.eventAssetExpired(asset);
        }
    }

    private void checkUnexpiring() {
        final List<AssetBase> assets = service.list(AssetBase.UNEXPIRING);
        assert Iterables.all(assets, AssetBase.IS_UNEXPIRING) : "Expected all assets to be unexpiring";

        LOG.info("Found {} unexpiring assets", assets.size());

        for (AssetBase asset : assets) {
            unexpiredEvent.eventAssetUnexpired(asset);
        }
    }

}