[Enhancement] Add Response Status (#209)

* [Enhancement] Add Response Status * Set the response code --------- Co-authored-by: Wenbo Han <hwbrzzl@gmail.com>
goravel · Jun 25, 2023 · ae90575 · ae90575
1 parent 3ce7337
commit ae90575
Show file tree

Hide file tree

Showing 6 changed files with 106 additions and 18 deletions.
diff --git a/contracts/http/mocks/Response.go b/contracts/http/mocks/Response.go
diff --git a/contracts/http/mocks/ResponseOrigin.go b/contracts/http/mocks/ResponseOrigin.go
diff --git a/contracts/http/mocks/ResponseStatus.go b/contracts/http/mocks/ResponseStatus.go
diff --git a/contracts/http/mocks/ResponseSuccess.go b/contracts/http/mocks/ResponseSuccess.go
diff --git a/contracts/http/response.go b/contracts/http/response.go
@@ -18,9 +18,17 @@ type Response interface {
 	Redirect(code int, location string)
 	String(code int, format string, values ...any)
 	Success() ResponseSuccess
+	Status(code int) ResponseStatus
 	Writer() http.ResponseWriter
 }
 
+//go:generate mockery --name=ResponseStatus
+type ResponseStatus interface {
+	Data(contentType string, data []byte)
+	Json(obj any)
+	String(format string, values ...any)
+}
+
 //go:generate mockery --name=ResponseSuccess
 type ResponseSuccess interface {
 	Data(contentType string, data []byte)

diff --git a/http/gin_response.go b/http/gin_response.go
@@ -56,6 +56,10 @@ func (r *GinResponse) Success() httpcontract.ResponseSuccess {
 	return NewGinSuccess(r.instance)
 }
 
+func (r *GinResponse) Status(code int) httpcontract.ResponseStatus {
+	return NewGinStatus(r.instance, code)
+}
+
 func (r *GinResponse) Writer() http.ResponseWriter {
 	return r.instance.Writer
 }
@@ -80,6 +84,27 @@ func (r *GinSuccess) String(format string, values ...any) {
 	r.instance.String(http.StatusOK, format, values...)
 }
 
+type GinStatus struct {
+	instance *gin.Context
+	status   int
+}
+
+func NewGinStatus(instance *gin.Context, code int) httpcontract.ResponseSuccess {
+	return &GinStatus{instance, code}
+}
+
+func (r *GinStatus) Data(contentType string, data []byte) {
+	r.instance.Data(r.status, contentType, data)
+}
+
+func (r *GinStatus) Json(obj any) {
+	r.instance.JSON(r.status, obj)
+}
+
+func (r *GinStatus) String(format string, values ...any) {
+	r.instance.String(r.status, format, values...)
+}
+
 func GinResponseMiddleware() httpcontract.Middleware {
 	return func(ctx httpcontract.Context) {
 		blw := &BodyWriter{body: bytes.NewBufferString("")}