From 42f7c90f9f40ca4557e0864c468517719e133243 Mon Sep 17 00:00:00 2001
From: Tang Cheng <cheng.tang@supwisdom.com>
Date: Thu, 12 Mar 2020 16:28:30 +0800
Subject: [PATCH] =?utf8?q?chore:=20=E5=A2=9E=E5=8A=A0=E9=94=99=E8=AF=AF?=
 =?utf8?q?=E8=BF=94=E5=9B=9E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 .../dlpay/framework/framework_util.kt         | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/payapi/src/main/kotlin/com/supwisdom/dlpay/framework/framework_util.kt b/payapi/src/main/kotlin/com/supwisdom/dlpay/framework/framework_util.kt
index 8e6deb5c..a4d76190 100644
--- a/payapi/src/main/kotlin/com/supwisdom/dlpay/framework/framework_util.kt
+++ b/payapi/src/main/kotlin/com/supwisdom/dlpay/framework/framework_util.kt
@@ -1,6 +1,7 @@
 package com.supwisdom.dlpay.framework
 
 import com.google.gson.Gson
+import com.nimbusds.oauth2.sdk.ErrorResponse
 import com.supwisdom.dlpay.api.APIRequestParam
 import com.supwisdom.dlpay.api.bean.ApiResponse
 import com.supwisdom.dlpay.api.exception.RequestParamCheckException
@@ -27,7 +28,7 @@ class ResponseBodyBuilder private constructor() {
             return ResponseEntity.ok(create().fail(bean, code, msg))
         }
 
-        fun <T> fail(status: HttpStatus, bean: T): ResponseEntity<*> {
+        fun <T> fail(status: HttpStatus, bean: T): ResponseEntity<T> {
             return ResponseEntity.status(status).body(bean)
         }
 
@@ -43,14 +44,26 @@ class ResponseBodyBuilder private constructor() {
             return fail(HttpStatus.NOT_FOUND, msg)
         }
 
+        fun notFound(error: ErrorResponse) : ResponseEntity<ErrorResponse> {
+            return fail(HttpStatus.NOT_FOUND, error)
+        }
+
         fun badRequest(msg: String): ResponseEntity<*> {
             return fail(HttpStatus.BAD_REQUEST, msg)
         }
 
+        fun badRequest(error: ErrorResponse): ResponseEntity<*> {
+            return fail(HttpStatus.BAD_REQUEST, error)
+        }
+
         fun conflict(msg: String): ResponseEntity<*> {
             return fail(HttpStatus.CONFLICT, msg)
         }
 
+        fun conflict(error: ErrorResponse): ResponseEntity<ErrorResponse> {
+            return fail(HttpStatus.CONFLICT, error)
+        }
+
         fun forbbiden(msg: String): ResponseEntity<*> {
             return fail(HttpStatus.FORBIDDEN, msg)
         }
@@ -62,6 +75,10 @@ class ResponseBodyBuilder private constructor() {
         fun internalServerError(msg: String): ResponseEntity<*> {
             return fail(HttpStatus.INTERNAL_SERVER_ERROR, msg)
         }
+
+        fun internalServerError(error: ErrorResponse): ResponseEntity<ErrorResponse> {
+            return fail(HttpStatus.INTERNAL_SERVER_ERROR, error)
+        }
     }
 
     private var retCode = INVALIDE_RETCODE
-- 
2.17.1