1
|
package eu.dnetlib.dli.resolver;
|
2
|
|
3
|
import java.net.URLEncoder;
|
4
|
|
5
|
import com.mongodb.DBObject;
|
6
|
import com.mongodb.MongoClient;
|
7
|
import com.mongodb.QueryBuilder;
|
8
|
import com.mongodb.client.FindIterable;
|
9
|
import com.mongodb.client.MongoCollection;
|
10
|
import com.mongodb.client.MongoCursor;
|
11
|
import com.mongodb.client.MongoDatabase;
|
12
|
import eu.dnetlib.dli.resolver.model.DLIResolvedObject;
|
13
|
import eu.dnetlib.pid.resolver.AbstractPIDResolver;
|
14
|
import groovy.transform.Synchronized;
|
15
|
import org.apache.commons.logging.Log;
|
16
|
import org.apache.commons.logging.LogFactory;
|
17
|
import org.bson.Document;
|
18
|
import org.bson.conversions.Bson;
|
19
|
import org.springframework.beans.factory.annotation.Autowired;
|
20
|
|
21
|
// TODO: Auto-generated Javadoc
|
22
|
|
23
|
/**
|
24
|
* The Class CrossrefDOIResolver.
|
25
|
*/
|
26
|
public class CrossrefResolver extends AbstractPIDResolver {
|
27
|
|
28
|
/**
|
29
|
* The Constant log.
|
30
|
*/
|
31
|
private static final Log log = LogFactory.getLog(CrossrefResolver.class);
|
32
|
|
33
|
/**
|
34
|
* The Constant baseUrlCrossref.
|
35
|
*/
|
36
|
private final static String baseUrlCrossref = "http://api.crossref.org/works/";
|
37
|
|
38
|
/**
|
39
|
* The parser.
|
40
|
*/
|
41
|
@Autowired
|
42
|
private CrossRefParserJSON parser;
|
43
|
|
44
|
|
45
|
@Autowired
|
46
|
private MongoClient mongoClient;
|
47
|
|
48
|
private MongoDatabase db;
|
49
|
|
50
|
@Override
|
51
|
protected boolean canResolvePid(final String pidType) {
|
52
|
return (pidType != null) && ("doi".equals(pidType.toLowerCase().trim()) || "handle".equals(pidType.toLowerCase().trim()));
|
53
|
}
|
54
|
|
55
|
@Override
|
56
|
@Synchronized
|
57
|
protected DLIResolvedObject resolve(final String pid, final String pidType) {
|
58
|
String response;
|
59
|
try {
|
60
|
// response = requestURL(baseUrlCrossref + URLEncoder.encode(pid, "UTF-8"));
|
61
|
response = retrieveCrossRefFromDump(pid);
|
62
|
} catch (Exception e) {
|
63
|
log.debug("unable to get response");
|
64
|
return null;
|
65
|
}
|
66
|
try {
|
67
|
|
68
|
log.debug("Obtained " + response);
|
69
|
if (response == null) return null;
|
70
|
if (response.contains("Resource not found")) return null;
|
71
|
DLIResolvedObject record = parser.parseRecord(response);
|
72
|
if (record.getPid() == null) return null;
|
73
|
return record;
|
74
|
} catch (Throwable e) {
|
75
|
log.error("Error on resolve pid " + pid, e);
|
76
|
}
|
77
|
return null;
|
78
|
}
|
79
|
|
80
|
|
81
|
private String retrieveCrossRefFromDump(final String pid) {
|
82
|
if (db == null) {
|
83
|
db = mongoClient.getDatabase("crossRef");
|
84
|
|
85
|
}
|
86
|
final MongoCollection<Document> crossRef = db.getCollection("dump");
|
87
|
|
88
|
DBObject query = QueryBuilder.start("_id").is(pid).get();
|
89
|
FindIterable<Document> documents = crossRef.find((Bson) query).limit(1);
|
90
|
MongoCursor<Document> iterator = documents.iterator();
|
91
|
if (iterator.hasNext()){
|
92
|
return iterator.next().toJson();
|
93
|
}
|
94
|
return null;
|
95
|
|
96
|
|
97
|
|
98
|
|
99
|
}
|
100
|
|
101
|
|
102
|
|
103
|
|
104
|
|
105
|
}
|