从 DuplicateKeyException 中检索实际复制的特定 MongoDB 键（Java/Spring）

Question

我正在使用 Spring-Data/MongoDB 并在保存/插入时正确捕获重复键。

例如，假设我有一个用户被保存到一个集合中。 User 对象用两个@Indexed(unique=true)（两个唯一键）注释。假设它们是“电子邮件”和“用户名”。如何检索在插入过程中实际复制的索引。

我得到的最接近的是当我执行这种类型的示例代码时：

public boolean createNewUser() {
    MongoTemplate operations = RepositoryFactory.getMongoOperationsInstance();
    try {
        log.debug("Saving new user to DB");
        operations.save(this);
        return true;
    } catch (DuplicateKeyException dke) {
        log.debug("User with same username or email found");    
        log.debug(operations.getDb().getLastError());
        return false;
    }
}

这会打印字符串：

{ "serverUsed" : "/127.0.0.1:27017" , "err" : "E11000 duplicate key error index: Collection.user.$username  dup key: { : \"user\" }" , "code" : 11000 , "n" : 0 , "connectionId" : 17 , "ok" : 1.0}

没有愚蠢的字符串操作或 Json 转换，有没有办法通过 Mongodriver API 提取Collection.user.$username？

我一直在搜索不成功。

Answer 1

并非如此，因为 Mongo Java 驱动程序已经将最后一个错误公开为构造字符串：

writeResult.getLastError().get("err") 返回如下内容：

insertDocument :: caused by :: 11000 E11000 duplicate key error index: test.person.$username dup key: { : "joe" }

我想这对于 shell 和每个驱动程序也是如此。

我认为一个合理的解决方案是使用自定义异常来解析这样的重复键异常：

public class DetailedDuplicateKeyException extends DuplicateKeyException {
    public DetailedDuplicateKeyException(String msg) {
        // Instead of just calling super parse the message here.
        super(msg);
    }
}

...自定义异常翻译器：

public class DetailedDuplicateKeyExceptionTransaltor extends MongoExceptionTranslator {

    @Override
    public DataAccessException translateExceptionIfPossible(RuntimeException ex) {
        if (ex instanceof MongoException.DuplicateKey) {
            return new DetailedDuplicateKeyException(ex.getMessage());
        }
        return super.translateExceptionIfPossible(ex);
    }
}

...并正确设置 Spring 配置：

@Bean
public MongoFactoryBean mongo() {
    MongoFactoryBean mongo = new MongoFactoryBean();
    mongo.setExceptionTranslator(new DetailedDuplicateKeyExceptionTransaltor());
    mongo.setHost("localhost");
    return mongo;
}

编辑

检查MongoTemplate 代码（1.4.1.RELEASE）后，似乎在内部使用SimpleMongoDbFactory 来检索默认MongoExceptionTranslator，因此使用MongoFactoryBean 创建的那个被隐藏了。错过了那部分。

解决方案是覆盖SimpleMongoDbFactory（忘记MongoFactoryBean，在这种情况下没用）：

public class MySimpleMongoDbFactory extends SimpleMongoDbFactory {

    PersistenceExceptionTranslator translator = new       
            DetailedDuplicateKeyExceptionTransaltor();

    public MySimpleMongoDbFactory(Mongo mongo, String databaseName) {
        super(mongo, databaseName);
    }

    @Override
    public PersistenceExceptionTranslator getExceptionTranslator() {
        return translator;
    }
}

现在您可以使用自定义MongoDbFactory 构建模板：

template = new MongoTemplate (new MySimpleMongoDbFactory(new MongoClient(), "test"));

试过了，这个对我有用。

Answer 2

如果您在使用spring-data-rest/spring-data-mongodb 时遇到此问题，我编写了一个@ControllerAdvice 类，它使用@ExceptionHandler 方法以与验证类相同的方式返回错误。

我似乎没有在接受的答案中使用类，这就是我发布这个的原因。

我愿意接受有关解决此问题的更好方法的建议（在 Spring Data 中）/实施此@ExceptionHandler。

@ControllerAdvice
public class ControllerExceptionHandler {

  @ExceptionHandler(DuplicateKeyException.class)
  @ResponseStatus(value = HttpStatus.CONFLICT)
  @ResponseBody
  public Map<String, Object> handleDuplicateKeyException(DuplicateKeyException e) {
    String entity = null;
    String message = null;
    String invalidValue = null;
    String property = null;

    String errorMessage = e.getMessage();

    Pattern pattern = Pattern.compile("\\.(.*?) index: (.*?) dup key: \\{ : \\\\\"(.*?)\\\\\"");
    Matcher matcher = pattern.matcher(errorMessage);
    if (matcher.find()) {
      entity = WordUtils.capitalize(matcher.group(1));
      property = matcher.group(2);
      invalidValue = matcher.group(3);
    }

    message = WordUtils.capitalize(property) + " must be unique";

    Map<String, String> uniqueIndexViolation = new HashMap<>();
    uniqueIndexViolation.put("entity", entity);
    uniqueIndexViolation.put("message", message);
    uniqueIndexViolation.put("invalidValue", invalidValue);
    uniqueIndexViolation.put("property", property);

    List<Object> errors = new ArrayList<Object>();
    errors.add(uniqueIndexViolation);

    Map<String, Object> responseBody = new HashMap<>();
    responseBody.put("errors", errors);

    return responseBody;
  }

}