63 lines
2.0 KiB
Java
63 lines
2.0 KiB
Java
package eu.mikroskeem.mongohackery;
|
|
|
|
import com.mongodb.MongoClient;
|
|
import com.mongodb.client.MongoDatabase;
|
|
import com.mongodb.client.gridfs.GridFSBucket;
|
|
import com.mongodb.client.gridfs.GridFSBuckets;
|
|
import com.mongodb.client.gridfs.model.GridFSUploadOptions;
|
|
import org.bson.Document;
|
|
import org.bson.types.ObjectId;
|
|
|
|
import java.io.BufferedReader;
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.InputStream;
|
|
import java.io.InputStreamReader;
|
|
import java.util.stream.Collectors;
|
|
|
|
import static com.mongodb.client.model.Filters.eq;
|
|
|
|
public class Main {
|
|
public static void main(String... args){
|
|
new Main();
|
|
}
|
|
|
|
public Main(){
|
|
MongoClient client = new MongoClient();
|
|
MongoDatabase leTest = client.getDatabase("leTest");
|
|
GridFSBucket gridfs = GridFSBuckets.create(leTest, "files");
|
|
|
|
/*
|
|
* Ransom data
|
|
* Should use file here actually
|
|
*/
|
|
InputStream is = new ByteArrayInputStream("topkek".getBytes());
|
|
|
|
/* Remove (if present) */
|
|
try {
|
|
gridfs.delete(gridfs.find(eq("filename", "topkek")).first().getId());
|
|
}
|
|
catch(NullPointerException ignored){}
|
|
|
|
/* Insert */
|
|
GridFSUploadOptions options = new GridFSUploadOptions()
|
|
.chunkSizeBytes(1024)
|
|
.metadata(new Document("contentType", "text/plain"));
|
|
/*
|
|
* NOTE: 'contentType' isn't standard! It is here, because I want to see
|
|
* It is there, because I tend to use it with GridFS
|
|
*/
|
|
|
|
ObjectId id = gridfs.uploadFromStream("topkek", is, options);
|
|
System.out.println(id);
|
|
|
|
/* Now read */
|
|
InputStream os = gridfs.openDownloadStream(gridfs.find(eq("filename", "topkek")).first().getId());
|
|
String res = new BufferedReader(new InputStreamReader(os))
|
|
.lines().collect(Collectors.joining("\n"));
|
|
System.out.println(res);
|
|
|
|
/* Remove */
|
|
gridfs.delete(gridfs.find(eq("filename", "topkek")).first().getId());
|
|
}
|
|
}
|