2015-04-24 23:11:47 +02:00
|
|
|
// Copyright (c) 2014 Couchbase, Inc.
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
|
|
|
|
// except in compliance with the License. You may obtain a copy of the License at
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
// Unless required by applicable law or agreed to in writing, software distributed under the
|
|
|
|
// License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
|
|
|
|
// either express or implied. See the License for the specific language governing permissions
|
|
|
|
// and limitations under the License.
|
|
|
|
|
|
|
|
package store
|
|
|
|
|
2016-01-27 23:51:07 +01:00
|
|
|
import "unsafe"
|
|
|
|
|
2015-04-24 23:11:47 +02:00
|
|
|
// At the moment this happens to be the same interface as described by
|
|
|
|
// RocksDB, but this may not always be the case.
|
|
|
|
|
|
|
|
type MergeOperator interface {
|
|
|
|
|
|
|
|
// FullMerge the full sequence of operands on top of the existingValue
|
|
|
|
// if no value currently exists, existingValue is nil
|
|
|
|
// return the merged value, and success/failure
|
|
|
|
FullMerge(key, existingValue []byte, operands [][]byte) ([]byte, bool)
|
|
|
|
|
|
|
|
// Partially merge these two operands.
|
|
|
|
// If partial merge cannot be done, return nil,false, which will defer
|
|
|
|
// all processing until the FullMerge is done.
|
|
|
|
PartialMerge(key, leftOperand, rightOperand []byte) ([]byte, bool)
|
|
|
|
|
|
|
|
// Name returns an identifier for the operator
|
|
|
|
Name() string
|
|
|
|
}
|
|
|
|
|
2016-01-27 23:51:07 +01:00
|
|
|
// NativeMergeOperator is a merge operator that also includeTermVectors
|
|
|
|
// a C implementation
|
|
|
|
type NativeMergeOperator interface {
|
|
|
|
MergeOperator
|
|
|
|
|
|
|
|
// a pointer to function in C with the signature
|
|
|
|
// char* (*full_merge)(void *state,
|
|
|
|
// const char* key, size_t key_length,
|
|
|
|
// const char* existing_value,
|
|
|
|
// size_t existing_value_length,
|
|
|
|
// const char* const* operands_list,
|
|
|
|
// const size_t* operands_list_length, int num_operands,
|
|
|
|
// unsigned char* success, size_t* new_value_length)
|
|
|
|
FullMergeC() unsafe.Pointer
|
|
|
|
|
|
|
|
// a pointer to function in C with the signature
|
|
|
|
// char* (*partial_merge)(void *state,
|
|
|
|
// const char* key, size_t key_length,
|
|
|
|
// const char* const* operands_list,
|
|
|
|
// const size_t* operands_list_length, int num_operands,
|
|
|
|
// unsigned char* success, size_t* new_value_length)
|
|
|
|
PartialMergeC() unsafe.Pointer
|
|
|
|
|
|
|
|
// a pointer to function in C with signature
|
|
|
|
// const char* merge_operator_name_fn(void *state)
|
|
|
|
NameC() unsafe.Pointer
|
|
|
|
}
|
|
|
|
|
2015-04-24 23:11:47 +02:00
|
|
|
type EmulatedMerge struct {
|
2015-09-23 20:25:47 +02:00
|
|
|
Merges map[string][][]byte
|
2015-04-24 23:11:47 +02:00
|
|
|
mo MergeOperator
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewEmulatedMerge(mo MergeOperator) *EmulatedMerge {
|
|
|
|
return &EmulatedMerge{
|
2015-09-23 20:25:47 +02:00
|
|
|
Merges: make(map[string][][]byte),
|
2015-04-24 23:11:47 +02:00
|
|
|
mo: mo,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m *EmulatedMerge) Merge(key, val []byte) {
|
2015-09-23 20:25:47 +02:00
|
|
|
ops, ok := m.Merges[string(key)]
|
2015-04-24 23:11:47 +02:00
|
|
|
if ok && len(ops) > 0 {
|
|
|
|
last := ops[len(ops)-1]
|
|
|
|
mergedVal, partialMergeOk := m.mo.PartialMerge(key, last, val)
|
|
|
|
if partialMergeOk {
|
|
|
|
// replace last entry with the result of the merge
|
|
|
|
ops[len(ops)-1] = mergedVal
|
|
|
|
} else {
|
|
|
|
// could not partial merge, append this to the end
|
|
|
|
ops = append(ops, val)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
ops = [][]byte{val}
|
|
|
|
}
|
2015-09-23 20:25:47 +02:00
|
|
|
m.Merges[string(key)] = ops
|
2015-04-24 23:11:47 +02:00
|
|
|
}
|