NebulaGraph学习笔记-自定义SessionPool

最近看了一下NebulaGraph图数据库连接的部分资料，发现还可以通过SessionPool的方式连接。下面是一个简单的DEMO记录。

组件项目

相关依赖包

<!-- SpringBoot依赖包 -->
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot</artifactId>
</dependency><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-autoconfigure</artifactId>
</dependency><!-- Client依赖包 -->
<dependency><groupId>com.vesoft</groupId><artifactId>client</artifactId><version>3.8.4</version>
</dependency>

NebulaGraph连接属性类

java">@Data
@ConfigurationProperties(prefix = "nebula-graph")
public class NebulaGraphProperties {/** 集群节点 */private String[] clusterNodes = null;/** 用户名 */private String username;/** 密码 */private String password;private List<CSessionPoolConfig> cSessionPoolConfigs = new ArrayList<>();@Datapublic static class CSessionPoolConfig {/** 空间名称 */private String spaceName;/** 最小Session数 */private int minSessionSize = 10;/** 最大Session数 */private int maxSessionSize = 100;/** 是否重连 */private boolean reconnect = true;/** Timeout */private int timeout = 0;/** WaitTime */private int waitTime = 0;/** CleanTime */private int cleanTime = 3600;/** IntervalTime */private int intervalTime = 0;/** Health Check Time */private int healthCheckTime = 600;/** Retry Connect Times */private int retryConnectTimes = 1;/** Retry Times */private int retryTimes = 3;}}

SessionPoolBeanDefinitionRegistrar类

java">public class SessionPoolBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar, EnvironmentAware {private NebulaGraphProperties nebulaGraphProperties;@Overridepublic void registerBeanDefinitions(AnnotationMetadata annotationMetadata, BeanDefinitionRegistry beanDefinitionRegistry) {if (ObjectUtil.isNull(nebulaGraphProperties)) {return;}List<NebulaGraphProperties.CSessionPoolConfig> cSessionPoolConfigs = nebulaGraphProperties.getCSessionPoolConfigs();if (cSessionPoolConfigs.isEmpty()) {return;}List<HostAddress> hostAddresses = new ArrayList<>();String[] clusterNodes = nebulaGraphProperties.getClusterNodes();for (int i = 0, len = clusterNodes.length; i < len; i++) {String clusterNode = clusterNodes[i];if (!clusterNode.contains(":")) {continue;}String[] ipAndPort = clusterNode.split(":");if (ipAndPort.length != 2 || !ipAndPort[1].matches("\\d+")) {throw new RuntimeException("Invalid Nebula Graph Node " + clusterNode);}hostAddresses.add(new HostAddress(ipAndPort[0], Integer.parseInt(ipAndPort[1])));}String username = nebulaGraphProperties.getUsername();String password = nebulaGraphProperties.getPassword();cSessionPoolConfigs.forEach(cSessionPoolConfig -> {String spaceName = cSessionPoolConfig.getSpaceName();if (CharSequenceUtil.isBlank(spaceName)) {return;}SessionPoolConfig sessionPoolConfig = new SessionPoolConfig(hostAddresses, spaceName, username, password);sessionPoolConfig.setMinSessionSize(cSessionPoolConfig.getMinSessionSize());sessionPoolConfig.setMaxSessionSize(cSessionPoolConfig.getMaxSessionSize());sessionPoolConfig.setTimeout(cSessionPoolConfig.getTimeout());sessionPoolConfig.setWaitTime(cSessionPoolConfig.getWaitTime());sessionPoolConfig.setCleanTime(cSessionPoolConfig.getCleanTime());sessionPoolConfig.setIntervalTime(cSessionPoolConfig.getIntervalTime());sessionPoolConfig.setHealthCheckTime(cSessionPoolConfig.getHealthCheckTime());sessionPoolConfig.setRetryConnectTimes(cSessionPoolConfig.getRetryConnectTimes());sessionPoolConfig.setRetryTimes(cSessionPoolConfig.getRetryTimes());sessionPoolConfig.setReconnect(cSessionPoolConfig.isReconnect());RootBeanDefinition rootBeanDefinition = new RootBeanDefinition();rootBeanDefinition.setBeanClass(SessionPool.class);rootBeanDefinition.getConstructorArgumentValues().addGenericArgumentValue(sessionPoolConfig);beanDefinitionRegistry.registerBeanDefinition(String.format("%sSessionPool", spaceName), rootBeanDefinition);});}@Overridepublic void setEnvironment(Environment environment) {BindResult<NebulaGraphProperties> bindResult = Binder.get(environment).bind(nebulaGraphPropertiesPrefix(), NebulaGraphProperties.class);if (ObjectUtil.isNotNull(bindResult) && bindResult.isBound()) {nebulaGraphProperties = bindResult.get();}}private String nebulaGraphPropertiesPrefix() {return Objects.requireNonNull(AnnotationUtils.getAnnotation(NebulaGraphProperties.class, ConfigurationProperties.class)).prefix();}}

SpringBoot自动配置

java">@EnableConfigurationProperties({ NebulaGraphProperties.class })
@Import({ SessionPoolBeanDefinitionRegistrar.class })
@Configuration
public class NebulaGraphAutoConfiguration {}

spring.factories文件开启自动配置

java">org.springframework.boot.autoconfigure.EnableAutoConfiguration=\com.component.nebula.graph.config.NebulaGraphAutoConfiguration

业务项目

引入组件项目

java"><!--ComponentNebulaGraph依赖包-->
<dependency><groupId>com.component</groupId><artifactId>component-nebula-graph</artifactId><version>1.0.0-SNAPSHOT</version>
</dependency>

项目引入配置

nebula-graph:cluster-nodes:- 192.168.0.1:9559- 192.168.0.1:9669username: rootpassword: 123456c-session-pool-configs:- space-name: tmp1_space_namemin-session-size: 10max-session-size: 200

项目引入部分代码

java">@Slf4j
@Service("nebulaGraphService")
public class NebulaGraphServiceImpl implements NebulaGraphService {@Autowiredprivate SessionPool sessionPool;protected String executeJsonWithParameter(String statement, Map<String, Object> paramsMap)throws ClientServerIncompatibleException, AuthFailedException, BindSpaceFailedException, IOErrorException {return sessionPool.executeJsonWithParameter(statement, paramsMap);}}