Bläddra i källkod

向量匹配优化

wangyunpeng 5 dagar sedan
förälder
incheckning
785994c197

+ 10 - 1
core/src/main/java/com/tzld/videoVector/config/db/PgVectorDBConfig.java

@@ -4,7 +4,9 @@ import com.zaxxer.hikari.HikariDataSource;
 import org.apache.ibatis.session.SqlSessionFactory;
 import org.mybatis.spring.SqlSessionFactoryBean;
 import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.jdbc.DataSourceBuilder;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
@@ -18,11 +20,18 @@ import javax.sql.DataSource;
 @EnableTransactionManagement
 public class PgVectorDBConfig {
 
+    @Value("${spring.datasource.pg-vector.connection-init-sql:}")
+    private String connectionInitSql;
+
     // 1. 配置 pgVector 数据源
     @Bean(name = "pgVectorDataSource")
     @ConfigurationProperties(prefix = "spring.datasource.pg-vector")
     public DataSource pgVectorDataSource() {
-        return new HikariDataSource();
+        HikariDataSource dataSource = (HikariDataSource) DataSourceBuilder.create().build();
+        if (connectionInitSql != null && !connectionInitSql.isEmpty()) {
+            dataSource.setConnectionInitSql(connectionInitSql);
+        }
+        return dataSource;
     }
 
     // 2. 配置 pgVector 专属 SqlSessionFactory(独立 Configuration,避免与 MySQL 数据源互相污染)