package com.hbbh.adapter.manager.impl;
|
|
import com.google.common.collect.Maps;
|
import com.hbbh.adapter.config.GoldenConfig;
|
import com.hbbh.adapter.covert.PointConvert;
|
import com.hbbh.adapter.dao.golden.*;
|
import com.hbbh.adapter.enums.TableEnum;
|
import com.hbbh.adapter.manager.DataSource;
|
import com.hbbh.adapter.pojo.Point;
|
import com.hbbh.adapter.pojo.golden.*;
|
import com.hbbh.adapter.utils.SearchConditionUtil;
|
import com.rtdb.enums.DataSort;
|
import com.rtdb.model.Entity;
|
import com.rtdb.model.FullPoint;
|
import com.rtdb.model.Table;
|
import com.rtdb.service.impl.BaseImpl;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.TreeMap;
|
|
@Service
|
public class DataSourceImpl implements DataSource
|
{
|
private static final Logger log = LoggerFactory.getLogger(Class.class);
|
|
|
@Autowired
|
GoldenConfig goldenConfig;
|
|
|
/**
|
* @param pointIds 标签点id数组
|
* @return
|
* @throws Exception
|
*/
|
public List<FullPoint> selectFullPoint(int[] pointIds) throws Exception {
|
BaseImpl base = new BaseImpl(goldenConfig.getServer());
|
|
Entity<FullPoint> pointsProperties = base.getPointsProperties(pointIds);
|
List<FullPoint> list = pointsProperties.getList();
|
return list;
|
}
|
|
|
/**
|
* 获取表与标签数量关系
|
* @return
|
* @throws Exception
|
*/
|
public Map<Table,Integer> getCount() throws Exception {
|
Map<Table,Integer> map = Maps.newHashMap();
|
|
BaseImpl base = new BaseImpl(goldenConfig.getServer());
|
|
int[] tableIds = base.getTableIds();
|
|
for (int tableId : tableIds) {
|
int i = base.getTableRealSizeById(tableId);
|
Table table = base.getTablePropertiesById(tableId);
|
map.put(table,i);
|
}
|
return map;
|
}
|
|
/*
|
public List<Point> selectEntity() throws Exception {
|
int count = 0;
|
BaseImpl base = new BaseImpl(goldenConfig.getServer());
|
|
int[] pointIds = base.search(SearchConditionUtil.getSearchCondition(),
|
count,
|
DataSort.SORT_BY_ID);
|
List<FullPoint> fullPoints = selectFullPoint(pointIds);
|
|
List<Point> results = PointConvert.fullpointsToPointEntity(fullPoints);
|
return results;
|
}*/
|
|
/**
|
* k - tableName
|
* v - point
|
* @return
|
* @throws Exception
|
*/
|
public Map<String,List<Point>> selectAllEntity() throws Exception {
|
Map<String, List<Point>> result = Maps.newTreeMap();
|
//Map<Table,List<Point>> result=Maps.newHashMap();
|
|
Map<Table, Integer> map = getCount();
|
|
BaseImpl base = new BaseImpl(goldenConfig.getServer());
|
|
map.forEach((k,v)->{
|
try {
|
int[] pointIds = base.search(SearchConditionUtil.getSearchCondition(),
|
v,
|
DataSort.SORT_BY_ID);
|
List<FullPoint> fullPoints = selectFullPoint(pointIds);
|
List<Point> points = PointConvert.fullpointsToPointEntity(fullPoints);
|
result.put(k.getName(),points);
|
}catch (Exception e){
|
throw new RuntimeException("取数据失败");
|
}
|
});
|
return result;
|
}
|
|
public void execute() throws Exception {
|
Map<String, List<Point>> tableListMap = selectAllEntity();
|
|
for (Map.Entry<String, List<Point>> entry : tableListMap.entrySet()) {
|
String k = entry.getKey();
|
List<Point> v = entry.getValue();
|
insert(k,v);
|
|
}
|
}
|
|
@Autowired
|
SMSDao smsDao;
|
@Autowired
|
DLZ_PLCDao dlzPlcDao;
|
@Autowired
|
DJW_PLCDao djwPlcDao;
|
@Autowired
|
BDS_PLCDao bdsPlcDao;
|
@Autowired
|
KTNZ2C_PLCDao ktnz2CPlcDao;
|
|
|
public void insert(String flag,List<Point> point){
|
System.out.println(flag);
|
|
switch (flag) {
|
case "SMS":
|
point.forEach(p->{
|
SMS data=PointConvert.pointToSMS(p);
|
smsDao.saveAndFlush(data);
|
});
|
break;
|
case "DLZ_PLC":
|
point.forEach(p->{
|
DLZ_PLC data=PointConvert.pointToDLZPLC(p);
|
dlzPlcDao.saveAndFlush(data);
|
});
|
break;
|
case "DJW_PLC":
|
point.forEach(p->{
|
DJW_PLC data=PointConvert.pointToDJWPLC(p);
|
djwPlcDao.saveAndFlush(data);
|
});
|
break;
|
case "BDS_PLC":
|
point.forEach(p->{
|
BDS_PLC data=PointConvert.pointToBDSPLC(p);
|
bdsPlcDao.saveAndFlush(data);
|
});
|
break;
|
case "KTNZ2C_PLC":
|
point.forEach(p->{
|
KTNZ2C_PLC data=PointConvert.pointToKTNZ2CPLC(p);
|
ktnz2CPlcDao.saveAndFlush(data);
|
});
|
break;
|
}
|
|
}
|
}
|
