Exemplo n.º 1
0
  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());
    }
  }
Exemplo n.º 2
0
  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);
  }
Exemplo n.º 3
0
  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());
    }
  }
Exemplo n.º 4
0
  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();
  }