当MongoDB主键为String时,mongoTemplate无法根据id查询的问题

MongoDB推荐使用ObjectId作为主键,但国内的开发都知道,事情往往不如人所愿,当我们真的出现了"_id"主键的类型为String时,且还必须想用mongoTemplate.findOne或findList时,直接使用该方法会导致查询结果为空。

因为mongoTemplate会在查询的时候将主键转换为ObjectId。

实际上这一步转换是MappingMongoConverter中做的,源码如下:

java 复制代码
	/**
	 * Converts the given raw id value into either {@link ObjectId} or {@link String}.
	 *
	 * @param id can be {@literal null}.
	 * @param targetType must not be {@literal null}.
	 * @return {@literal null} if source {@literal id} is already {@literal null}.
	 * @since 2.2
	 */
@Nullable
	default Object convertId(@Nullable Object id, Class<?> targetType) {

		if (id == null || ClassUtils.isAssignableValue(targetType, id)) {
			return id;
		}

		if (ClassUtils.isAssignable(ObjectId.class, targetType)) {

			if (id instanceof String) {

				if (ObjectId.isValid(id.toString())) {
					return new ObjectId(id.toString());
				}

				// avoid ConversionException as convertToMongoType will return String anyways.
				return id;
			}
		}

		try {
			return getConversionService().canConvert(id.getClass(), targetType)
					? getConversionService().convert(id, targetType)
					: convertToMongoType(id, (TypeInformation<?>) null);
		} catch (ConversionException o_O) {
			return convertToMongoType(id,(TypeInformation<?>)  null);
		}
	}

注意示例版本为spring-data-mongodb:3.3.5,一些更古早的版本,如2.1.9,需要重写其他部分,如果有人需要请留言,我会抽时间补充,但鉴于之前的版本有比较重大的漏洞,所以推荐升级版本。

if (ObjectId.isValid(id.toString())) { return new ObjectId(id.toString()); }

这段就是mongoTemplate将String转换为ObjectId的步骤。

所以我们要做的就是重写这部分,并且替换掉原有的:

1、新建一个Converter

java 复制代码
public class MyMongoConverter extends MappingMongoConverter {
    public MyMongoConverter(DbRefResolver dbRefResolver, MappingContext<? extends MongoPersistentEntity<?>, MongoPersistentProperty> mappingContext) {
        super(dbRefResolver, mappingContext);
    }

    @Override
    public Object convertId(@Nullable Object id, Class<?> targetType) {
        if (id == null || ClassUtils.isAssignableValue(targetType, id)) {
            return id;
        }

        if (ClassUtils.isAssignable(ObjectId.class, targetType)) {
            if (id instanceof String) {
                return id;
            }
        }

        try {
            return getConversionService().canConvert(id.getClass(), targetType)
                    ? getConversionService().convert(id, targetType)
                    : convertToMongoType(id, (TypeInformation<?>) null);
        } catch (ConversionException o_O) {
            return convertToMongoType(id,(TypeInformation<?>)  null);
        }
    }
}

2、在构建MongoTemplate时用我们自己写的替换

java 复制代码
MongoTemolate mongoTemplate = new MongoTemplate(mongoDbFactory(), getDefaultMongoConverter(mongoDbFactory()))

protected static MongoConverter getDefaultMongoConverter(MongoDatabaseFactory factory) {

        DbRefResolver dbRefResolver = new DefaultDbRefResolver(factory);
        MongoCustomConversions conversions = new MongoCustomConversions(Collections.emptyList());

        MongoMappingContext mappingContext = new MongoMappingContext();
        mappingContext.setSimpleTypeHolder(conversions.getSimpleTypeHolder());
        mappingContext.afterPropertiesSet();

        MyMongoConverter converter = new MyMongoConverter(dbRefResolver, mappingContext);

        converter.setCustomConversions(conversions);
        converter.afterPropertiesSet();
        converter.setTypeMapper(new DefaultMongoTypeMapper(null));

        return converter;
    }
    /**
     * @return {@link MongoDatabaseFactory}
     */
    protected MongoDatabaseFactory mongoDbFactory() {
        ServerAddress serverAddress = new ServerAddress(host, port);

        MongoClientSettings.Builder setting = MongoClientSettings.builder()
                .applyToClusterSettings(builder -> builder.hosts(Collections.singletonList(serverAddress)))
                .retryWrites(false);
        if (ObjectUtil.isNotEmpty(username)) {
            MongoCredential credential = MongoCredential.createCredential(username, authenticationDatabase, password.toCharArray());
            setting.credential(credential);
        }
        MongoClient mongoClient = MongoClients.create(setting.build());
        return new SimpleMongoClientDatabaseFactory(mongoClient, database);
    }

host,port,username,authenticationDatabase,password,database

分别为MongoDB的地址,端口号,用户名,认证数据库,密码,数据库名

相关推荐
爬山算法8 分钟前
Redis(69)Redis分布式锁的优点和缺点是什么?
数据库·redis·分布式
RestCloud12 分钟前
从数据库到价值:ETL 工具如何打通南大通用数据库与企业应用
数据库
惜月_treasure1 小时前
Text2SQL与工作流实现:让数据库查询变得轻松又高效
数据库·人工智能·python
-睡到自然醒~1 小时前
[go 面试] 并发与数据一致性:事务的保障
数据库·面试·golang
为乐ovo1 小时前
19.DCL-用户管理
数据库
一个天蝎座 白勺 程序猿1 小时前
金仓数据库KingbaseES实现MongoDB平滑迁移全攻略:从架构适配到性能调优的完整实践
数据库·mongodb·数据迁移·kingbasees·金仓数据库
武子康1 小时前
Java-153 深入浅出 MongoDB 全面的适用场景分析与选型指南 场景应用指南
java·开发语言·数据库·mongodb·性能优化·系统架构·nosql
2401_837088501 小时前
Redis通用命令
数据库·redis·缓存
程序边界1 小时前
MongoDB迁移到KES实战全纪录(上):迁移准备与实施指南
数据库·mongodb
weixin_421133412 小时前
django xadmin 结合 minio
数据库·django·sqlite