public static void insertPost(Post post, String type) {
try {
String postId = post.getId();
IdNameEntity idNameEntity = post.getFrom();
String userId = idNameEntity.getId();
String userName = idNameEntity.getName();
String commentsCount = post.getComments() == null ? "0" : post.getComments().size() + "";
String likeCount = post.getLikes() == null ? "0" : post.getLikes().size() + "";
String columnFamily = COLUMN_FAMILY_FACEBOOK_POST;
String rowKey = type;
String superColumn = postId + "-" + userId + "-" + userName;
Clock clock = new Clock(System.nanoTime());
Column column = new Column();
String columnName = commentsCount + "-" + likeCount;
column.setName(columnName.getBytes(UTF8));
String columnValue = post.getMessage();
if (!StringUtils.isBlank(columnValue)) {
column.setValue(columnValue.getBytes(UTF8));
} else {
column.setValue("".getBytes(UTF8));
}
column.setTimestamp(clock.timestamp);
cassandra.insertSuperColumn(columnFamily, rowKey, superColumn, column);
} catch (UnsupportedEncodingException e) {
logger.error("[Info: encoding invalid] - [Error: {}]", e.toString());
}
}
public static void removePost(Post post, String type) {
String postId = post.getId();
IdNameEntity idNameEntity = post.getFrom();
String userId = idNameEntity.getId();
String userName = idNameEntity.getName();
String columnFamily = COLUMN_FAMILY_FACEBOOK_POST;
String rowKey = type;
String superColumn = postId + "-" + userId + "-" + userName;
cassandra.removeSuperColumn(columnFamily, rowKey, superColumn);
}
public static void insertPostPopular(Post post, String type, Integer position) {
try {
String postId = post.getId();
IdNameEntity idNameEntity = post.getFrom();
String userId = idNameEntity.getId();
String userName = idNameEntity.getName();
String columnFamily = COLUMN_FAMILY_FACEBOOK_POST_POPULAR;
String rowKey = (type + "-" + position);
Clock clock = new Clock(System.nanoTime());
Column column = new Column();
String columnName = (postId + "-" + userId + "-" + userName);
column.setName(columnName.getBytes(UTF8));
String columnValue = post.getMessage();
column.setValue(columnValue.getBytes(UTF8));
column.setTimestamp(clock.timestamp);
cassandra.insertColumn(columnFamily, rowKey, column);
} catch (UnsupportedEncodingException e) {
logger.error("[Info: encoding invalid] - [Error: {}]", e.toString());
}
}
public static XContentBuilder toJson(
Post post,
String riverName,
Map<String, List<EntityReference>> entities,
Map<String, List<String>> mauiTopics,
List<Map<String, String>> bayesianAnalyzedText)
throws IOException {
XContentBuilder out = XContentFactory.jsonBuilder().startObject();
out.field("PostId", post.getId());
out.field("PostLink", post.getLink() != null ? post.getLink().toString() : null);
out.field("PostCreatedTime", post.getCreatedTime());
IdNameEntity idName = null;
if ((idName = post.getFrom()) != null) {
out.field("PostFromId", idName.getId());
out.field("PostFromName", idName.getName());
}
if (post.getStatusType() != null) {
out.field("PostStatusType", post.getStatusType());
}
out.field("Post", post.getMessage());
if (entities != null) {
if (entities.isEmpty()) {
out.field("entities", "{}");
} else {
out.startArray("entities");
for (Map.Entry<String, List<EntityReference>> entityEntry : entities.entrySet()) {
out.startObject();
out.startArray(entityEntry.getKey());
for (EntityReference er : entityEntry.getValue()) {
out.startObject();
out.field("entity-label", er.getEntityLabel());
out.field("entity-reference", er.getEntityReference());
out.field("entity-confidence", er.getConfidence());
out.startArray("entity-type");
for (String s : er.getEntityTypeList()) {
out.startObject();
out.field("type", s);
out.endObject();
}
out.endArray();
out.endObject();
}
out.endArray();
out.endObject();
}
out.endArray();
}
}
if (mauiTopics != null) {
if (mauiTopics.isEmpty()) {
out.field("topics", "{}");
} else {
out.startArray("topics");
out.startObject();
for (Map.Entry<String, List<String>> e : mauiTopics.entrySet()) {
out.startArray(e.getKey());
for (String s : e.getValue()) {
out.startObject();
out.field("text", s);
out.endObject();
}
out.endArray();
}
out.endObject();
out.endArray();
}
}
if (bayesianAnalyzedText != null) {
if (bayesianAnalyzedText.isEmpty()) {
out.field("BayesianAnalysers", "{}");
} else {
out.startArray("BayesianAnalysers");
out.startObject();
for (Map<String, String> m : bayesianAnalyzedText) {
if (m == null || m.isEmpty()) {
continue;
}
for (Map.Entry<String, String> e : m.entrySet()) {
out.startArray(e.getKey());
out.startObject();
out.field("text", e.getValue());
out.endObject();
out.endArray();
}
}
out.endObject();
out.endArray();
}
}
if (riverName != null) {
out.field("RiverName", riverName);
}
return out.endObject();
}
