mirror of
https://gitee.com/johng/gf
synced 2026-06-06 02:25:47 +08:00
7ffdff37e4
This PR includes the following changes: - **Upgrade `.golangci.yml`**: Updated the configuration file to align with the latest golangci-lint version, ensuring compatibility and leveraging new features. - **Refactor GitHub Action workflow**: Modified `golangci-lint.yml` in the GitHub Actions workflow to reflect the updated configuration and improve CI performance. - **Codebase optimization**: Refactored code to address issues and warnings raised by the updated golangci-lint rules, including: - Improved function length and complexity. - Enhanced error handling and variable naming conventions. - Fixed minor issues such as unused imports and formatting inconsistencies. These changes aim to maintain code quality, ensure compatibility with the latest tools, and improve overall maintainability.
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
//
|
|
|
|
package response
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
)
|
|
|
|
// BufferWriter is the custom writer for http response with buffer.
|
|
type BufferWriter struct {
|
|
*Writer // The underlying BufferWriter.
|
|
Status int // HTTP status.
|
|
buffer *bytes.Buffer // The output buffer.
|
|
}
|
|
|
|
func NewBufferWriter(writer http.ResponseWriter) *BufferWriter {
|
|
return &BufferWriter{
|
|
Writer: NewWriter(writer),
|
|
buffer: bytes.NewBuffer(nil),
|
|
}
|
|
}
|
|
|
|
// RawWriter returns the underlying BufferWriter.
|
|
func (w *BufferWriter) RawWriter() http.ResponseWriter {
|
|
return w.Writer
|
|
}
|
|
|
|
// Write implements the interface function of http.BufferWriter.Write.
|
|
func (w *BufferWriter) Write(data []byte) (int, error) {
|
|
return w.buffer.Write(data)
|
|
}
|
|
|
|
// WriteString writes string content to internal buffer.
|
|
func (w *BufferWriter) WriteString(data string) (int, error) {
|
|
return w.buffer.WriteString(data)
|
|
}
|
|
|
|
// Buffer returns the buffered content as []byte.
|
|
func (w *BufferWriter) Buffer() []byte {
|
|
return w.buffer.Bytes()
|
|
}
|
|
|
|
// BufferString returns the buffered content as string.
|
|
func (w *BufferWriter) BufferString() string {
|
|
return w.buffer.String()
|
|
}
|
|
|
|
// BufferLength returns the length of the buffered content.
|
|
func (w *BufferWriter) BufferLength() int {
|
|
return w.buffer.Len()
|
|
}
|
|
|
|
// SetBuffer overwrites the buffer with `data`.
|
|
func (w *BufferWriter) SetBuffer(data []byte) {
|
|
w.buffer.Reset()
|
|
w.buffer.Write(data)
|
|
}
|
|
|
|
// ClearBuffer clears the response buffer.
|
|
func (w *BufferWriter) ClearBuffer() {
|
|
w.buffer.Reset()
|
|
}
|
|
|
|
// WriteHeader implements the interface of http.BufferWriter.WriteHeader.
|
|
func (w *BufferWriter) WriteHeader(status int) {
|
|
w.Status = status
|
|
}
|
|
|
|
// Flush outputs the buffer to clients and clears the buffer.
|
|
func (w *BufferWriter) Flush() {
|
|
if w.IsHijacked() {
|
|
return
|
|
}
|
|
|
|
if w.Status != 0 && !w.IsHeaderWrote() {
|
|
w.Writer.WriteHeader(w.Status)
|
|
}
|
|
// Default status text output.
|
|
if w.Status != http.StatusOK && w.buffer.Len() == 0 && w.BytesWritten() == 0 {
|
|
w.buffer.WriteString(http.StatusText(w.Status))
|
|
}
|
|
if w.buffer.Len() > 0 {
|
|
_, _ = w.Writer.Write(w.buffer.Bytes())
|
|
w.buffer.Reset()
|
|
if flusher, ok := w.RawWriter().(http.Flusher); ok {
|
|
flusher.Flush()
|
|
}
|
|
}
|
|
}
|