diff --git a/src/main/java/me/zhyd/oauth/model/AuthCallback.java b/src/main/java/me/zhyd/oauth/model/AuthCallback.java index bce57c8..b6558e2 100644 --- a/src/main/java/me/zhyd/oauth/model/AuthCallback.java +++ b/src/main/java/me/zhyd/oauth/model/AuthCallback.java @@ -5,6 +5,7 @@ import lombok.Builder; import lombok.Getter; import lombok.NoArgsConstructor; import lombok.Setter; +import me.zhyd.oauth.utils.StringUtils; import java.io.Serializable; @@ -68,4 +69,8 @@ public class AuthCallback implements Serializable { * @see error response */ private String error; + + public String getCode() { + return StringUtils.isEmpty(code) ? auth_code : code; + } }