8
8
9
9
import javax .servlet .http .HttpServletRequest ;
10
10
import java .lang .reflect .Field ;
11
+ import java .util .List ;
11
12
12
13
/**
13
14
* @author pantao
@@ -21,8 +22,6 @@ public class BaseController {
21
22
22
23
private boolean checkSensitiveData = false ;
23
24
24
- private String sensitiveDataTip = "******" ;
25
-
26
25
public BaseController () {}
27
26
28
27
public BaseController (boolean checkSensitiveData ) {
@@ -69,25 +68,40 @@ protected ResultObject parseResult(String okMsg, String errMsg, boolean isOk) {
69
68
return isOk ? new ResultObject (okMsg ) : CheckResult .getErrorResult (errMsg );
70
69
}
71
70
71
+ protected <T > ResultObject <List <T >> parseResult (String errMsg , List <T > list ) {
72
+ if (Checker .isEmpty (list )) {
73
+ return CheckResult .getErrorResult (errMsg );
74
+ }
75
+ if (checkSensitiveData ) {
76
+ list .forEach (this ::setSensitiveData );
77
+ }
78
+ return new ResultObject <>(list );
79
+ }
80
+
72
81
protected <T > ResultObject <T > parseResult (String okMsg , String errMsg , T t ) {
73
82
if (Checker .isNull (t )) {
74
83
return CheckResult .getErrorResult (errMsg );
75
84
}
76
85
if (checkSensitiveData ) {
77
- Field [] fields = t .getClass ().getDeclaredFields ();
78
- try {
79
- for (Field field : fields ) {
80
- SensitiveData sensitiveData = field .getAnnotation (SensitiveData .class );
81
- if (Checker .isNotNull (sensitiveData ) && field .getType () == String .class ) {
82
- field .setAccessible (true );
83
- field .set (t , sensitiveDataTip );
84
- }
86
+ setSensitiveData (t );
87
+ }
88
+ return new ResultObject <>(okMsg , t );
89
+ }
90
+
91
+ private <T > void setSensitiveData (T t ) {
92
+ Field [] fields = t .getClass ().getDeclaredFields ();
93
+ String sensitiveDataTip = "******" ;
94
+ try {
95
+ for (Field field : fields ) {
96
+ SensitiveData sensitiveData = field .getAnnotation (SensitiveData .class );
97
+ if (Checker .isNotNull (sensitiveData ) && field .getType () == String .class ) {
98
+ field .setAccessible (true );
99
+ field .set (t , sensitiveDataTip );
85
100
}
86
- } catch (IllegalAccessException e ) {
87
- LoggerUtils .error ("set sensitive data error: {}" , e .getMessage ());
88
101
}
102
+ } catch (IllegalAccessException e ) {
103
+ LoggerUtils .error ("set sensitive data error: {}" , e .getMessage ());
89
104
}
90
- return new ResultObject <>(okMsg , t );
91
105
}
92
106
93
107
protected <T > ResultObject <T > parseResult (String errMsg , T t ) {
0 commit comments