luxolus
/
open-vault
2
0
Fork 0
Code Issues 1 Pull Requests Releases Activity

Added a single retry after a reconnection

This commit is contained in:
Brian Kassouf 2017-02-06 11:38:38 -08:00
parent af1847f2b4
commit 09049c2787
1 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
builtin/audit/socket/backend.go
View File

@ -122,13 +122,14 @@ func (b *Backend) LogRequest(auth *logical.Auth, req *logical.Request, outerErr
b.Lock()
defer b.Unlock()
b.connection.SetDeadline(time.Now().Add(b.writeDuration))
_, err := b.connection.Write(buf.Bytes())
err := b.write(buf.Bytes())
if err != nil {
rErr := b.reconnect()
if rErr != nil {
err = multierror.Append(err, rErr)
} else {
// Try once more after reconnecting
err = b.write(buf.Bytes())
}
}
@ -145,19 +146,34 @@ func (b *Backend) LogResponse(auth *logical.Auth, req *logical.Request,
b.Lock()
defer b.Unlock()
b.connection.SetDeadline(time.Now().Add(b.writeDuration))
_, err := b.connection.Write(buf.Bytes())
err := b.write(buf.Bytes())
if err != nil {
rErr := b.reconnect()
if rErr != nil {
err = multierror.Append(err, rErr)
} else {
// Try once more after reconnecting
err = b.write(buf.Bytes())
}
}
return err
}
func (b *Backend) write(buf []byte) error {
err := b.connection.SetWriteDeadline(time.Now().Add(b.writeDuration))
if err != nil {
return err
}
_, err = b.connection.Write(buf)
if err != nil {
return err
}
return err
}
func (b *Backend) reconnect() error {
conn, err := net.Dial(b.socketType, b.address)
if err != nil {