D-Net

package eu.dnetlib.data.claims;

import eu.dnetlib.data.claims.entity.Claim;

import eu.dnetlib.data.claims.entity.Result;

import eu.dnetlib.data.claims.handler.*;

import eu.dnetlib.data.claims.sql.SQLStoreException;

import eu.dnetlib.data.claims.utils.*;

import org.apache.log4j.BasicConfigurator;

import org.apache.log4j.Level;

import org.apache.log4j.Logger;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.context.ApplicationContext;

import org.springframework.context.annotation.PropertySource;

import org.springframework.context.support.ClassPathXmlApplicationContext;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import java.io.BufferedWriter;

import java.io.FileWriter;

import java.io.PrintWriter;

import java.util.ArrayList;

import java.util.List;

import static junit.framework.Assert.assertNotNull;

@RunWith(SpringJUnit4ClassRunner.class)

@ContextConfiguration(locations = {"springContext-claims.xml"})

@PropertySource("springContext-claims.properties")

public class CopyFromBetaTest {

	private Logger log = Logger.getLogger(this.getClass());

	ClaimHandler claimHandler = null;

    FetchClaimHandler fetchClaimHandler = null;

    private PrintWriter out;

    @Before

	public void init() throws Exception {

		BasicConfigurator.configure();

        ApplicationContext context = new ClassPathXmlApplicationContext("eu/dnetlib/data/claims/springContext-claims.xml");

        claimHandler = context.getBean(ClaimHandler.class);

        fetchClaimHandler = context.getBean(FetchClaimHandler.class);

	}

    @Test

    public void CopyFromBetaToProduction(){

        log.setLevel(Level.ERROR);

        QueryGenerator q = new QueryGenerator();

        System.out.println(q.generateCreateClaimTablesQuery());

        try {

            fetchClaimHandler.getQueryGenerator().setMigrationTable("claims_beta");

            ArrayList<String> communities = new ArrayList<>();

            communities.add("ee");

//            communities.add("mes");

//            communities.add("fam");

//            communities.add("ni");

//            communities.add("dh-ch");

//            communities.add("covid-19");

            String startDate = "2018-03-01";

            String endDate = "2020-11-22";

            fetchClaimHandler.getSqlDAO().getSqlStore().setDbUrl("jdbc:postgresql://scoobydoo.di.uoa.gr:5432/claims_beta");

            List<Claim> claims = fetchClaimHandler.fetchClaimsByDateForDashboards(startDate,endDate,  null,null,true, communities);

            System.out.println("Beta claims: ");

            System.out.println(claims.size());

            claimHandler.getSqlDAO().getSqlStore().setDbUrl("jdbc:postgresql://scoobydoo.di.uoa.gr:5432/movedb");

            out = new PrintWriter(new BufferedWriter(new FileWriter("/home/argirok/claims_migratedFromBeta_record_paths.txt", true)));

            for (Claim claim : claims) {

                if(out!=null) {

                    if(!claim.getTargetType().equals("project") && !claim.getTargetType().equals("context")){

                        System.out.println(((Result)claim.getTarget()).getRecordPath());

                        out.println("cp beta_records/" + ((Result)claim.getTarget()).getRecordPath() + " records/"+((Result)claim.getTarget()).getRecordPath());

                    }

                    if(!claim.getSourceType().equals("project") && !claim.getSourceType().equals("context")){

                        System.out.println(((Result)claim.getSource()).getRecordPath());

                        out.println("cp beta_records/" + ((Result)claim.getSource()).getRecordPath() + " records/"+((Result)claim.getSource()).getRecordPath());

                    }

                }

                System.out.println(claim.getId());

//                claimHandler.saveClaim(claim);

            }

            if(out!=null) {

                out.close();

            }

            List<Claim> claimsmoveDB = fetchClaimHandler.fetchClaimsByDate(startDate,endDate, null, null, true);

            System.out.println("Movedb claims:");

            System.out.println(claimsmoveDB.size());

        } catch (Exception | SQLStoreException e) {

            log.error("Error in migration",e);

        }

    }

}