diff options
author | Martin Czygan <martin.czygan@gmail.com> | 2021-07-05 23:38:38 +0200 |
---|---|---|
committer | Martin Czygan <martin.czygan@gmail.com> | 2021-07-05 23:38:38 +0200 |
commit | e9a3ce1f7112a55bffeeb6d7a9dd26584da8276d (patch) | |
tree | 490cfc38b56f77963bbb044f1c2d3b9e7fe551b5 /skate | |
parent | db954963e097238f9df13e6c7820f19d9d83f0f0 (diff) | |
download | refcat-e9a3ce1f7112a55bffeeb6d7a9dd26584da8276d.tar.gz refcat-e9a3ce1f7112a55bffeeb6d7a9dd26584da8276d.zip |
we need a safe encoder, not just a safe writer
Diffstat (limited to 'skate')
-rw-r--r-- | skate/reduce.go | 2 | ||||
-rw-r--r-- | skate/xio/util.go | 19 |
2 files changed, 20 insertions, 1 deletions
diff --git a/skate/reduce.go b/skate/reduce.go index 8325bc0..61e8cd6 100644 --- a/skate/reduce.go +++ b/skate/reduce.go @@ -64,7 +64,7 @@ func groupLogf(g *zipkey.Group, s string, vs ...interface{}) { // match result, e.g. for doi matches. func ZippyExact(releases, refs io.Reader, matchResult MatchResult, w io.Writer) error { var ( - enc = json.NewEncoder(xio.NewSingleWriter(w)) + enc = xio.NewSafeEncoder(json.NewEncoder(w)) keyer = makeKeyFunc("\t", 1) i = 0 bref BiblioRef diff --git a/skate/xio/util.go b/skate/xio/util.go index 49f38a3..9f6cc26 100644 --- a/skate/xio/util.go +++ b/skate/xio/util.go @@ -9,6 +9,25 @@ import ( "sync" ) +type Encoder interface { + Encode(interface{}) error +} + +type SafeEncoder struct { + sync.Mutex + enc Encoder +} + +func NewSafeEncoder(enc Encoder) *SafeEncoder { + return &SafeEncoder{enc: enc} +} + +func (s *SafeEncoder) Encode(v interface{}) error { + s.Lock() + defer s.Unlock() + return s.enc.Encode(v) +} + // SingleWriter makes any writer thread safe. type SingleWriter struct { sync.Mutex |