前言
最近为了工作也为了更加深入了解掌握java注解的使用,决定自定义注解来实现数据验证。
api开发中经常会遇到一些对请求数据进行验证的情况,这时候如果使用注解就有两个好处,一是验证逻辑和业务逻辑分离,代码清晰,二是验证逻辑可以轻松复用,只需要在要验证的地方加上注解就可以。
java提供了一些基本的验证注解,比如@notnull、@size,但是更多情况下需要自定义验证逻辑,这时候就可以自己实现一个验证注解,方法很简单,仅需要两个东西:
- 一个自定义的注解,并且指定验证器
- 一个验证器的实现
自定义验证注解
考虑有一个api,接收一个student对象,并希望对象里的age域的值是奇数,这时候就可以创建以下注解:
1
2
3
4
5
6
7
8
|
@target (elementtype.field) @retention (retentionpolicy.runtime) @constraint (validatedby = agevalidator. class ) public @interface odd { string message() default "age must be odd" ; class <?>[] groups() default {}; class <? extends payload>[] payload() default {}; } |
其中:
- @target指明这个注解要作用在什么地方,可以是对象、域、构造器等,因为要作用在age域上,因此这里选择field
- @retention指明了注解的生命周期,可以有source(仅保存在源码中,会被编译器丢弃),class(在class文件中可用,会被vm丢弃)以及runtime(在运行期也被保留),这里选择了生命周期最长的runtime
- @constraint是最关键的,它表示这个注解是一个验证注解,并且指定了一个实现验证逻辑的验证器
- message()指明了验证失败后返回的消息,此方法为@constraint要求
- groups()和payload()也为@constraint要求,可默认为空,详细用途可以查看@constraint文档
创建验证器
有了注解之后,就需要一个验证器来实现验证逻辑:
1
2
3
4
5
6
7
8
9
10
|
public class agevalidator implements constraintvalidator<odd,integer> { @override public void initialize(odd constraintannotation) { } @override public boolean isvalid(integer age, constraintvalidatorcontext constraintvalidatorcontext) { return age % 2 != 0 ; } } |
其中:
- 验证器有两个类型参数,第一个是所属的注解,第二个是注解作用地方的类型,这里因为作用在age上,因此这里用了integer
- initialize()可以在验证开始前调用注解里的方法,从而获取到一些注解里的参数,这里用不到
- isvalid()就是判断是否合法的地方
应用注解
注解和验证器创建好之后,就可以使用注解了:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
public class student { @odd private int age; private string name; public string getname() { return name; } public void setname(string name) { this .name = name; } public int getage() { return age; } public void setage( int age) { this .age = age; } } |
1
2
3
4
5
6
7
|
@restcontroller public class studentresource { @postmapping ( "/student" ) public string addstudent( @valid @requestbody student student) { return "student created" ; } } |
在需要启用验证的地方加上@valid注解,这时候如果请求里的student年龄不是奇数,就会得到一个400响应:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
{ "timestamp" : "2018-08-15t17:01:44.598+0000" , "status" : 400 , "error" : "bad request" , "errors" : [ { "codes" : [ "odd.student.age" , "odd.age" , "odd.int" , "odd" ], "arguments" : [ { "codes" : [ "student.age" , "age" ], "arguments" : null , "defaultmessage" : "age" , "code" : "age" } ], "defaultmessage" : "age must be odd" , "objectname" : "student" , "field" : "age" , "rejectedvalue" : 12 , "bindingfailure" : false , "code" : "odd" } ], "message" : "validation failed for object='student'. error count: 1" , "path" : "/student" } |
也可以手动来处理错误,加上一个bindingresult来接收验证结果即可:
1
2
3
4
5
6
7
8
9
10
|
@restcontroller public class studentresource { @postmapping ( "/student" ) public string addstudent( @valid @requestbody student student, bindingresult validateresult) { if (validateresult.haserrors()) { return validateresult.getallerrors().get( 0 ).getdefaultmessage(); } return "student created" ; } } |
这时候如果验证出错,便只会返回一个状态为200,内容为age must be odd的响应。
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对服务器之家的支持。
原文链接:https://www.cnblogs.com/xz816111/p/9484902.html