| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 package rawdatastore | |
| 6 | |
| 7 import ( | |
| 8 "bytes" | |
| 9 ) | |
| 10 | |
| 11 // Buffer is the interface which corresponds to the subset of *bytes.Buffer | |
| 12 // that this package requires. | |
| 13 type Buffer interface { | |
| 14 String() string | |
| 15 Grow(int) | |
| 16 | |
| 17 Read([]byte) (int, error) | |
| 18 ReadByte() (byte, error) | |
| 19 | |
| 20 Write([]byte) (int, error) | |
| 21 WriteByte(c byte) error | |
| 22 WriteString(s string) (int, error) | |
| 23 } | |
| 24 | |
| 25 var _ Buffer = (*bytes.Buffer)(nil) | |
| 26 | |
| 27 // InvertibleBuffer is just like Buffer, except that it also has a stateful | |
| 28 // Invert() method, which will cause all reads and writes to/from it to be | |
| 29 // inverted (e.g. every byte XOR 0xFF). | |
| 30 // | |
| 31 // Implementing queries requires manipulating the index entries (e.g. | |
| 32 // synthesizing them, parsing them, etc.). In particular, when you have | |
| 33 // a reverse-sorted field (e.g. high to low instead of low to high), it's | |
| 34 // achieved by having all the bits inverted. | |
| 35 // | |
| 36 // All the serialization formats include delimiter information, which the | |
| 37 // parsers only know to parse non-inverted. If we don't have this buffer, we'd | |
| 38 // basically have to invert every byte in the []byte array when we're trying to | |
| 39 // decode a reverse-ordered field (including the bytes of all fields after the | |
| 40 // one we intend to parse) so that the parser can consume as many bytes as it | |
| 41 // needs (and it only knows the number of bytes it needs as it decodes them). | |
| 42 // This InvertibleBuffer lets that happen on the fly without having to flip the | |
| 43 // whole []byte. | |
| 44 // | |
| 45 // If you know you need it, you'll know it's the right thing. If you're not sure | |
| 46 // then you definitely don't need it! | |
| 47 type InvertibleBuffer interface { | |
| 48 Buffer | |
| 49 Invert() | |
| 50 } | |
| 51 | |
| 52 type invertibleBuffer struct { | |
| 53 Buffer | |
| 54 invert bool | |
| 55 } | |
| 56 | |
| 57 // Invertible returns an InvertibleBuffer based on the Buffer. | |
| 58 func Invertible(b Buffer) InvertibleBuffer { | |
| 59 return &invertibleBuffer{b, false} | |
| 60 } | |
| 61 | |
| 62 func (ib *invertibleBuffer) Read(bs []byte) (int, error) { | |
| 63 n, err := ib.Buffer.Read(bs) | |
| 64 if ib.invert { | |
| 65 for i, b := range bs { | |
| 66 bs[i] = b ^ 0xFF | |
| 67 } | |
| 68 } | |
| 69 return n, err | |
| 70 } | |
| 71 | |
| 72 func (ib *invertibleBuffer) WriteString(s string) (int, error) { | |
| 73 if ib.invert { | |
| 74 ib.Grow(len(s)) | |
| 75 for i := 0; i < len(s); i++ { | |
| 76 if err := ib.Buffer.WriteByte(s[i] ^ 0xFF); err != nil { | |
| 77 return i, err | |
| 78 } | |
| 79 } | |
| 80 return len(s), nil | |
| 81 } | |
| 82 return ib.Buffer.WriteString(s) | |
| 83 } | |
| 84 | |
| 85 func (ib *invertibleBuffer) Write(bs []byte) (int, error) { | |
| 86 if ib.invert { | |
| 87 ib.Grow(len(bs)) | |
| 88 for i, b := range bs { | |
| 89 if err := ib.Buffer.WriteByte(b ^ 0xFF); err != nil { | |
| 90 return i, err | |
| 91 } | |
| 92 } | |
| 93 return len(bs), nil | |
| 94 } | |
| 95 return ib.Buffer.Write(bs) | |
| 96 } | |
| 97 | |
| 98 func (ib *invertibleBuffer) WriteByte(b byte) error { | |
| 99 if ib.invert { | |
| 100 b = b ^ 0xFF | |
| 101 } | |
| 102 return ib.Buffer.WriteByte(b) | |
| 103 } | |
| 104 | |
| 105 func (ib *invertibleBuffer) ReadByte() (byte, error) { | |
| 106 ret, err := ib.Buffer.ReadByte() | |
| 107 if ib.invert { | |
| 108 ret = ret ^ 0xFF | |
| 109 } | |
| 110 return ret, err | |
| 111 } | |
| 112 | |
| 113 func (ib *invertibleBuffer) Invert() { | |
| 114 ib.invert = !ib.invert | |
| 115 } | |
| OLD | NEW |