gcube-cms-suite/use-cases/src/test/java/org/gcube/application/cms/usecases/mocks/ConcessionPublisherThread.java

package org.gcube.application.cms.usecases.mocks;

import lombok.*;
import lombok.extern.slf4j.Slf4j;
import org.gcube.application.cms.tests.model.TestModel;
import org.gcube.application.geoportal.client.utils.Serialization;
import org.gcube.application.geoportal.common.model.legacy.Concessione;
import org.gcube.application.geoportal.common.model.legacy.report.ValidationReport;
import org.gcube.application.geoportal.common.rest.MongoConcessioni;
import org.gcube.application.geoportal.common.utils.FileSets;
import org.gcube.application.geoportal.common.utils.Files;
import org.gcube.application.geoportal.common.utils.StorageUtils;

import java.io.File;
import java.util.List;
import java.util.Map;
import java.util.NavigableSet;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicLong;

import static org.gcube.application.geoportal.client.GeoportalAbstractPlugin.mongoConcessioni;

@Slf4j
@RequiredArgsConstructor
@ToString
public class ConcessionPublisherThread implements Runnable{



    //Threads
    private static AtomicLong queued=new AtomicLong(0);
    private static AtomicLong completed = new AtomicLong(0);

    //Concessioni
    @Data
    static class Report{
        private ConcurrentLinkedQueue<Concessione> error=new ConcurrentLinkedQueue<>();
        private ConcurrentLinkedQueue<Concessione> success=new ConcurrentLinkedQueue<>();
        private ConcurrentLinkedQueue<Concessione> warning=new ConcurrentLinkedQueue<>();
        private ConcurrentLinkedQueue<Concessione> noReport=new ConcurrentLinkedQueue<>();
    }

    private static ConcessionPublisherThread.Report report=new ConcessionPublisherThread.Report();

    private static ExecutorService exec=Executors.newFixedThreadPool(5);

    public static void submit(ConcessionPublisherThread c){
        log.info("Submitting {}",c);
        exec.submit(c);
        log.info("Submitted {} ",queued.incrementAndGet());
    }

    public static ConcessionPublisherThread.Report waitCompletion(){
        try {
            while (!exec.awaitTermination(2, TimeUnit.MINUTES)) {
                log.info("Waiting .. completed {}, out of {} ",completed.get(),queued.get());
                if(completed.get()==queued.get()) exec.shutdown();
            }
        }catch(InterruptedException e){
            log.info("Process finished");
        }
        return report;
    }

    @NonNull
    private File baseDir;
    @NonNull
    private String positionPath;
    @NonNull
    private String projectName;





    @Override
    public void run() {
        Concessione c=null;
        try {
            MongoConcessioni client=mongoConcessioni().build();
            StorageUtils storage=new StorageUtils();

            //NB raggruppa per file
            Map.Entry<String, List<File>> posSets = Files.getAllShapeSet(new File(baseDir,positionPath),true).
                    entrySet().stream().findFirst().get();
            Map<String, List<File>> pianteSets = Files.getAllShapeSet(new File(baseDir,projectName),true);

            if(pianteSets.size()>1)
                pianteSets.remove(posSets.getKey());



            log.debug("Entry {} pos Size {} piante {} ",projectName,posSets.getValue().size(),pianteSets.size());


             c = createMock(projectName,baseDir.getName(), pianteSets, posSets.getValue(), client, storage);

            switch(c.getReport().getStatus()){
                case PASSED: report.getSuccess().add(c);
                break;
                case ERROR: report.getError().add(c);
                break;
                case WARNING:report.getWarning().add(c);
                break;
            }

        }catch(Throwable t){
            log.error("Problematic entry "+this,t);
            report.getNoReport().add(c);
        }finally{
            log.info("Completed N {}", completed.incrementAndGet());
        }
    }



    private static Concessione createMock(String baseName, String packageName, Map<String,List<File>> piante, List<File> pos,
                                          MongoConcessioni client, StorageUtils storage) throws Exception {

        Concessione c= TestModel.prepareConcessione(piante.size(), 1);
        c.setNome("Mock for "+baseName+" ("+packageName+")");
        c= client.createNew(c);
        String mongoId=c.getMongo_id();

        // TEST DATA, DO NOT CARE
        client.registerFileSet(mongoId, FileSets.prepareRequest(storage,
                Concessione.Paths.RELAZIONE,new File (TestModel.getBaseFolder(),"relazione.pdf")));

        client.registerFileSet(mongoId, FileSets.prepareRequest(storage,
                Concessione.Paths.imgByIndex(0),Files.getSiblings(TestModel.getBaseFolder(),"immagine").get(0)));

        // POSIZIONAMENTO

        client.registerFileSet(mongoId, FileSets.prepareRequest(storage,
                Concessione.Paths.POSIZIONAMENTO,pos.toArray(new File[pos.size()])));

        // PIANTE
        Map.Entry<String,List<File>>[] entries= piante.entrySet().toArray(new Map.Entry[0]);
        for( int i= 0; i< piante.size();i++) {
            // Set layer name
            c=client.getById(mongoId);
            String path=Concessione.Paths.piantaByIndex(i);
            c.getContentByPath(path).setTitolo(" Pianta from "+entries[i].getKey());
            client.update(mongoId, Serialization.write(c));

            //Set fileset
            client.registerFileSet(mongoId, FileSets.prepareRequest(storage,path, entries[i].getValue().toArray(new File[0])));
        }

        c=client.publish(mongoId);

        System.out.println("@@@ Concessione "+c.getNome()+"\t STATUS : "+ c.getReport().getStatus());

        return c;
    }
}