| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 package memory | 5 package memory |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "bytes" | 8 "bytes" |
| 9 "runtime" | 9 "runtime" |
| 10 "sync" | 10 "sync" |
| (...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 62 // memStore is a gkvlite.Store which will panic for anything which might | 62 // memStore is a gkvlite.Store which will panic for anything which might |
| 63 // otherwise return an error. | 63 // otherwise return an error. |
| 64 // | 64 // |
| 65 // This is reasonable for in-memory Store objects, since the only errors that | 65 // This is reasonable for in-memory Store objects, since the only errors that |
| 66 // should occur happen with file IO on the underlying file (which of course | 66 // should occur happen with file IO on the underlying file (which of course |
| 67 // doesn't exist). | 67 // doesn't exist). |
| 68 type memStore gkvlite.Store | 68 type memStore gkvlite.Store |
| 69 | 69 |
| 70 func newMemStore() *memStore { | 70 func newMemStore() *memStore { |
| 71 ret, err := gkvlite.NewStore(nil) | 71 ret, err := gkvlite.NewStore(nil) |
| 72 » if err != nil { | 72 » memoryCorruption(err) |
| 73 » » panic(err) | |
| 74 » } | |
| 75 return (*memStore)(ret) | 73 return (*memStore)(ret) |
| 76 } | 74 } |
| 77 | 75 |
| 78 func (ms *memStore) Snapshot() *memStore { | 76 func (ms *memStore) Snapshot() *memStore { |
| 79 ret := (*memStore)((*gkvlite.Store)(ms).Snapshot()) | 77 ret := (*memStore)((*gkvlite.Store)(ms).Snapshot()) |
| 80 runtime.SetFinalizer((*gkvlite.Store)(ret), func(s *gkvlite.Store) { | 78 runtime.SetFinalizer((*gkvlite.Store)(ret), func(s *gkvlite.Store) { |
| 81 go s.Close() | 79 go s.Close() |
| 82 }) | 80 }) |
| 83 return ret | 81 return ret |
| 84 } | 82 } |
| (...skipping 13 matching lines...) Expand all Loading... |
| 98 // memCollection is a gkvlite.Collection which will panic for anything which | 96 // memCollection is a gkvlite.Collection which will panic for anything which |
| 99 // might otherwise return an error. | 97 // might otherwise return an error. |
| 100 // | 98 // |
| 101 // This is reasonable for in-memory Store objects, since the only errors that | 99 // This is reasonable for in-memory Store objects, since the only errors that |
| 102 // should occur happen with file IO on the underlying file (which of course | 100 // should occur happen with file IO on the underlying file (which of course |
| 103 // doesn't exist. | 101 // doesn't exist. |
| 104 type memCollection gkvlite.Collection | 102 type memCollection gkvlite.Collection |
| 105 | 103 |
| 106 func (mc *memCollection) Get(k []byte) []byte { | 104 func (mc *memCollection) Get(k []byte) []byte { |
| 107 ret, err := (*gkvlite.Collection)(mc).Get(k) | 105 ret, err := (*gkvlite.Collection)(mc).Get(k) |
| 108 » if err != nil { | 106 » memoryCorruption(err) |
| 109 » » panic(err) | |
| 110 » } | |
| 111 return ret | 107 return ret |
| 112 } | 108 } |
| 113 | 109 |
| 114 func (mc *memCollection) MinItem(withValue bool) *gkvlite.Item { | 110 func (mc *memCollection) MinItem(withValue bool) *gkvlite.Item { |
| 115 ret, err := (*gkvlite.Collection)(mc).MinItem(withValue) | 111 ret, err := (*gkvlite.Collection)(mc).MinItem(withValue) |
| 116 » if err != nil { | 112 » memoryCorruption(err) |
| 117 » » panic(err) | |
| 118 » } | |
| 119 return ret | 113 return ret |
| 120 } | 114 } |
| 121 | 115 |
| 122 func (mc *memCollection) Set(k, v []byte) { | 116 func (mc *memCollection) Set(k, v []byte) { |
| 123 » if err := (*gkvlite.Collection)(mc).Set(k, v); err != nil { | 117 » err := (*gkvlite.Collection)(mc).Set(k, v) |
| 124 » » panic(err) | 118 » memoryCorruption(err) |
| 125 » } | |
| 126 } | 119 } |
| 127 | 120 |
| 128 func (mc *memCollection) Delete(k []byte) bool { | 121 func (mc *memCollection) Delete(k []byte) bool { |
| 129 ret, err := (*gkvlite.Collection)(mc).Delete(k) | 122 ret, err := (*gkvlite.Collection)(mc).Delete(k) |
| 130 » if err != nil { | 123 » memoryCorruption(err) |
| 131 » » panic(err) | |
| 132 » } | |
| 133 return ret | 124 return ret |
| 134 } | 125 } |
| 135 | 126 |
| 136 func (mc *memCollection) VisitItemsAscend(target []byte, withValue bool, visitor
gkvlite.ItemVisitor) { | 127 func (mc *memCollection) VisitItemsAscend(target []byte, withValue bool, visitor
gkvlite.ItemVisitor) { |
| 137 » if err := (*gkvlite.Collection)(mc).VisitItemsAscend(target, withValue,
visitor); err != nil { | 128 » err := (*gkvlite.Collection)(mc).VisitItemsAscend(target, withValue, vis
itor) |
| 138 » » panic(err) | 129 » memoryCorruption(err) |
| 139 » } | |
| 140 } | 130 } |
| 141 | 131 |
| 142 func (mc *memCollection) GetTotals() (numItems, numBytes uint64) { | 132 func (mc *memCollection) GetTotals() (numItems, numBytes uint64) { |
| 143 numItems, numBytes, err := (*gkvlite.Collection)(mc).GetTotals() | 133 numItems, numBytes, err := (*gkvlite.Collection)(mc).GetTotals() |
| 144 » if err != nil { | 134 » memoryCorruption(err) |
| 145 » » panic(err) | |
| 146 » } | |
| 147 return | 135 return |
| 148 } | 136 } |
| OLD | NEW |