improve GitBucketDataStore to crawl issue comments
This commit is contained in:
parent
a4c73a664e
commit
c7841013ab
1 changed files with 26 additions and 2 deletions
|
@ -41,6 +41,9 @@ import org.elasticsearch.common.xcontent.json.JsonXContent;
|
|||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
import com.fasterxml.jackson.core.type.TypeReference;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
|
||||
/**
|
||||
* @author Keiichi Watanabe
|
||||
*/
|
||||
|
@ -310,8 +313,8 @@ public class GitBucketDataStoreImpl extends AbstractDataStoreImpl {
|
|||
logger.warn("Failed to access to " + issueUrl, e);
|
||||
}
|
||||
|
||||
// FIXME: Get issue comments from `commentsUrl`
|
||||
// How to parse JSON-style list?
|
||||
final String commentsStr = String.join("\n", getIssueComments(issueUrl, authToken));
|
||||
contentStr += "\n" + commentsStr;
|
||||
|
||||
dataMap.put("content", contentStr);
|
||||
dataMap.put("url", viewUrl);
|
||||
|
@ -325,6 +328,27 @@ public class GitBucketDataStoreImpl extends AbstractDataStoreImpl {
|
|||
return;
|
||||
}
|
||||
|
||||
private List<String> getIssueComments(final String issueUrl, final String authToken) {
|
||||
final String commentsUrl = issueUrl + "/comments";
|
||||
final List<String> commentList = new ArrayList<String>();
|
||||
|
||||
try (CurlResponse curlResponse = Curl.get(commentsUrl).header("Authorization", "token " + authToken).execute()) {
|
||||
final String commentsJson = curlResponse.getContentAsString();
|
||||
List<Map<String, Object>> comments = new ObjectMapper().readValue(commentsJson, new TypeReference<List<Map<String, Object>>>() {
|
||||
});
|
||||
|
||||
for (Map<String, Object> comment : comments) {
|
||||
if (comment.containsKey("body")) {
|
||||
commentList.add((String) comment.get("body"));
|
||||
}
|
||||
}
|
||||
} catch (final Exception e) {
|
||||
logger.warn("Failed to access to " + issueUrl, e);
|
||||
}
|
||||
|
||||
return commentList;
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
private void storeWikiContents(final String rootURL, final String authToken, final String wikiLabel, final String owner,
|
||||
final String name, final List<String> roleList, final CrawlingConfig crawlingConfig, final IndexUpdateCallback callback,
|
||||
|
|
Loading…
Add table
Reference in a new issue