Skip to content

B
brandkbs2
Commit 4cd3f9da by 陈健智
Options

Merge remote-tracking branch 'origin/feature' into feature

parents 82a27aa9 b57cf324
Showing with 309 additions and 60 deletions
src/main/java/com/zhiwei/brandkbs2/common/ChannelType.java 0 → 100644
package com.zhiwei.brandkbs2.common;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import java.util.Arrays;
import java.util.List;
/**
* @ClassName: ChannelType
* @Description 渠道类型
* @author: sjj
* @date: 2023-04-14 17:27
*/
public class ChannelType {
private static final List<String> MAIN_BODY_TYPES = Arrays.asList("媒体", "团体", "普通用户", "企业", "名人", "政务机构");
public static final List<String> POLITICS_LEVELS = Arrays.asList("央级", "省级", "地级");
private static final List<String> FIELDS = Arrays.asList("情感", "综合", "财经", "家居", "社会", "时尚", "文化", "历史", "母婴育儿", "体育", "娱乐", "动漫", "汽车", "美食", "健康养生", "国际", "音乐", "游戏", "科技", "旅游", "军事", "时事", "教育");
private static final JSONArray REGION = GlobalPojo.CHINA_AREA_TAGS;
public static JSONObject channelOption = new JSONObject();
public static final List<String> COMMON_FIELDS = Arrays.asList("财经", "科技", "时政", "社会民生", "交通");
static {
channelOption.put("mainBodyTypes", MAIN_BODY_TYPES);
channelOption.put("politicsLevels", POLITICS_LEVELS);
channelOption.put("fields", FIELDS);
channelOption.put("regions", REGION);
}
}
src/main/java/com/zhiwei/brandkbs2/common/GenericAttribute.java
......@@ -111,6 +111,25 @@ public class GenericAttribute {
public static final String ES_LINKED_GROUP_ID = "linked_group_id";
public static final String ES_SENSITIVE_CHANNEL = "sensitive_channel";
/**
* 行政级别
*/
public static final String ES_POLITICS_LEVEL = "politics_level";
/**
* 主体类型
*/
public static final String ES_MAIN_BODY_TYPE = "main_body_type";
/**
* 领域
*/
public static final String ES_REGION = "region";
/**
* 地域
*/
public static final String ES_FIELD = "field";
// public static final String ES_CACHE_MAP_PROJECT = "brandkbs_cache_maps.project_id.keyword";
// public static final String ES_CACHE_MAP_LINKED_GROUP_ID = "brandkbs_cache_maps.linked_group_id.keyword";
// public static final String PLATFORM = "platform";
......
src/main/java/com/zhiwei/brandkbs2/common/GlobalPojo.java
package com.zhiwei.brandkbs2.common;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.zhiwei.brandkbs2.pojo.ChannelTag;
import com.zhiwei.brandkbs2.pojo.Project;
import com.zhiwei.brandkbs2.service.SystemInfoService;
import com.zhiwei.brandkbs2.util.Tools;
import com.zhiwei.qbjc.bean.pojo.common.MessagePlatform;
import com.zhiwei.qbjc.bean.pojo.common.Tag;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
......@@ -29,6 +33,10 @@ import java.util.stream.Collectors;
public class GlobalPojo {
public static final Logger log = LogManager.getLogger(GlobalPojo.class);
@Value("classpath:chinaAreaTag.json")
private org.springframework.core.io.Resource chinaAreaTag;
@Resource(name = "systemInfoServiceImpl")
private SystemInfoService systemInfoService;
......@@ -62,6 +70,8 @@ public class GlobalPojo {
*/
public static List<JSONObject> YU_QING_PROJECTS;
public static JSONArray CHINA_AREA_TAGS;
public static final List<String> PERMANENT_PLATFORM_NAMES = Arrays.asList("网媒", "微博", "微信", "今日头条");
public static final String ELSE_PLATFORM_NAME = "其他自媒体";
......@@ -69,6 +79,7 @@ public class GlobalPojo {
@PostConstruct
public void start() {
try {
initPojo();
updatePojo("启动获取");
} catch (Exception e) {
log.info("启动获取-出错", e);
......@@ -85,6 +96,10 @@ public class GlobalPojo {
}
}
private void initPojo() throws IOException {
CHINA_AREA_TAGS = JSONArray.parseArray(Tools.readJsonFile(chinaAreaTag.getInputStream()));
}
private void updatePojo(String logMsg) {
try {
PLATFORMS = systemInfoService.getPlatforms();
......
src/main/java/com/zhiwei/brandkbs2/dao/impl/QbjcPojoDaoImpl.java
......@@ -5,7 +5,6 @@ import com.zhiwei.qbjc.bean.pojo.common.MessagePlatform;
import com.zhiwei.qbjc.bean.pojo.common.Tag;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Component;
......@@ -27,7 +26,7 @@ public class QbjcPojoDaoImpl implements QbjcPojoDao {
@Override
public List<MessagePlatform> findMessagePlatformAll() {
return mongoTemplate.find(new Query().with(Sort.by(Sort.Order.asc("_id"))), MessagePlatform.class);
return mongoTemplate.find(new Query().with(Sort.by(Sort.Order.asc("createAt"))), MessagePlatform.class);
}
@Override
......
src/main/java/com/zhiwei/brandkbs2/enmus/EventTagEnum.java
......@@ -8,6 +8,7 @@ import com.alibaba.fastjson.JSONObject;
* @author: sjj
* @date: 2022-05-26 15:51
*/
@Deprecated
public enum EventTagEnum{
EVENT_ATTRIBUTE("事件属性"),
......
src/main/java/com/zhiwei/brandkbs2/enmus/ImportantChannelEnum.java
......@@ -11,6 +11,7 @@ import java.util.stream.Collectors;
* @description 重要渠道枚举类
* @date 2019/9/6 14:08
*/
@Deprecated
public enum ImportantChannelEnum {
/**
* 央级
......
src/main/java/com/zhiwei/brandkbs2/es/EsQueryTools.java
package com.zhiwei.brandkbs2.es;
import com.zhiwei.brandkbs2.common.ChannelType;
import com.zhiwei.brandkbs2.common.GenericAttribute;
import com.zhiwei.brandkbs2.common.GlobalPojo;
import com.zhiwei.brandkbs2.config.Constant;
import com.zhiwei.brandkbs2.enmus.ImportantChannelEnum;
import com.zhiwei.brandkbs2.pojo.external.BrandkbsYuQingConfig;
import com.zhiwei.pushlog.tools.Tools;
import com.zhiwei.qbjc.bean.pojo.common.MessagePlatform;
import org.apache.commons.collections4.CollectionUtils;
......@@ -95,14 +97,15 @@ public class EsQueryTools {
}
public static BoolQueryBuilder assembleCacheMapsQuery(String projectId, String contendId) {
return assembleCacheMapsQuery(projectId, contendId, null);
return assembleCacheMapsQuery(projectId, contendId, null, null, null, null, null);
}
public static BoolQueryBuilder assembleCacheMapsQuery(String projectId, String contendId, List<String> mediaTypes) {
return assembleCacheMapsQuery(projectId, contendId, mediaTypes,null);
public static BoolQueryBuilder assembleCacheMapsQuery(String projectId, String contendId, String politicsLevel, String field, String region, String mainBodyType) {
return assembleCacheMapsQuery(projectId, contendId, politicsLevel, field, region, mainBodyType, null);
}
public static BoolQueryBuilder assembleCacheMapsQuery(String projectId,String contendId, List<String> mediaTypes, Integer channelEmotion) {
public static BoolQueryBuilder assembleCacheMapsQuery(String projectId,String contendId, String politicsLevel, String field, String region,
String mainBodyType,Integer channelEmotion) {
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
String key = concat(projectId, contendId);
BoolQueryBuilder nestedBoolBuilder = QueryBuilders.boolQuery();
......@@ -113,20 +116,20 @@ public class EsQueryTools {
nestedBoolBuilder.must(QueryBuilders.termQuery("brandkbs_cache_maps.channel_emotion", channelEmotion));
}
boolQueryBuilder.must(cacheMapsNestedQuery(nestedBoolBuilder));
// mediaTypes
addMediaTypes(mediaTypes, boolQueryBuilder);
// sensitiveChannel
addSensitiveChannel(politicsLevel, field, region, mainBodyType, boolQueryBuilder);
return boolQueryBuilder;
}
public static BoolQueryBuilder assembleCacheMapsQueryWithProject(String projectId, List<String> mediaTypes) {
public static BoolQueryBuilder assembleCacheMapsQueryWithProject(String projectId, BrandkbsYuQingConfig config) {
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
BoolQueryBuilder nestedBoolBuilder = QueryBuilders.boolQuery();
// 必要条件
nestedBoolBuilder.must(QueryBuilders.termQuery("brandkbs_cache_maps.project_id.keyword", projectId));
// 或需要添加其余nested字段
boolQueryBuilder.must(cacheMapsNestedQuery(nestedBoolBuilder));
// mediaTypes
addMediaTypes(mediaTypes, boolQueryBuilder);
// sensitiveChannel
addSensitiveChannel(config, boolQueryBuilder);
// BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
// mediaTypes.forEach(e -> {
// BoolQueryBuilder mediaTypeQueryBuilder = QueryBuilders.boolQuery();
......@@ -325,6 +328,43 @@ public class EsQueryTools {
return regex.toString();
}
public static void addSensitiveChannel(BrandkbsYuQingConfig config, BoolQueryBuilder boolQueryBuilder) {
BoolQueryBuilder hitBoolQuery = QueryBuilders.boolQuery();
if (!Tools.isEmpty(config.getPoliticsLevels())) {
List<String> hitList = config.getPoliticsLevels();
if (config.getPoliticsLevels().contains("全部")) {
hitList = ChannelType.POLITICS_LEVELS;
}
hitList.forEach(politicsLevel -> hitBoolQuery.should(QueryBuilders.termQuery(GenericAttribute.ES_SENSITIVE_CHANNEL + "." + GenericAttribute.ES_POLITICS_LEVEL + ".keyword", politicsLevel)));
}
if (!Tools.isEmpty(config.getFields())) {
List<String> hitList = config.getFields();
if (config.getFields().contains("全部")) {
hitList = ChannelType.COMMON_FIELDS;
}
hitList.forEach(field -> hitBoolQuery.should(QueryBuilders.termQuery(GenericAttribute.ES_SENSITIVE_CHANNEL + "." + GenericAttribute.ES_FIELD + ".keyword", field)));
}
boolQueryBuilder.must(hitBoolQuery);
}
public static void addSensitiveChannel(String politicsLevel, String field, String region, String mainBodyType, BoolQueryBuilder boolQueryBuilder) {
if (null != politicsLevel) {
boolQueryBuilder.must(QueryBuilders.termQuery(GenericAttribute.ES_SENSITIVE_CHANNEL + "." + GenericAttribute.ES_POLITICS_LEVEL + ".keyword",
politicsLevel));
}
if (null != mainBodyType) {
boolQueryBuilder.must(QueryBuilders.matchPhraseQuery(GenericAttribute.ES_SENSITIVE_CHANNEL + "." + GenericAttribute.ES_MAIN_BODY_TYPE + ".keyword", mainBodyType));
}
if (null != field) {
boolQueryBuilder.must(QueryBuilders.termQuery(GenericAttribute.ES_SENSITIVE_CHANNEL + "." + GenericAttribute.ES_FIELD + ".keyword", field));
}
if (null != region) {
// 使用matchPhraseQuery,例如宁波市 能查询到 “宁波市|鄞州区”
boolQueryBuilder.must(QueryBuilders.matchPhraseQuery(GenericAttribute.ES_SENSITIVE_CHANNEL + "." + GenericAttribute.ES_REGION + ".keyword", region));
}
}
@Deprecated
private static void addMediaTypes(List<String> mediaTypes, BoolQueryBuilder boolQueryBuilder) {
if (CollectionUtils.isNotEmpty(mediaTypes)) {
BoolQueryBuilder mediaTypesBoolQueryBuilder = QueryBuilders.boolQuery();
......
src/main/java/com/zhiwei/brandkbs2/pojo/dto/MarkSearchDTO.java
......@@ -73,9 +73,33 @@ public class MarkSearchDTO {
* 媒体类型
*/
@ApiModelProperty(value = "重要发声方")
@Deprecated
private List<String> mediaTypes;
/**
* 行政级别
*/
@ApiModelProperty(value = "行政级别")
private String politicsLevel;
/**
* 领域
*/
@ApiModelProperty(value = "领域")
private String field;
/**
* 地域
*/
@ApiModelProperty(value = "地域")
private String region;
/**
* 主体类型
*/
@ApiModelProperty(value = "主体类型")
private String mainBodyType;
/**
* 标签筛选
*/
@ApiModelProperty(value = "舆情标签筛选")
......
src/main/java/com/zhiwei/brandkbs2/pojo/external/BrandkbsYuQingConfig.java
......@@ -10,10 +10,13 @@ public class BrandkbsYuQingConfig {
private String configName;
/**
* 发声渠道
* 行政级别
*/
private List<String> channelTypes;
private List<String> politicsLevels;
/**
* 领域
*/
private List<String> fields;
/**
* 文章类型
*/
......
src/main/java/com/zhiwei/brandkbs2/service/impl/CommonServiceImpl.java
......@@ -83,7 +83,16 @@ public class CommonServiceImpl implements CommonService {
@Override
public List<JSONObject> getQbjcPlatform(String... includeFields) {
List<JSONObject> collect = GlobalPojo.PLATFORMS.stream().filter(messagePlatform -> !"外媒".equals(messagePlatform.getName())).map(messagePlatform -> {
List<JSONObject> collect = GlobalPojo.PLATFORMS.stream().filter(messagePlatform -> {
// if ("小红书".equals(messagePlatform.getName())) {
// return false;
// }
// if ("抖音".equals(messagePlatform.getName())) {
// return false;
// }
return !"外媒".equals(messagePlatform.getName());
}
).map(messagePlatform -> {
JSONObject json = new JSONObject();
json.put("id", messagePlatform.getId());
json.put("name", messagePlatform.getName());
......
src/main/java/com/zhiwei/brandkbs2/service/impl/EsSearchServiceImpl.java
package com.zhiwei.brandkbs2.service.impl;
import com.zhiwei.brandkbs2.common.GenericAttribute;
import com.zhiwei.brandkbs2.common.GlobalPojo;
import com.zhiwei.brandkbs2.es.EsClientDao;
import com.zhiwei.brandkbs2.es.EsQueryTools;
import com.zhiwei.brandkbs2.function.ArticleFunction;
import com.zhiwei.brandkbs2.pojo.dto.MarkSearchDTO;
import com.zhiwei.brandkbs2.service.EsSearchService;
import com.zhiwei.brandkbs2.util.Tools;
import com.zhiwei.qbjc.bean.pojo.common.MessagePlatform;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.tuple.Pair;
......@@ -23,7 +21,6 @@ import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
......@@ -77,7 +74,7 @@ public class EsSearchServiceImpl implements EsSearchService {
String contendId = dto.getContendId();
// PostFilter 后置过滤器
// 新增channelEmotion
BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQuery(projectId, contendId, dto.getMediaTypes(), dto.getChannelEmotion());
BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQuery(projectId, contendId, dto.getPoliticsLevel(), dto.getField(), dto.getRegion(), dto.getMainBodyType(), dto.getChannelEmotion());
// time
postFilter.must(QueryBuilders.rangeQuery(dto.getTimeType()).gte(dto.getStartTime()).lt(dto.getEndTime()));
// platform
......@@ -143,7 +140,7 @@ public class EsSearchServiceImpl implements EsSearchService {
String projectId = dto.getProjectId();
String contendId = dto.getContendId();
// PostFilter 后置过滤器
BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQuery(projectId, contendId, dto.getMediaTypes());
BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQuery(projectId, contendId, dto.getPoliticsLevel(), dto.getField(), dto.getRegion(), dto.getMainBodyType(), dto.getChannelEmotion());
// time
postFilter.must(QueryBuilders.rangeQuery(dto.getTimeType()).gte(dto.getStartTime()).lt(dto.getEndTime()));
// platform
......@@ -249,14 +246,15 @@ public class EsSearchServiceImpl implements EsSearchService {
*/
private Map<String, Long> platformCount(int page, List<String> platforms, BoolQueryBuilder postFilter, EsClientDao esClientDao) throws IOException {
Map<String, Long> counts = new HashMap<>();
// TODO 平台进量暂不统计
//平台计量统计 仅第一页且无平台限制
if (1 == page && CollectionUtils.isEmpty(platforms)) {
for (MessagePlatform platform : GlobalPojo.PLATFORMS) {
BoolQueryBuilder queryBuilder = QueryBuilders.boolQuery().must(postFilter).must(EsQueryTools.assemblePlatformQuery(Collections.singletonList(platform)));
Long count = esClientDao.count(esClientDao.getIndexes(), queryBuilder, null);
counts.put(platform.getName(), count);
}
}
// if (1 == page && CollectionUtils.isEmpty(platforms)) {
// for (MessagePlatform platform : GlobalPojo.PLATFORMS) {
// BoolQueryBuilder queryBuilder = QueryBuilders.boolQuery().must(postFilter).must(EsQueryTools.assemblePlatformQuery(Collections.singletonList(platform)));
// Long count = esClientDao.count(esClientDao.getIndexes(), queryBuilder, null);
// counts.put(platform.getName(), count);
// }
// }
return counts;
}
......
src/main/java/com/zhiwei/brandkbs2/service/impl/MarkDataServiceImpl.java
......@@ -8,6 +8,7 @@ import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import com.zhiwei.base.entity.subclass.mark.MarkInfo;
import com.zhiwei.brandkbs2.auth.UserThreadLocal;
import com.zhiwei.brandkbs2.common.ChannelType;
import com.zhiwei.brandkbs2.common.GenericAttribute;
import com.zhiwei.brandkbs2.common.GlobalPojo;
import com.zhiwei.brandkbs2.common.RedisKeyPrefix;
......@@ -296,9 +297,11 @@ public class MarkDataServiceImpl implements MarkDataService {
return json;
}).collect(Collectors.toList()));
// 重要发声方
result.put("mediaTypeList", ImportantChannelEnum.getAllTagExceptSpec());
result.put("channelTypeList", ChannelType.channelOption);
// 其他标签组(筛选器)
result.put("tagFilterList", tagFilterService.getTagFilter());
// 地域
result.put("regionList", getFilterRegionList());
// 自定义标签列表
List<CustomTagVo> customTagList = customTagService.findCustomTagList(1, 1).getList();
if (CollectionUtils.isNotEmpty(customTagList)) {
......@@ -307,6 +310,39 @@ public class MarkDataServiceImpl implements MarkDataService {
return result;
}
private List<JSONObject> getFilterRegionList() {
List<JSONObject> res = new ArrayList<>();
JSONObject beijing = new JSONObject();
beijing.put("name","北京");
beijing.put("id","北京市");
res.add(beijing);
JSONObject shanghai = new JSONObject();
shanghai.put("name","上海");
shanghai.put("id","上海市");
res.add(shanghai);
JSONObject guangzhou = new JSONObject();
guangzhou.put("name","广州");
guangzhou.put("id","广东省|广州市");
res.add(guangzhou);
JSONObject shenzhen = new JSONObject();
shenzhen.put("name","深圳");
shenzhen.put("id","广东省|深圳市");
res.add(shenzhen);
JSONObject zhejiang = new JSONObject();
zhejiang.put("name","浙江");
zhejiang.put("id","浙江省");
res.add(zhejiang);
JSONObject sichuan = new JSONObject();
sichuan.put("name","四川");
sichuan.put("id","四川省");
res.add(sichuan);
JSONObject chongqing = new JSONObject();
chongqing.put("name","重庆");
chongqing.put("id","重庆市");
res.add(chongqing);
return res;
}
@Override
public JSONObject getMobileYuqingMarkCriteria() {
String projectId = UserThreadLocal.getProjectId();
......
src/main/java/com/zhiwei/brandkbs2/service/impl/MarkFlowServiceImpl.java
......@@ -29,10 +29,7 @@ import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.*;
import java.util.stream.Collectors;
/**
......@@ -212,24 +209,16 @@ public class MarkFlowServiceImpl implements MarkFlowService {
sourceDetails.put("followersNum", followersNum);
}
// 渠道标签
String channelTag = tJson.getString("channel_tag");
if (null != channelTag) {
sourceDetails.put("channelTag", channelTag.replaceAll(",", "|"));
JSONObject sensitiveChannel = tJson.getJSONObject(GenericAttribute.ES_SENSITIVE_CHANNEL);
if (null != sensitiveChannel && !sensitiveChannel.isEmpty()) {
StringBuilder sb = new StringBuilder();
addChannelTag(sensitiveChannel.getString(GenericAttribute.ES_POLITICS_LEVEL), sb);
addChannelTag(sensitiveChannel.getString(GenericAttribute.ES_MAIN_BODY_TYPE), sb);
addChannelTag(sensitiveChannel.getString(GenericAttribute.ES_FIELD), sb);
addChannelTagRegion(sensitiveChannel.getString(GenericAttribute.ES_REGION), sb);
sourceDetails.put("channelTag", sb.toString());
}
// 渠道倾向及id(调整为快照属性)
// Channel channel = channelDao.queryUnique(ChannelIndex.createChannelIndex(tJson, projectId, contendId));
// if (null != channel) {
// sourceDetails.put("channelId", channel.getId());
// sourceDetails.put("channelEmotion", ChannelEmotion.getNameFromState(channel.getEmotion()));
// } else {
// List<Map<String, Object>> cacheMaps = (List<Map<String, Object>>) tJson.get(GenericAttribute.ES_BRANDKBS_CACHE_MAPS);
// if (null != cacheMaps) {
// Map<String, Object> hitMap =
// cacheMaps.stream().filter(map -> projectId.equals(map.get("project_id")) && contendId.equals(map.get("contend_id"))).findAny().orElse(Collections.emptyMap());
// sourceDetails.put("channelId", hitMap.get("channel_id"));
// sourceDetails.put("channelEmotion", ChannelEmotion.getNameFromState(hitMap.get("channel_emotion")));
// }
// }
// 渠道倾向及id
JSONObject brandkbsHitJson = Tools.getBrandkbsHitMap(tJson, Tools.concat(markInfoSource.getProjectId(), markInfoSource.getContendId()));
if (null != brandkbsHitJson.get("channel_emotion")) {
sourceDetails.put("channelEmotion", ChannelEmotion.getNameFromState(brandkbsHitJson.getIntValue("channel_emotion")));
......@@ -315,4 +304,35 @@ public class MarkFlowServiceImpl implements MarkFlowService {
return res;
}
private void addChannelTag(String value, StringBuilder channelTag) {
if (null != value) {
if (0 == channelTag.length()) {
channelTag.append(value);
} else {
channelTag.append("|").append(value);
}
}
}
private void addChannelTagRegion(String region, StringBuilder channelTag) {
if (null != region) {
StringBuilder newRegion = new StringBuilder();
String[] split = region.split("\\|", 2);
for (String s : split) {
if (0 == newRegion.length()) {
newRegion.append(s);
} else {
if (!newRegion.toString().contains(s)) {
newRegion.append(" ").append(s);
}
}
}
if (0 == channelTag.length()) {
channelTag.append(newRegion);
} else {
channelTag.append("|").append(newRegion);
}
}
}
}
src/main/java/com/zhiwei/brandkbs2/service/impl/ProjectWarnServiceImpl.java
......@@ -3,6 +3,7 @@ package com.zhiwei.brandkbs2.service.impl;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.zhiwei.brandkbs2.common.ChannelType;
import com.zhiwei.brandkbs2.common.RedisKeyPrefix;
import com.zhiwei.brandkbs2.config.Constant;
import com.zhiwei.brandkbs2.dao.ChannelDao;
......@@ -12,10 +13,7 @@ import com.zhiwei.brandkbs2.enmus.EmotionEnum;
import com.zhiwei.brandkbs2.es.EsClientDao;
import com.zhiwei.brandkbs2.es.EsQueryTools;
import com.zhiwei.brandkbs2.model.ResponseResult;
import com.zhiwei.brandkbs2.pojo.AbstractProject;
import com.zhiwei.brandkbs2.pojo.BaseMap;
import com.zhiwei.brandkbs2.pojo.Channel;
import com.zhiwei.brandkbs2.pojo.Project;
import com.zhiwei.brandkbs2.pojo.*;
import com.zhiwei.brandkbs2.pojo.external.*;
import com.zhiwei.brandkbs2.pojo.vo.CrisisCaseWarnVO;
import com.zhiwei.brandkbs2.service.ProjectService;
......@@ -105,7 +103,8 @@ public class ProjectWarnServiceImpl implements ProjectWarnService {
Map<String, Object> res = new HashMap<>();
switch (type) {
case "舆情动态":
CriteriaHelper.warnCriteriaAddSimple(res, "channelTypes", Arrays.asList("央级", "门户", "财经", "科技", "其他"), Collections.singletonList("央级"));
CriteriaHelper.warnCriteriaAddSimple(res, "politicsLevels", Arrays.asList("央级", "省级", "地级"), Collections.singletonList("央级"));
CriteriaHelper.warnCriteriaAddSimple(res, "fields", ChannelType.COMMON_FIELDS, Collections.singletonList("财经"));
CriteriaHelper.warnCriteriaAdd(res, "primary", Arrays.asList("原创", "转发"), Arrays.asList(false, true), Collections.singletonList("原创"));
CriteriaHelper.warnCriteriaAddSimple(res, "markerTags", Arrays.asList("正面", "中性", "负面"), Collections.singletonList("负面"));
List<JSONObject> brands = projectService.getBrands(projectId, false);
......@@ -172,7 +171,8 @@ public class ProjectWarnServiceImpl implements ProjectWarnService {
switch (type) {
case "舆情动态":
BrandkbsYuQingConfig config = new BrandkbsYuQingConfig();
config.setChannelTypes(Collections.singletonList("央级"));
config.setPoliticsLevels(Collections.singletonList("央级"));
config.setFields(Collections.singletonList("财经"));
config.setPrimary(Collections.singletonList(Boolean.TRUE));
config.setMarkerTags(Collections.singletonList("负面"));
config.setDuplicate(Boolean.FALSE);
......@@ -221,13 +221,43 @@ public class ProjectWarnServiceImpl implements ProjectWarnService {
// key1
String key1 = "【品见】舆情动态-" + project.getProjectName();
// key2
String key2Header = "全部";
// String key2Header = "全部";
String key2Header = "";
StringBuilder mediaChannel = new StringBuilder();
config.getChannelTypes().forEach(channelType -> mediaChannel.append(channelType).append("、"));
List<String> politicsLevels = config.getPoliticsLevels();
List<String> fields = config.getFields();
if(Tools.isEmpty(politicsLevels)){
key2Header += "【全部级别】";
}else if(politicsLevels.contains("全部")){
key2Header += "【重点级别】";
}else{
politicsLevels.forEach(politicsLevel -> mediaChannel.append(politicsLevel).append("、"));
}
if (Tools.isEmpty(fields)) {
if (0 != key2Header.length()) {
key2Header += "、";
key2Header += "【全部领域】,";
} else {
key2Header += "【全部领域】";
}
} else if (fields.contains("全部")) {
if (0 != key2Header.length()) {
key2Header += "、";
key2Header += "【重点领域】,";
} else {
key2Header += "【重点领域】";
}
} else {
fields.forEach(field -> mediaChannel.append(field).append("、"));
}
if (0 != mediaChannel.length()) {
key2Header = mediaChannel.deleteCharAt(mediaChannel.length() - 1).toString();
if (0 != key2Header.length()) {
key2Header += "、";
}
// 替换“全部” 并去掉末位的、
key2Header += mediaChannel.deleteCharAt(mediaChannel.length() - 1).toString();
key2Header += "媒体,";
}
key2Header += "媒体,";
StringBuilder dataType = new StringBuilder();
config.getMarkerTags().forEach(emotion -> dataType.append(emotion).append(" "));
dataType.append("(本品)");
......@@ -593,7 +623,7 @@ public class ProjectWarnServiceImpl implements ProjectWarnService {
EsClientDao.SearchHelper helper = EsClientDao.createSearchHelper();
// 发声渠道
BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQueryWithProject(projectId, config.getChannelTypes());
BoolQueryBuilder postFilter = EsQueryTools.assembleCacheMapsQueryWithProject(projectId, config);
// 时间(标注时间)
postFilter.must(QueryBuilders.rangeQuery("mtime").gte(start).lt(end));
// 文章类型
......
src/main/java/com/zhiwei/brandkbs2/util/Tools.java
......@@ -40,6 +40,7 @@ import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.WritableByteChannel;
import java.nio.charset.StandardCharsets;
import java.text.ParseException;
import java.util.*;
import java.util.regex.Matcher;
......@@ -580,6 +581,28 @@ public class Tools {
return result;
}
/**
* 读取json文件，返回json串
*
* @param fileInputStream
* @return
*/
public static String readJsonFile(InputStream fileInputStream) {
try {
Reader reader = new InputStreamReader(fileInputStream, StandardCharsets.UTF_8);
int ch = 0;
StringBuilder sb = new StringBuilder();
while ((ch = reader.read()) != -1) {
sb.append((char) ch);
}
reader.close();
return sb.toString();
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
public static Map<String, Integer> sortMap(Map<String, Integer> map, Integer maxSize) {
Map<String, Integer> result = new LinkedHashMap<>();
List<Map.Entry<String, Integer>> collect;
......
src/main/resources/chinaAreaTag.json 0 → 100644
This source diff could not be displayed because it is too large. You can view the blob instead.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment