Use the same interface to realize different login methods

1. Define parent class DTO

package com.zzx.dto;


import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonSubTypes;
import com.fasterxml.jackson.annotation.JsonTypeInfo;
import lombok.Data;

@Data
@JsonTypeInfo(
    use = JsonTypeInfo.Id.NAME,
    include = JsonTypeInfo.As.EXISTING_PROPERTY,
    property = "productCode",
    visible = true)
@JsonSubTypes({
    
    @JsonSubTypes.Type(value = UserPasswordDTO.class, name = LoginDTO.USERNAME_PASSWORD),
    @JsonSubTypes.Type(value = UserPhoneDTO.class,name = LoginDTO.PHONE_SMS_PRODUCT),
})
@JsonIgnoreProperties(ignoreUnknown = true)
public class LoginDTO {
    

    public static final String PHONE_SMS_PRODUCT = "phoneSms";

    public static final String USERNAME_PASSWORD = "password";


    private String productCode;

    public LoginDTO() {
    
    }
}

2. Define subclasses that require login types

   • Account password login

     package com.zzx.dto;
     
     import lombok.Data;
     
     @Data
     public class UserPasswordDTO extends LoginDTO {
             
     
         private String userName;
     
         private String password;
     
         @Override
         public String getProductCode() {
             
             return LoginDTO.USERNAME_PASSWORD;
         }
     }
     
     

   • Mobile number verification code login

     package com.zzx.dto;
     
     import lombok.AllArgsConstructor;
     import lombok.Data;
     import lombok.NoArgsConstructor;
     
     @Data
     public class UserPhoneDTO extends LoginDTO {
             
     
         /** *  Location code  */
         private String areaCode;
     
         /** *  cell-phone number  */
         private String phoneNum;
     
         /** *  Verification Code  */
         private String verifyCode;
     
         @Override
         public String getProductCode() {
             
             return LoginDTO.PHONE_SMS_PRODUCT;
         }
     }
     
     

3. test ( Be careful , Incoming productCode Must be with @JsonSubTypes.Type Defined in name The value is consistent. )

package com.zzx;

import java.util.HashMap;
import java.util.Map;

import com.alibaba.fastjson.JSON;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.zzx.dto.LoginDTO;
import com.zzx.dto.UserPasswordDTO;
import com.zzx.dto.UserPhoneDTO;

public class LoginTest {
    
    public static void main(String[] args) throws JsonProcessingException {
    

        Map<String, String> params = new HashMap<>();
        params.put("productCode", "password");
        params.put("userName", "userName");
        params.put("password", "password");
        params.put("phoneNum", null);
        params.put("verifyCode", null);
        ObjectMapper mapper = new ObjectMapper();
        String res = mapper.writerWithDefaultPrettyPrinter().writeValueAsString(params);

        LoginDTO loginDTO = mapper.readValue(res, LoginDTO.class);

        System.out.println("loginDTO : "+JSON.toJSONString(loginDTO));

        if (loginDTO instanceof UserPasswordDTO){
    
            System.out.println("userPassword : "+JSON.toJSONString(loginDTO));
        }

        if (loginDTO instanceof UserPhoneDTO){
    
            System.out.println("userPhone : "+JSON.toJSONString(loginDTO));
        }

    }
}