wanglihui-ip-learning-graph/ip-learning-spark/src/main/java/cn/ac/iie/dao/BaseArangoData.java

package cn.ac.iie.dao;

import cn.ac.iie.config.ApplicationConfig;
import cn.ac.iie.service.read.ReadHistoryArangoData;
import cn.ac.iie.utils.ArangoDBConnect;
import cn.ac.iie.utils.ExecutorThreadPool;
import com.arangodb.ArangoCursor;
import com.arangodb.entity.BaseDocument;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CountDownLatch;

/**
 * 获取arangoDB历史数据
 *
 * @author wlh
 */
public class BaseArangoData {
    private static final Logger LOG = LoggerFactory.getLogger(BaseArangoData.class);
    private static ArangoDBConnect arangoDBConnect = ArangoDBConnect.getInstance();

    private ExecutorThreadPool threadPool = ExecutorThreadPool.getInstance();

    public <T extends BaseDocument> ConcurrentHashMap<Integer, ConcurrentHashMap<String, T>> readHistoryData(String table, Class<T> type) {
        ConcurrentHashMap<Integer, ConcurrentHashMap<String, T>> historyMap = new ConcurrentHashMap<>();
        try {
            LOG.warn("开始更新" + table);
            long start = System.currentTimeMillis();
            for (int i = 0; i < ApplicationConfig.THREAD_POOL_NUMBER(); i++) {
                historyMap.put(i, new ConcurrentHashMap<>());
            }
            CountDownLatch countDownLatch = new CountDownLatch(ApplicationConfig.THREAD_POOL_NUMBER());
            Long countTotal = getCountTotal(table);
            for (int i = 0; i < ApplicationConfig.THREAD_POOL_NUMBER(); i++) {
                String sql = getQuerySql(countTotal, i, table);
                ReadHistoryArangoData<T> readHistoryArangoData = new ReadHistoryArangoData<>(arangoDBConnect, sql, historyMap, type, table, countDownLatch);
                threadPool.executor(readHistoryArangoData);
            }
            countDownLatch.await();
            long last = System.currentTimeMillis();
            LOG.warn("读取" + table + " arangoDB 共耗时：" + (last - start));
        } catch (Exception e) {
            e.printStackTrace();
        }
        return historyMap;
    }

    private Long getCountTotal(String table){
        long start = System.currentTimeMillis();
        Long cnt = 0L;
        String sql = "RETURN LENGTH("+table+")";
        try {
            ArangoCursor<Long> longs = arangoDBConnect.executorQuery(sql, Long.class);
            while (longs.hasNext()){
                cnt = longs.next();
            }
        }catch (Exception e){
            LOG.error(sql +"执行异常");
        }
        long last = System.currentTimeMillis();
        LOG.warn(sql+" 结果："+cnt+" 执行时间："+(last-start));
        return cnt;
    }

    private String getQuerySql(Long cnt,int threadNumber, String table){
        long sepNum = cnt / ApplicationConfig.THREAD_POOL_NUMBER() + 1;
        long offsetNum = threadNumber * sepNum;
        if (sepNum >= ApplicationConfig.ARANGODB_READ_LIMIT() * 10000){
            sepNum = ApplicationConfig.ARANGODB_READ_LIMIT() * 10000;
        }
        return "FOR doc IN " + table + " limit "+offsetNum+","+sepNum+" RETURN doc";
    }


}