sis-geotk-plugin/src/test/java/org/gcube/data/transfer/plugins/sis/TestGetMetadata.java

130 lines
3.5 KiB
Java

package org.gcube.data.transfer.plugins.sis;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import org.apache.sis.storage.DataStoreException;
import org.apache.sis.storage.UnsupportedStorageException;
import org.gcube.data.transfer.plugins.thredds.sis.SisPlugin;
public class TestGetMetadata {
public static void main(String[] args) throws UnsupportedStorageException, DataStoreException, IOException, NoSuchAlgorithmException {
String[] toCheckFiles=new String[] {
"/Users/FabioISTI/Downloads/Aaptos_aaptos.nc",
"/Users/FabioISTI/Downloads/NASA_Precipitations_1950_2100_rcp45.nc",
};
for(String f:toCheckFiles) {
System.out.println("checking "+f);
try{
check(false,f);
}catch(UnsupportedStorageException e) {
System.err.println(e.getMessage());
// System.out.println("Trying opening "+f+" manually");
// open(f);
}
}
System.out.println("Done");
}
// private static void open(String path) throws IOException, UnsupportedStorageException, DataStoreException {
// System.out.println("Opening "+path);
// NetcdfFile f=NetcdfFiles.open(path);
// System.out.println(f.getFileTypeDescription()+"\t"+f.getFileTypeId());
//
//
// NetcdfDataset ncDs=NetcdfDatasets.openDataset(path);
//// NcMLReaderNew.
////
// System.out.println(ncDs.getFileTypeDescription()+"\t"+ncDs.getFileTypeId());
//
//// NetcdfDatasets.openFile(path,null);
//
//
// }
private static void check(boolean checkCopy,String src) throws UnsupportedStorageException, DataStoreException, IOException, NoSuchAlgorithmException {
File original=new File(src);
File dataset=original;
if(checkCopy) {
File temp=File.createTempFile("temp", ".temp");
File copied=File.createTempFile("copied", ".nc");
transferStream(new FileInputStream(original),new FileOutputStream(temp));
System.out.println("copied. Moving..");
System.out.println("Checksum original : "+getChecksum(original));
System.out.println("Checksum temp : "+getChecksum(temp));
Files.move(temp.toPath(), copied.toPath(),StandardCopyOption.ATOMIC_MOVE,StandardCopyOption.REPLACE_EXISTING);
if(copied.length()<original.length()) throw new RuntimeException("Different size after moving");
dataset=copied;
}
System.out.println(SisPlugin.getMetaFromFile(dataset));
}
private static String getChecksum(File datafile) throws NoSuchAlgorithmException, IOException {
MessageDigest md = MessageDigest.getInstance("SHA1");
FileInputStream fis = new FileInputStream(datafile);
byte[] dataBytes = new byte[1024];
int nread = 0;
while ((nread = fis.read(dataBytes)) != -1) {
md.update(dataBytes, 0, nread);
};
byte[] mdbytes = md.digest();
//convert the byte to hex format
StringBuffer sb = new StringBuffer("");
for (int i = 0; i < mdbytes.length; i++) {
sb.append(Integer.toString((mdbytes[i] & 0xff) + 0x100, 16).substring(1));
}
return sb.toString();
}
private static void transferStream(InputStream in, OutputStream out){
long receivedTotal=0l;
try{
byte[] internalBuf=new byte[1024];
int received=0;
while ((received=in.read(internalBuf))!=-1){
out.write(internalBuf,0,received);
receivedTotal+=received;
}
out.flush();
}catch(IOException e){
throw new RuntimeException("Unable to read from source.");
}
}
}