D-Net

package eu.dnetlib.goldoa.service.utils;

import eu.dnetlib.goldoa.domain.ManagerException;

import eu.dnetlib.goldoa.domain.Publication;

import eu.dnetlib.goldoa.domain.Request;

import eu.dnetlib.goldoa.domain.stats.Tuple;

import eu.dnetlib.goldoa.service.PublicationManager;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.jdbc.core.JdbcTemplate;

import org.springframework.jdbc.core.RowMapper;

import javax.sql.DataSource;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Types;

import java.util.List;

import java.util.logging.StreamHandler;

/**

 * Created by antleb on 3/7/16.

 */

public class LicenseUpdateTask implements Runnable {

	private static final String GET_MISSING_LICENSES = "select pub.id, pub.doi from publication pub join request r on r.publication=pub.id where doi is not null and license is null and r.status=" + Request.RequestStatus.ACCOUNTING_PAID.getCode();

	private static final String UPDATE_LICENSE = "update publication set license=? where id=?";

	private DataSource dataSource;

	private PublicationManager publicationManager;

	public LicenseUpdateTask(DataSource dataSource, PublicationManager publicationManager) {

		this.dataSource = dataSource;

		this.publicationManager = publicationManager;

	}

	@Override

	public void run() {

		System.out.println("Updating paid request licenses!");

		JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

		List<Tuple<String, String>> pubs = jdbcTemplate.query(GET_MISSING_LICENSES, new RowMapper<Tuple<String, String>>() {

			@Override

			public Tuple<String, String> mapRow(ResultSet rs, int rowNum) throws SQLException {

				return new Tuple<String, String>(rs.getString("id"), rs.getString("doi"));

			}

		});

		for (Tuple<String, String> pub:pubs) {

			try {

				Publication publication = publicationManager.resolveDOI(pub.getSecond());

				if (publication.getLicense() != null) {

					jdbcTemplate.update(UPDATE_LICENSE, new String[] {publication.getLicense(), pub.getFirst()}, new int[] {Types.VARCHAR, Types.VARCHAR});

				}

			} catch (ManagerException e) {

				System.out.print("Error updating license " + e.getErrorCause());

			} catch (Exception e) {

				System.out.print("Error updating license " + e.getMessage());

			}

		}

	}

}

package eu.dnetlib.goldoa.service.utils;

import eu.dnetlib.goldoa.service.PublicationManager;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

import javax.sql.DataSource;

import java.util.concurrent.Executors;

import java.util.concurrent.ScheduledExecutorService;

import java.util.concurrent.TimeUnit;

/**

 * Created by antleb on 3/7/16.

 */

public class Scheduler {

	@Autowired

	private DataSource dataSource;

	@Autowired

	private PublicationManager publicationManager;

	private ScheduledExecutorService executor = Executors.newScheduledThreadPool(1);

	@PostConstruct

	public void init() {

		executor.scheduleAtFixedRate(new LicenseUpdateTask(dataSource, publicationManager), 0, 3,TimeUnit.HOURS);

	}

	public void destroy() {

		executor.shutdown();

	}

}