54531d0a44
This PR cldean up remaining left over of github.com/golang/protobuf in coredns, as github.com/golang/protobuf has been deprecated. Note the github.com/golang/protobuf is now only indirectly used by external libraries. Manually updating coredns' dependency to futher clean up has been tried. However, it was not possible as external library usages of github.com/golang/protobuf are too interleaved. This PR fixes 4939. Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
40 lines
857 B
Go
40 lines
857 B
Go
package dnstap
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
tap "github.com/dnstap/golang-dnstap"
|
|
fs "github.com/farsightsec/golang-framestream"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// encoder wraps a golang-framestream.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() }
|