@ -22,20 +22,30 @@ import com.fr.decision.webservice.bean.datasource.JDBCConnectionBean;
import com.fr.decision.webservice.utils.DecisionServiceConstants ;
import com.fr.decision.webservice.v10.datasource.connection.processor.impl.ConnectionProcessorFactory ;
import com.fr.decision.webservice.v10.datasource.connection.processor.impl.JDBCConnectionProcessor ;
import com.fr.event.Event ;
import com.fr.event.EventDispatcher ;
import com.fr.event.Listener ;
import com.fr.event.Null ;
import com.fr.log.FineLoggerFactory ;
import com.fr.module.tool.ActivatorToolBox ;
import com.fr.security.encryption.transmission.TransmissionEncryptionManager ;
import com.fr.stable.Constants ;
import com.fr.stable.StringUtils ;
import com.fr.third.fasterxml.jackson.databind.ObjectMapper ;
import com.fr.third.guava.cache.CacheBuilder ;
import com.fr.third.guava.cache.CacheLoader ;
import com.fr.third.guava.cache.LoadingCache ;
import com.fr.third.springframework.beans.BeanUtils ;
import com.fr.workspace.server.repository.WorkplaceConstants ;
import com.fr.workspace.server.repository.connection.ConnectionCacheEvent ;
import com.fr.workspace.server.repository.connection.ConnectionRepository ;
import com.fr.workspace.server.repository.tabledata.DataEncryptionHelper ;
import java.util.HashMap ;
import java.util.Map ;
import java.util.Optional ;
import java.util.concurrent.ExecutionException ;
import java.util.concurrent.TimeUnit ;
/ * *
* 数据连接传输工具类
@ -50,19 +60,38 @@ public class ConnectionInfoBeanHelper {
private static final int ORACLE_DEFAULT_FETCH_SIZE = 128 ;
private static final int DB2_DEFAULT_FETCH_SIZE = 50 ;
private static final int POSTGRE_DEFAULT_FETCH_SIZE = 10000 ;
public static final int CACHE_MAX_SIZE = 1 ;
public static final String VALUE = "value" ;
private static LoadingCache < String , Map < String , Connection > > cache = CacheBuilder . newBuilder ( )
// 缓存容量1
. maximumSize ( CACHE_MAX_SIZE )
// 默认写入过60S后过期
. expireAfterWrite ( 60 , TimeUnit . SECONDS )
. build ( new CacheLoader < String , Map < String , Connection > > ( ) {
@Override
public Map < String , Connection > load ( String s ) throws Exception {
return updateCache ( ) ;
}
} ) ;
static {
FETCH_SIZE_MAP . put ( "oracle" , ORACLE_DEFAULT_FETCH_SIZE ) ;
FETCH_SIZE_MAP . put ( "ibm-db2" , DB2_DEFAULT_FETCH_SIZE ) ;
FETCH_SIZE_MAP . put ( "postgresql" , POSTGRE_DEFAULT_FETCH_SIZE ) ;
// 监听数据连接缓存失效
EventDispatcher . listen ( ConnectionCacheEvent . REMOVE , new Listener < Null > ( ) {
@Override
public void on ( Event event , Null aNull ) {
cache . invalidateAll ( ) ;
}
} ) ;
}
/ * *
* 获取连接Map
* /
public static Map < String , Connection > createConnectionMap ( ConnectionInfoBean [ ] beans ) {
private static Map < String , Connection > updateCache ( ) {
Map < String , Connection > ans = new HashMap < > ( ) ;
ConnectionInfoBean [ ] beans = ConnectionRepository . getInstance ( ) . getAll ( ) ;
for ( ConnectionInfoBean infoBean : beans ) {
try {
if ( JDBCConnectionProcessor . KEY . acceptConnectionTypes ( ) . contains ( infoBean . getConnectionType ( ) ) ) {
@ -77,23 +106,33 @@ public class ConnectionInfoBeanHelper {
return ans ;
}
/ * *
* 获取连接Map
* /
public static Map < String , Connection > getCurrentConnectionMap ( ) {
try {
return cache . get ( VALUE ) ;
} catch ( ExecutionException e ) {
throw new RuntimeException ( e ) ;
}
}
/ * *
* 根据bean创建连接
* /
public static Connection createConnection ( ConnectionInfoBean infoBean ) {
Connection connection = null ;
try {
if ( JDBCConnectionProcessor . KEY . acceptConnectionTypes ( ) . contains ( infoBean . getConnectionType ( ) ) ) {
connection = createJDBCConnection ( infoBean ) ;
} else {
connection = ConnectionProcessorFactory . createConnection ( infoBean , true ) ;
}
connection . setConnectionName ( infoBean . getConnectionName ( ) ) ;
} catch ( Exception e ) {
FineLoggerFactory . getLogger ( ) . error ( e . getMessage ( ) , e ) ;
return connection ;
}
return connection ;
try {
Connection connection = cache . get ( VALUE ) . get ( infoBean . getConnectionName ( ) ) ;
if ( connection ! = null ) {
connection . setConnectionName ( infoBean . getConnectionName ( ) ) ;
}
return connection ;
} catch ( ExecutionException e ) {
throw new RuntimeException ( e ) ;
}
}
/ * *
