確認需求
需求與前文類似，只不過這里需要是在 RequestBody 中使用。與前文不同的是，這種請求是通過 Http Body 的方式傳輸到后端，通常是 json 或 xml 格式，Spring 默認借助 Jackson 反序列化為對象。
同樣的，我們需要在枚舉中定義 int 類型的 id、String 類型的 code，id 取值不限于序號（即從 0 開始的 orinal 數據），code 不限于 name。客戶端請求過程中，可以傳 id，可以傳 code，也可以傳 name。服務端只需要在對象中定義一個枚舉參數，不需要額外的轉換，即可得到枚舉值。
好了，接下來我們定義一下枚舉對象。
定義枚舉和對象
先定義我們的枚舉類GenderIdCodeEnum，包含 id 和 code 兩個屬性：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
public enum GenderIdCodeEnum implements IdCodeBaseEnum { MALE(1, "male"), FEMALE(2, "female"); private final Integer id; private final String code; GenderIdCodeEnum(Integer id, String code) { this.id = id; this.code = code; } @Override public String getCode() { return code; } @Override public Integer getId() { return id; }} |
這個枚舉類的要求與前文一致，不清楚的可以再去看一下。
在定義一個包裝類GenderIdCodeRequestBody，用于接收 json 數據的請求體：
|
1
2
3
4
5
6
|
@Datapublic class GenderIdCodeRequestBody { private String name; private GenderIdCodeEnum gender; private long timestamp;} |
除了GenderIdCodeEnum參數外，其他都是示例，所以隨便定義一下。
實現轉換邏輯
前奏鋪墊好，接下來入正題了。Jackson 提供了兩種方案：
- 方案一：精準攻擊，指定需要轉換的字段，不影響其他類對象中的字段
- 方案二：全范圍攻擊，所有借助 Jackson 反序列化的枚舉字段，全部具備自動轉換功能
方案一：精準攻擊
這種方案中，我們首先需要實現JsonDeserialize抽象類：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
public class IdCodeToEnumDeserializer extends JsonDeserializer<BaseEnum> { @Override public BaseEnum deserialize(JsonParser jsonParser, DeserializationContext deserializationContext) throws IOException { final String param = jsonParser.getText();// 1 final JsonStreamContext parsingContext = jsonParser.getParsingContext();// 2 final String currentName = parsingContext.getCurrentName();// 3 final Object currentValue = parsingContext.getCurrentValue();// 4 try { final Field declaredField = currentValue.getClass().getDeclaredField(currentName);// 5 final Class<?> targetType = declaredField.getType();// 6 final Method createMethod = targetType.getDeclaredMethod("create", Object.class);// 7 return (BaseEnum) createMethod.invoke(null, param);// 8 } catch (NoSuchMethodException | InvocationTargetException | IllegalAccessException | NoSuchFieldException e) { throw new CodeBaseException(ErrorResponseEnum.PARAMS_ENUM_NOT_MATCH, new Object[] {param}, "", e); } }} |
然后在指定枚舉字段上定義@JsonDeserialize注解，比如：
|
1
2
|
@JsonDeserialize(using = IdCodeToEnumDeserializer.class)private GenderIdCodeEnum gender; |
具體說一下每行的作用：
- 獲取參數值。根據需要，此處可能是 id、code 或 name，也就是源值，需要將其轉換為枚舉；
- 獲取轉換上線文，這個是為 3、4 步做準備的；
- 獲取標記@JsonDeserialize注解的字段，此時currentName的值是gender；
- 獲取包裝對象，也就是GenderIdCodeRequestBody對象；
- 根據包裝對象的Class對象，以及字段名gender獲取Field對象，為第 5 步做準備；
- 獲取gender字段對應的枚舉類型，也即是GenderIdCodeEnum。之所以這樣做，是要實現一個通用的反序列化類；
- 這里是寫死的一種實現，就是在枚舉類中，需要定義一個靜態方法，方法名是create，請求參數是Object；
- 通過反射調用create方法，將第一步獲取的請求參數傳入。
我們來看一下枚舉類中定義的create方法：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
public static GenderIdCodeEnum create(Object code) { final String stringCode = code.toString(); final Integer intCode = BaseEnum.adapter(stringCode); for (GenderIdCodeEnum item : values()) { if (Objects.equals(stringCode, item.name())) { return item; } if (Objects.equals(item.getCode(), stringCode)) { return item; } if (Objects.equals(item.getId(), intCode)) { return item; } } return null;} |
為了性能考慮，我們可以提前定義三組 map，分別以 id、code、name 為 key，以枚舉值為 value，這樣就可以通過 O(1) 的時間復雜度返回了。可以參考前文的Converter類的實現邏輯。
這樣，我們就可以實現精準轉換了。
方案二：全范圍攻擊
這種方案是全范圍攻擊了，只要是 Jackson 參與的反序列化，只要其中有目標枚舉參數，就會受到這種進入這種方案的邏輯中。這種方案是在枚舉類中定義一個靜態轉換方法，通過@JsonCreator注解注釋，Jackson 就會自動轉換了。
這個方法的定義與方案一中的create方法完全一致，所以只需要在create方法上加上注解即可：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
@JsonCreator(mode = Mode.DELEGATING)public static GenderIdCodeEnum create(Object code) { final String stringCode = code.toString(); final Integer intCode = BaseEnum.adapter(stringCode); for (GenderIdCodeEnum item : values()) { if (Objects.equals(stringCode, item.name())) { return item; } if (Objects.equals(item.getCode(), stringCode)) { return item; } if (Objects.equals(item.getId(), intCode)) { return item; } } return null;} |
其中Mode類有四個值：DEFAULT、DELEGATING、PROPERTIES、DISABLED，這四種的差別會在原理篇中說明。還是那句話，對于應用類技術，我們可以先知其然，再知其所以然，也一定要知其所以然。
測試
先定義一個 controller 方法：
|
1
2
3
4
5
|
@PostMapping("gender-id-code-request-body")public GenderIdCodeRequestBody bodyGenderIdCode(@RequestBody GenderIdCodeRequestBody genderRequest) { genderRequest.setTimestamp(System.currentTimeMillis()); return genderRequest;} |
然后定義測試用例，還是借助 JUnit5：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
@ParameterizedTest@ValueSource(strings = {"\"MALE\"", "\"male\"", "\"1\"", "1"})void postGenderIdCode(String gender) throws Exception { final String result = mockMvc.perform( MockMvcRequestBuilders.post("/echo/gender-id-code-request-body") .contentType(MediaType.APPLICATION_JSON_UTF8) .accept(MediaType.APPLICATION_JSON_UTF8) .content("{\"gender\": " + gender + ", \"name\": \"看山\"}") ) .andExpect(MockMvcResultMatchers.status().isOk()) .andDo(MockMvcResultHandlers.print()) .andReturn() .getResponse() .getContentAsString(); ObjectMapper objectMapper = new ObjectMapper(); final GenderIdCodeRequestBody genderRequest = objectMapper.readValue(result, GenderIdCodeRequestBody.class); Assertions.assertEquals(GenderIdCodeEnum.MALE, genderRequest.getGender()); Assertions.assertEquals("看山", genderRequest.getName()); Assertions.assertTrue(genderRequest.getTimestamp() > 0);} |
總結
本文主要說明了如何在 RequestBody 中優雅的使用枚舉參數，借助了 Jackson 的反序列化擴展，可以定制類型轉換邏輯。
本篇文章就到這里了，希望能給你帶來幫助，也希望您能夠多多關注服務器之家的更多內容！
原文鏈接：https://blog.csdn.net/liuxinghao/article/details/119881628
