Move http write functions to writers.go

This commit is contained in:
2025-07-29 09:33:03 +03:00
parent ec94df5f4a
commit 1f36f2d7bc
2 changed files with 33 additions and 18 deletions

View File

@@ -1,10 +1,5 @@
package rpc package rpc
import (
"encoding/json"
"net/http"
)
const ( const (
ErrParseError = -32700 ErrParseError = -32700
ErrParseErrorS = "Parse error" ErrParseErrorS = "Parse error"
@@ -24,16 +19,3 @@ const (
ErrContextVersion = -32010 ErrContextVersion = -32010
ErrContextVersionS = "Invalid context version" ErrContextVersionS = "Invalid context version"
) )
func WriteRouterError(w http.ResponseWriter, status int, e *RPCError) error {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)
data, err := json.Marshal(e)
if err != nil {
return err
}
_, err = w.Write(data)
return err
}

View File

@@ -0,0 +1,33 @@
package rpc
import (
"encoding/json"
"fmt"
"net/http"
)
func write(w http.ResponseWriter, status int, msg any) error {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)
switch m := msg.(type) {
case RPCError, *RPCError,
RPCResponse, *RPCResponse:
data, err := json.Marshal(m)
if err != nil {
return err
}
_, err = w.Write(data)
return err
default:
return fmt.Errorf("invalid RPC structure: %T", msg)
}
}
func WriteRouterError(w http.ResponseWriter, status int, errm *RPCError) error {
return write(w, status, errm)
}
func WriteResponse(w http.ResponseWriter, response *RPCResponse) error {
return write(w, http.StatusOK, response)
}