コード例 #1
0
        /**
         * Load table meta data.
         *
         * @param logicTableName logic table name
         * @param databaseType database type
         * @return table meta data
         * @throws SQLException SQL exception
         */
        public TableMetaData Load(string logicTableName, IDatabaseType databaseType)
        {
            TableRule tableRule = shardingRule.GetTableRule(logicTableName);

            if (!isCheckingMetaData)
            {
                DataNode dataNode = tableRule.ActualDataNodes.First();
                var      rawMasterDataSourceName = shardingRule.ShardingDataSourceNames.GetRawMasterDataSourceName(dataNode.GetDataSourceName());
                return(TableMetaDataLoader.Load(dataSourceMap[rawMasterDataSourceName], dataNode.GetTableName(), databaseType.GetName()));
            }
            var dataNodeGroups = tableRule.GetDataNodeGroups();
            ConcurrentDictionary <string, TableMetaData> actualTableMetaDataMap = new ConcurrentDictionary <string, TableMetaData>();

            var tasks = dataNodeGroups.SelectMany(o => o.Value.Select(dataNode => Task.Run(() =>
            {
                var tableMetaData = Load(dataNode, databaseType);
                actualTableMetaDataMap.TryAdd(dataNode.GetTableName(), tableMetaData);
            }))).ToArray();

            Task.WaitAll(tasks);


            CheckUniformed(logicTableName, actualTableMetaDataMap);
            return(actualTableMetaDataMap.Select(o => o.Value).FirstOrDefault());
        }
コード例 #2
0
 private TableMetaData Load(DataNode dataNode, IDatabaseType databaseType)
 {
     try
     {
         return(TableMetaDataLoader.Load(dataSourceMap[dataNode.GetDataSourceName()], dataNode.GetTableName(), databaseType.GetName()));
     }
     catch (Exception e)
     {
         throw new ShardingException($"SqlException for DataNode={dataNode} and databaseType={databaseType.GetName()}", e);
     }
 }