org.teasoft.honey.sharding.engine.mongodb.MongodbShardingSelectFunEngine Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of honey Show documentation
Honey is the implementation of the Bee.
The newest version!
/*
 * Copyright 2016-2023 the original author.All rights reserved.
 * Kingstar([email protected])
 * The license,see the LICENSE file.
 */

package org.teasoft.honey.sharding.engine.mongodb;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.CompletionService;
import java.util.concurrent.ExecutorCompletionService;
import java.util.concurrent.ExecutorService;

import org.teasoft.bee.mongodb.MongoSqlStruct;
import org.teasoft.bee.mongodb.MongodbBeeSql;
import org.teasoft.bee.osql.FunctionType;
import org.teasoft.honey.osql.core.HoneyContext;
import org.teasoft.honey.osql.core.ShardingLogReg;
import org.teasoft.honey.osql.core.StringConst;
import org.teasoft.honey.sharding.ShardingUtil;
import org.teasoft.honey.sharding.engine.ResultMergeEngine;
import org.teasoft.honey.sharding.engine.ShardingFunResultEngine;
import org.teasoft.honey.sharding.engine.ThreadPoolUtil;

/**
 * 分片的select操作
 * 主要是对函数操作的分片.
 * @author AiTeaSoft
 * @since  2.0
 */
public class MongodbShardingSelectFunEngine {

	public  String asynProcess(Class entityClass, MongodbBeeSql mongodbBeeSql,
			MongoSqlStruct struct) {

		List list;
		String dsArray[];
		String tabArray[];

		if (ShardingUtil.hadShardingFullSelect()) {// 全域查询 或某些DS的某表全查询
			list = MongodbShardingRouter._findDsTabForFull(entityClass);
		} else {
			list = MongodbShardingRouter._findDsTab(); // 涉及部分分片
		}
		dsArray = list.get(0);
		tabArray = list.get(1);

		final List> tasks = new ArrayList<>(); 

		for (int i = 0; dsArray != null && i < dsArray.length; i++) {
			tasks.add(new ShardingBeeSQLFunExecutorEngine(tabArray[i], i + 1, mongodbBeeSql,
					dsArray[i], entityClass, struct));
		}

		if (dsArray != null) ShardingLogReg.log(dsArray.length);

		int size=tasks.size();
		if(size==0) return null;
		
		ExecutorService executor = ThreadPoolUtil.getThreadPool(size);
		CompletionService completionService = new ExecutorCompletionService<>(executor);
		for (int i = 0; tasks != null && i < size; i++) {
			completionService.submit(tasks.get(i));
		}

		List rsList = ResultMergeEngine.mergeFunResult(completionService, size);

		executor.shutdown();

		return ShardingFunResultEngine.funResultEngine(rsList);
	}

//	Return String 	
	@SuppressWarnings("rawtypes")
	private class ShardingBeeSQLFunExecutorEngine
			extends ShardingAbstractMongoBeeSQLExecutorEngine {

		private Class entityClass;
		private MongoSqlStruct struct;

		public ShardingBeeSQLFunExecutorEngine(String tab, int index,
				MongodbBeeSql mongodbBeeSql, String ds, Class entityClass,
				MongoSqlStruct struct) {
			super(tab, index, mongodbBeeSql, ds);
			this.entityClass = entityClass;
			this.struct = struct.copy();
			this.struct.setTableName(tab);
		}

		@SuppressWarnings("unchecked")
		public String shardingWork() {
			ShardingLogReg.regShardingSqlLog("select fun SQL", index, tab);
			String rsStr = "";
			String typeTag = "";
			typeTag = "select fun";

			String funType = HoneyContext.getSysCommStrInheritableLocal(StringConst.FunType);
			if (FunctionType.COUNT.getName().equalsIgnoreCase(funType)) {
				rsStr = mongodbBeeSql.count(struct, entityClass);
			} else {
				rsStr = mongodbBeeSql.selectWithFun(struct, entityClass);
			}

			ShardingLogReg.regShardingSqlLog("    | <--  " + typeTag, index, rsStr);

			return rsStr;
		}
	}

}