写代码其实有很多小技巧,熟练运用可以写出更加优雅健壮的代码,下面就介绍几个。
一、表单验证
controller
层接收请求参数,对有些参数我们需要进行校验,比如校验某个字段的格式、校验某个字段不能为空等。常规做法是写if
判断,当字段校验不通过就抛出异常。不过使用表单验证,可以让代码更加优雅。下面是伪代码:
@Data
public class UserDto {
@NotBlank(message = "name不能为空")
private String name;
@NotNull(message = "age不能为空")
@Min(value = 1, message = "年龄不能小于1")
private Integer age;
}
直接在接收参数的对象字段上加上相关注解,然后在controller
层中获取校验结果:
@PostMapping("/register")
public String register(@Valid UserDto userDto, BindingResult bindingResult){
// 校验参数
checkParams(bindingResult);
// 业务逻辑
return "";
}
private void checkParams(BindingResult bindingResult) {
if (bindingResult.hasErrors()){
List<ObjectError> allErrors = bindingResult.getAllErrors();
String errorMsg = allErrors.stream().map(ObjectError::getDefaultMessage).collect(Collectors.joining(";"));
if (StringUtils.isNotBlank(errorMsg)){
throw new BusinessException(BusinessExceptionEnum.bad_request_param.getCode(), errorMsg);
}
}
}
附上常用的校验注解:
注解 | 说明 |
---|---|
@Null | 只能为null |
@NotNull | 不能为null |
@AssertFalse | 只能为false |
@AssertTrue | 只能为true |
@DecimalMax(value) | 限制最大值,value是字符串形式 |
@DecimalMin(value) | 限制最小值,value是字符串形式 |
@Digits(x, y) | 只能是小数,且整数部分不能超过x位,小数部分不能超过y位 |
@Future | 必须是大于当前时间的日期 |
@Max(value) | 限制最大值,value是整数 |
@Min(value) | 限制最小值,value为整数 |
@Past | 必须是小于当前时间的日期 |
@Pattern(value) | value是正则表达式 |
@Size(max,min) | 限制字符长度必须在min到max之间 |
@NotEmpty | 字符串长度不为0,集合元素个数不为0 |
@NotBlank | 不为null,不为空白字符 |
必须符合邮箱格式 |
二、lombok的一些其他玩法
lombok
最常见的用法可能就是@Setter
、@Getter
和@Data
,其实他还有很多强大的地方。
1. @Accessors(chain = true)
我们经常在工具包或者源码中看到下面这种写法:
User user = new User().setAge(1).setName("aa");
这就是链式编程,如何实现链式编程,其实只需要在对象的set
方法中return this
就可以了,比如:
public class User {
private String name;
private int age;
public String getName() {
return name;
}
public User setName(String name) {
this.name = name;
return this;
}
public int getAge() {
return age;
}
public User setAge(int age) {
this.age = age;
return this;
}
}
如果使用lombok
,只需要在类上加上注解@Accessors(chain = true)
就搞定了,User
类可以简化这样:
@Setter
@Getter
@Accessors(chain = true)
public class User {
private String name;
private int age;
}
2. @RequiredArgsConstructor(staticName = "xxx")
和@NonNull
在创建对象的时候,可能有某个字段是必传的,那么可以将这个字段放到构造方法中,只有传入该字段才能创建对象。在类上加上@RequiredArgsConstructor(staticName = "xxx")
,xxx
表示方法名,然后在不能为空的字段上加上@NonNull
。
@Setter
@Getter
@Accessors(chain = true)
@RequiredArgsConstructor(staticName = "of")
public class User {
@NonNull
private String name;
private int age;
}
那么创建对象就可以这样:
User user = User.of("tom").setAge(6);
3. @Builder
我们也常常看到这样的代码:
HttpClientBuilder.create().setConnectionTimeToLive(100, TimeUnit.SECONDS).disableAuthCaching().build();
这其实是建造者模式的一种,如果要自己实现,可以像下面这样:
@Setter
@Getter
public class User {
private String name;
private int age;
public static Builder builder(){
return new Builder();
}
public static class Builder{
private String name;
private int age;
public Builder name(String name){
this.name = name;
return this;
}
public Builder age(int age){
this.age = age;
return this;
}
public User build(){
User user = new User();
user.setName(name);
user.setAge(age);
return user;
}
}
}
创建实例的时候就可以这样:
User user = User.builder().name("tom").age(6).build();
这样其实挺麻烦的,如果使用lombok
,只需要在User
类加上@Builder
注解即可。
4. @Cleanup
从java7
开始,我们可以使用try with resource
来确保资源得到释放而不用写finally
,例如:
public static void main(String[] args) throws Exception {
try (InputStream inputStream = new FileInputStream("test.txt");
OutputStream outputStream = new FileOutputStream(args[1])) {
byte[] b = new byte[10000];
while (true){
int r = inputStream.read();
if (r == -1){
break;
}
outputStream.write(b, 0, r);
}
}
}
这种写法也挺简单了,不过lombok
提供了@Cleanup
注解,加上它,连try
都不用写了,用法如下:
public static void main(String[] args) throws Exception {
@Cleanup InputStream inputStream = new FileInputStream("test.txt");
@Cleanup OutputStream outputStream = new FileOutputStream(args[1]);
byte[] b = new byte[10000];
while (true) {
int r = inputStream.read();
if (r == -1) {
break;
}
outputStream.write(b, 0, r);
}
}