Gin How to rewrite the response body and empty the original content in the middleware?

I want to rewrite response body in middleware,Only want to output replace data,Refer to How to rewrite response body in middleware?, modify my code as Requests are routed to

    ToolsGroup := Router.Group("")
    ToolsGroup .Use(middleware.ToolsGroupPermission())
    {
        ToolsGroup .GET("/ptr", func(c *gin.Context) {
            c.Data(http.StatusOK, "text/plain", []byte("origin data"))
        })
    }

Middleware is

package middleware

import (
    "bytes"
    "github.com/gin-gonic/gin"
)

func ToolsGroupPermission() gin.HandlerFunc {
    return func(c *gin.Context) {
        wb := &toolBodyWriter{
            body:           &bytes.Buffer{},
            ResponseWriter: c.Writer,
        }
        c.Writer = wb

        c.Next()

        wb.body.Reset()
        wb.Write([]byte("replace data"))
        //c.Data(http.StatusOK, "text/plain", []byte("replace data"))
    }
}

type toolBodyWriter struct {
    gin.ResponseWriter
    body *bytes.Buffer
}

func (r toolBodyWriter) Write(b []byte) (int, error) {
    return r.body.Write(b)
}

When the above code is modified, all [] bytes cannot be output. But the reference is changed to how do i get response body in after router middleware? when commenting:

func (r toolBodyWriter) Write(b []byte) (int, error) {
    r.body.Write(b)
    return r.ResponseWriter.Write(b)
}

Will output origin data+replace data at the same time. But the requirement is to output replace data

Comment From: kuustudio

The following two are the solutions I wrote, I hope you can provide better suggestions

package middleware

import (
    "bytes"
    "github.com/gin-gonic/gin"
)

func ToolsGroupPermission() gin.HandlerFunc {
    return func(c *gin.Context) {
        wb := &toolBodyWriter{
            body:           &bytes.Buffer{},
            ResponseWriter: c.Writer,
        }
        c.Writer = wb

        c.Next()

        originBytes := wb.body
        fmt.Printf("%s", originBytes)
        // clear Origin Buffer
        wb.body = &bytes.Buffer{}
        wb.Write([]byte("replace data"))
        wb.ResponseWriter.Write(wb.body.Bytes())
        //c.Data(http.StatusOK, "text/plain", []byte("replace data"))
    }
}

type toolBodyWriter struct {
    gin.ResponseWriter
    body *bytes.Buffer
}

func (r toolBodyWriter) Write(b []byte) (int, error) {
    return r.body.Write(b)
}

Or

package middleware

import (
    "bytes"
    "github.com/gin-gonic/gin"
)

func ToolsGroupPermission() gin.HandlerFunc {
    return func(c *gin.Context) {
        wb := &toolBodyWriter{
            body:           &bytes.Buffer{},
            ResponseWriter: c.Writer,
            status:         Origin,
        }
        c.Writer = wb

        c.Next()
        wb.status = Replace
        originBytes := wb.body
        fmt.Printf("%s", originBytes)
        // clear Origin Buffer
        wb.body = &bytes.Buffer{}
        wb.Write([]byte("replace data"))
        //c.Data(http.StatusOK, "text/plain", []byte("replace data"))
    }
}

type toolBodyWriter struct {
    gin.ResponseWriter
    body *bytes.Buffer
    status byte
}
const (
    Origin  byte = 0x0
    Replace      = 0x1
)

func (r toolBodyWriter) Write(b []byte) (int, error) {
    if r.status == 0x1 {
        r.body.Write(b)
        return r.ResponseWriter.Write(b)
    } else {
        return r.body.Write(b) //r.ResponseWriter.Write(b)
    }
}