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);

        try {

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

            // 5 connect communities: 793 - done

//            communities.add("ee");

//            communities.add("mes");

//            communities.add("fam");

//            communities.add("ni");

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

            //2nd group - 1073 - done

            //duplicates ids [20587 -22214]

//            select id, source_id, claimed_in_dashboard from claim where id>20586 and id < 22215 and (source_id like 'elixir-gr%' or source_id like 'enermaps%' or source_id like 'dariah%' or source_id like 'science-innovation-policy%' or source_id like 'oa-pg%');

//            communities.add("elixir-gr");

//            communities.add("enermaps");

//            communities.add("dariah");

//            communities.add("science-innovation-policy");

//            communities.add("oa-pg");

            // 3rd group - 555 - done

//            communities.add("egi");

//            communities.add("rda");

            // 4th group - 32673 - done

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

            String startDate = "2018-03-01";

            String endDate = "2020-11-24";

            //local

            fetchClaimHandler.getSqlDAO().getSqlStore().setDbUrl("jdbc:postgresql://localhost:5432/claims_beta");

            //beta

//            fetchClaimHandler.getSqlDAO().getSqlStore().setDbUser("dnet");

//            fetchClaimHandler.getSqlDAO().getSqlStore().setDbPassword("dnetPwd");

//            fetchClaimHandler.getSqlDAO().getSqlStore().setDbUrl("jdbc:postgresql://beta.services.openaire.eu:5432/dnet_openaireplus");

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

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

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

//            fetchClaimHandler.getSqlDAO().getSqlStore().setDbUser("postgres");

//            fetchClaimHandler.getSqlDAO().getSqlStore().setDbPassword("snowflakes");

//            claimHandler.getSqlDAO().getSqlStore().setDbUrl("jdbc:postgresql://localhost:5432/movedb");

            //prod

            claimHandler.getSqlDAO().getSqlStore().setDbUser("dnet");

            claimHandler.getSqlDAO().getSqlStore().setDbPassword("dnetPwd");

            claimHandler.getSqlDAO().getSqlStore().setDbUrl("jdbc:postgresql://postgresql.services.openaire.eu:5432/dnet_openaireplus");

           /* out = new PrintWriter(new BufferedWriter(new FileWriter("/home/argirok/claims_migration/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.countAllClaims(startDate,endDate, null, null, true);

//            System.out.println("Prod claims:");

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

        } catch (Exception | SQLStoreException e) {

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

        }

    }

}