修复同一个邮箱有多条验证码时查询报错问题

atjiu · May 24, 2019 · e1bdd49 · e1bdd49
1 parent 371d390
commit e1bdd49
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/co/yiiu/pybbs/service/CodeService.java b/src/main/java/co/yiiu/pybbs/service/CodeService.java
@@ -13,6 +13,7 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.util.Date;
+import java.util.List;
 
 /**
  * Created by tomoya.
@@ -59,7 +60,9 @@ public Code selectNotUsedCode(Integer userId, String email, String mobile) {
     }
     lambda.eq(Code::getUsed, false);
     lambda.gt(Code::getExpireTime, new Date());
-    return codeMapper.selectOne(wrapper);
+    List<Code> codes = codeMapper.selectList(wrapper);
+    if (codes.isEmpty()) return null;
+    return codes.get(0);
   }
 
   // 创建一条验证码记录