mirror of
https://github.com/coredns/coredns.git
synced 2025-11-02 10:13:14 -05:00
a136b7128b
* plugin/dnstap: remove encoder*.go Those files reimplemented parts of the dnstap spec, we can just use the dnstap functions for that. This leaves all the queuing that is enabled and drops messages if the dnstap reader can't keep up. In the new code flush() would never return an error (at least I couldn't make it do so), so the reconnect functionally is moved to kick off when we get write errors. Some smaller cosmetic changes as well, `d.socket` is now `proto`, which makes the dial() function smaller. Total testing time is now <1s (which was the impetus to look into this plugin *again*). See #4238 The buffered channel needs to be sized correctly, as we may need to do some queing if the dnstap reader can't keep up. Signed-off-by: Miek Gieben <miek@miek.nl> * add missing file Signed-off-by: Miek Gieben <miek@miek.nl> * update doc on queing Signed-off-by: Miek Gieben <miek@miek.nl>
42 lines
907 B
Go
42 lines
907 B
Go
// Package dnstapio is a small wrapper around golang-framestream
|
|
package dnstapio
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
tap "github.com/dnstap/golang-dnstap"
|
|
fs "github.com/farsightsec/golang-framestream"
|
|
"github.com/golang/protobuf/proto"
|
|
)
|
|
|
|
// Encoder wraps a fs.Encoder.
|
|
type Encoder struct {
|
|
fs *fs.Encoder
|
|
}
|
|
|
|
func newEncoder(w io.Writer, timeout time.Duration) (*Encoder, error) {
|
|
fs, err := fs.NewEncoder(w, &fs.EncoderOptions{
|
|
ContentType: []byte("protobuf:dnstap.Dnstap"),
|
|
Bidirectional: true,
|
|
Timeout: timeout,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Encoder{fs}, nil
|
|
}
|
|
|
|
func (e *Encoder) writeMsg(msg *tap.Dnstap) error {
|
|
buf, err := proto.Marshal(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = e.fs.Write(buf) // n < len(buf) should return an error
|
|
return err
|
|
}
|
|
|
|
func (e *Encoder) flush() error { return e.fs.Flush() }
|
|
func (e *Encoder) close() error { return e.fs.Close() }
|