基于SpringDataElasticsearch+SpEL表达式实现ES动态索引

女爷i 2024-05-06 21:30 88阅读 0赞

### 前言 ###

一般情况下，当我们使用 `SpringDataElasticsearch` 去操作 `ES` 时，索引名称都会在 `@Document` 注解中写死，每次都是对这个固定的索引进行操作。

假如我们现在处于一个多租户系统中，每个租户都有自己所对应的用户数据，而这些用户数据都会被导入到 `ES` 中，那怎么实现各个租户的用户数据索引隔离呢？

换言之，在同一个索引结构的情况下怎么实现一个租户一个索引？

解决方案：使用 `SpEL` 表达式动态获取索引。

### 实现 ###

#### 动态获取索引类 ####

**DynamicIndex.java**

package cn.xeblog.userprovider.es;
    
    import cn.hutool.core.util.StrUtil;
    import org.springframework.stereotype.Component;
    
    /**
     * 动态索引
     *
     * @author anlingyi
     * @date 2022/2/19 6:52 下午
     */
    @Component
    public class DynamicIndex {
    
        private static final ThreadLocal<String> THREAD_LOCAL = new ThreadLocal<>();
    
        /**
         * 获取索引名称后缀
         *
         * @return
         */
        public String getSuffix() {
            return THREAD_LOCAL.get();
        }
    
        /**
         * 设置索引名称后缀
         *
         * @param suffix
         */
        public void setSuffix(String suffix) {
            THREAD_LOCAL.set(suffix);
        }
    
        /**
         * 移除当前索引
         */
        public void remove() {
            THREAD_LOCAL.remove();
        }
    
        /**
         * 获取当前索引
         *
         * @return
         */
        public String getIndex() {
            if (StrUtil.isBlank(getSuffix())) {
                return null;
            }
    
            return "user_" + getSuffix();
        }
    
    }

原理：一般在请求后台接口的时候，我们会根据前端传过来的 `Token` ，解析出当前的用户信息，然后放置在当前请求线程的 `ThreadLocal` 中，当调用 `getIndex()` 方法时，会从当前线程的 `ThreadLocal` 中获取出用户的编号（索引后缀），然后拼接为一个完整的索引返回。

我这里为了方便测试，提供了 `setSuffix()、remove()` 等方法，用于手动设置或移除当前索引后缀。

#### 索引数据模型 ####

**EsUserInfo.java**

package cn.xeblog.userprovider.es.model;
    
    import lombok.Data;
    import org.springframework.data.annotation.Id;
    import org.springframework.data.elasticsearch.annotations.Document;
    
    /**
     * 用户信息
     *
     * @author anlingyi
     * @date 2022/2/19 6:47 下午
     */
    @Data
    @Document(indexName = "#{@dynamicIndex.getIndex()}", type = "_doc", createIndex = false)
    public class EsUserInfo {
    
        @Id
        private Long id;
    
        /**
         * 用户名
         */
        private String username;
    
        /**
         * 性别
         */
        private String gender;
    
        /**
         * 年龄
         */
        private Integer age;
    
    }

将`indexName` 设置为 `#{@dynamicIndex.getIndex()}` ，这是一个 `SpEL` 表达式，`dynamicIndex` 就是我们上面创建的动态获取索引类的对象，当需要获取索引名称的时候，`getIndex()` 方法就会被调用。

`createIndex` 一定要设置为 `false`，避免当项目启动时索引被自动创建。

#### ES存储库实现 ####

**EsUserInfoRepository.java**

无需定义任何方法

package cn.xeblog.userprovider.es;
    
    import cn.xeblog.userprovider.es.model.EsUserInfo;
    import org.springframework.data.elasticsearch.repository.ElasticsearchRepository;
    
    /**
     * @author anlingyi
     * @date 2022/2/19 6:55 下午
     */
    public interface EsUserInfoRepository extends ElasticsearchRepository<EsUserInfo, Long> {
    
    }

#### 测试 ####

package cn.xeblog.userprovider.es;
    
    import cn.xeblog.userprovider.es.model.EsUserInfo;
    import org.junit.jupiter.api.Test;
    import org.springframework.boot.test.context.SpringBootTest;
    
    import javax.annotation.Resource;
    
    import static org.junit.jupiter.api.Assertions.*;
    
    /**
     * @author anlingyi
     * @date 2022/2/19 6:57 下午
     */
    @SpringBootTest
    class EsUserInfoRepositoryTest {
    
        @Resource
        private EsUserInfoRepository esUserInfoRepository;
        @Resource
        private DynamicIndex dynamicIndex;
    
        @Test
        public void addUserInfo() {
            EsUserInfo userInfo = new EsUserInfo();
            userInfo.setId(1L);
            userInfo.setUsername("张三");
            userInfo.setGender("男");
            userInfo.setAge(18);
            // 索引后缀为当前租户ID：10001
            dynamicIndex.setSuffix("10001");
            // 为租户10001添加用户
            esUserInfoRepository.save(userInfo);
            // 移除后缀
            dynamicIndex.remove();
    
            EsUserInfo userInfo2 = new EsUserInfo();
            userInfo2.setId(2L);
            userInfo2.setUsername("李四");
            userInfo2.setGender("男");
            userInfo2.setAge(21);
            // 索引后缀为当前租户ID：10002
            dynamicIndex.setSuffix("10002");
            // 为租户10002添加用户
            esUserInfoRepository.save(userInfo2);
            // 移除后缀
            dynamicIndex.remove();
        }
    
    }

我这里分别为 `租户10001` 和 `租户10002` 各创建了一个用户。

![2e8d2b7361774a41942fa255ee91df36.png][]

### 注意 ###

除了 `createIndex` 一定要设置为 `false` 之外，还有一个需要特别注意的地方：

**`DynamicIndex` 的 `getIndex()` 方法在获取不到当前的索引后缀的情况下，一定要返回null ！！！**

/**
         * 获取当前索引
         *
         * @return
         */
        public String getIndex() {
            if (StrUtil.isBlank(getSuffix())) {
    	    // 一定要返回null
                return null;
            }
    
            return "user_" + getSuffix();
        }

为什么呢？

浅看一下 `ElasticsearchRepository.java` 源码你就懂了。

`AbstractElasticsearchRepository.java` 是 `ElasticsearchRepository.java` 的具体实现类，我们看一下这个类的 `save()` 方法的实现代码

@Override
    	public <S extends T> S save(S entity) {
    
    		Assert.notNull(entity, "Cannot save 'null' entity.");
    
    		elasticsearchOperations.index(createIndexQuery(entity));
    		elasticsearchOperations.refresh(entityInformation.getIndexName());
    
    		return entity;
    	}

当执行到 `elasticsearchOperations.refresh(entityInformation.getIndexName());` 这行代码时，获取到的索引后缀可能为空。

原因在于 `entityInformation.getIndexName()`

**MappingElasticsearchEntityInformation.java**

@Override
    	public String getIndexName() {
    		return indexName != null ? indexName : entityMetadata.getIndexName();
    	}

在项目启动时，`SpringDataElasticsearch` 会去解析一次 `@Document` 注解获取出索引名称，并将索引名称保存到 `MappingElasticsearchEntityInformation.java` 类的 `indexName` 字段中，后续调用 `entityInformation.getIndexName()` 时，`indexName` 字段值不为 `null` 时会直接返回，不会再去解析 `@Document` 注解。

这样就存在一个问题，当项目启动的时候 `getSuffix()` 返回的肯定是 `null`，如果在 `getIndex()` 方法中去掉判空代码，第一次调用时，返回的索引名称肯定会是 `user_null`，这样就会出现索引不存在的问题。

[2e8d2b7361774a41942fa255ee91df36.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/05/06/b70756f7a3414108ac5a96e1239ba54f.png