mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-12 15:21:09 +00:00
refactor: remove mutex from CustomEvent and update TCP connection handling in tests
This commit is contained in:
@@ -9,7 +9,6 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type stringWriter interface {
|
type stringWriter interface {
|
||||||
@@ -48,13 +47,13 @@ var dataReplacer = strings.NewReplacer(
|
|||||||
"\n", "\n",
|
"\n", "\n",
|
||||||
"\r", "\\r")
|
"\r", "\\r")
|
||||||
|
|
||||||
|
// CustomEvent does not synchronize writes to the response writer. Streaming
|
||||||
|
// callers must serialize event writes at the stream level.
|
||||||
type CustomEvent struct {
|
type CustomEvent struct {
|
||||||
Event string
|
Event string
|
||||||
Id string
|
Id string
|
||||||
Retry uint
|
Retry uint
|
||||||
Data interface{}
|
Data interface{}
|
||||||
|
|
||||||
Mutex sync.Mutex
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func encode(writer io.Writer, event CustomEvent) error {
|
func encode(writer io.Writer, event CustomEvent) error {
|
||||||
@@ -76,8 +75,6 @@ func (r CustomEvent) Render(w http.ResponseWriter) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r CustomEvent) WriteContentType(w http.ResponseWriter) {
|
func (r CustomEvent) WriteContentType(w http.ResponseWriter) {
|
||||||
r.Mutex.Lock()
|
|
||||||
defer r.Mutex.Unlock()
|
|
||||||
header := w.Header()
|
header := w.Header()
|
||||||
header["Content-Type"] = writeContentType
|
header["Content-Type"] = writeContentType
|
||||||
|
|
||||||
|
|||||||
@@ -364,7 +364,7 @@ func TestSMTPPlainAuthRejectsRemotePlaintextConnection(t *testing.T) {
|
|||||||
SMTPFrom = "sender@example.com"
|
SMTPFrom = "sender@example.com"
|
||||||
SMTPToken = "secret"
|
SMTPToken = "secret"
|
||||||
|
|
||||||
conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", server.host, server.port))
|
conn, err := net.Dial("tcp", net.JoinHostPort(server.host, strconv.Itoa(server.port)))
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
client, err := smtp.NewClient(conn, SMTPServer)
|
client, err := smtp.NewClient(conn, SMTPServer)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|||||||
Reference in New Issue
Block a user