mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
88 lines
2.3 KiB
Go
88 lines
2.3 KiB
Go
/*
|
|
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
|
Copyright (C) ITsysCOM GmbH
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
|
|
package cores
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/gob"
|
|
"io"
|
|
"log"
|
|
"net/rpc"
|
|
)
|
|
|
|
// All of the logic follows standard library implementation in this file
|
|
// Only here to use as object
|
|
|
|
func newGobServerCodec(conn io.ReadWriteCloser) rpc.ServerCodec {
|
|
buf := bufio.NewWriter(conn)
|
|
return &gobServerCodec{
|
|
rwc: conn,
|
|
dec: gob.NewDecoder(conn),
|
|
enc: gob.NewEncoder(buf),
|
|
encBuf: buf,
|
|
}
|
|
}
|
|
|
|
type gobServerCodec struct {
|
|
rwc io.ReadWriteCloser
|
|
dec *gob.Decoder
|
|
enc *gob.Encoder
|
|
encBuf *bufio.Writer
|
|
closed bool
|
|
}
|
|
|
|
func (c *gobServerCodec) ReadRequestHeader(r *rpc.Request) error {
|
|
return c.dec.Decode(r)
|
|
}
|
|
|
|
func (c *gobServerCodec) ReadRequestBody(body interface{}) error {
|
|
return c.dec.Decode(body)
|
|
}
|
|
|
|
func (c *gobServerCodec) WriteResponse(r *rpc.Response, body interface{}) (err error) {
|
|
if err = c.enc.Encode(r); err != nil {
|
|
if c.encBuf.Flush() == nil {
|
|
// Gob couldn't encode the header. Should not happen, so if it does,
|
|
// shut down the connection to signal that the connection is broken.
|
|
log.Println("rpc: gob error encoding response:", err)
|
|
c.Close()
|
|
}
|
|
return
|
|
}
|
|
if err = c.enc.Encode(body); err != nil {
|
|
if c.encBuf.Flush() == nil {
|
|
// Was a gob problem encoding the body but the header has been written.
|
|
// Shut down the connection to signal that the connection is broken.
|
|
log.Println("rpc: gob error encoding body:", err)
|
|
c.Close()
|
|
}
|
|
return
|
|
}
|
|
return c.encBuf.Flush()
|
|
}
|
|
|
|
func (c *gobServerCodec) Close() error {
|
|
if c.closed {
|
|
// Only call c.rwc.Close once; otherwise the semantics are undefined.
|
|
return nil
|
|
}
|
|
c.closed = true
|
|
return c.rwc.Close()
|
|
}
|