This repository has been archived by the owner on Feb 11, 2022. It is now read-only.
forked from aerospike/aerospike-client-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
batch_command_exists.go
128 lines (106 loc) · 3.62 KB
/
batch_command_exists.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// Copyright 2013-2020 Aerospike, 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 aerospike
import (
"bytes"
. "github.com/aerospike/aerospike-client-go/types"
Buffer "github.com/aerospike/aerospike-client-go/utils/buffer"
)
type batchCommandExists struct {
batchCommand
keys []*Key
existsArray []bool
index int
}
func newBatchCommandExists(
node *Node,
batch *batchNode,
policy *BatchPolicy,
keys []*Key,
existsArray []bool,
) *batchCommandExists {
res := &batchCommandExists{
batchCommand: batchCommand{
baseMultiCommand: *newMultiCommand(node, nil),
policy: policy,
batch: batch,
},
keys: keys,
existsArray: existsArray,
}
res.oneShot = false
return res
}
func (cmd *batchCommandExists) cloneBatchCommand(batch *batchNode) batcher {
res := *cmd
res.node = batch.Node
res.batch = batch
return &res
}
func (cmd *batchCommandExists) writeBuffer(ifc command) error {
return cmd.setBatchIndexReadCompat(cmd.policy, cmd.keys, cmd.batch, nil, _INFO1_READ|_INFO1_NOBINDATA)
}
// Parse all results in the batch. Add records to shared list.
// If the record was not found, the bins will be nil.
func (cmd *batchCommandExists) parseRecordResults(ifc command, receiveSize int) (bool, error) {
//Parse each message response and add it to the result array
cmd.dataOffset = 0
for cmd.dataOffset < receiveSize {
if err := cmd.readBytes(int(_MSG_REMAINING_HEADER_SIZE)); err != nil {
return false, err
}
resultCode := ResultCode(cmd.dataBuffer[5] & 0xFF)
// The only valid server return codes are "ok" and "not found".
// If other return codes are received, then abort the batch.
if resultCode != 0 && resultCode != KEY_NOT_FOUND_ERROR {
if resultCode == FILTERED_OUT {
cmd.filteredOutCnt++
} else {
return false, NewAerospikeError(resultCode)
}
}
info3 := cmd.dataBuffer[3]
// If cmd is the end marker of the response, do not proceed further
if (int(info3) & _INFO3_LAST) == _INFO3_LAST {
return false, nil
}
batchIndex := int(Buffer.BytesToUint32(cmd.dataBuffer, 14))
fieldCount := int(Buffer.BytesToUint16(cmd.dataBuffer, 18))
opCount := int(Buffer.BytesToUint16(cmd.dataBuffer, 20))
if opCount > 0 {
return false, NewAerospikeError(PARSE_ERROR, "Received bins that were not requested!")
}
key, err := cmd.parseKey(fieldCount)
if err != nil {
return false, err
}
var offset int
offset = batchIndex
if bytes.Equal(key.digest[:], cmd.keys[offset].digest[:]) {
// only set the results to true; as a result, no synchronization is needed
if resultCode == 0 {
cmd.existsArray[offset] = true
}
} else {
return false, NewAerospikeError(PARSE_ERROR, "Unexpected batch key returned: "+key.namespace+","+Buffer.BytesToHexString(key.digest[:])+". Expected: "+Buffer.BytesToHexString(cmd.keys[offset].digest[:]))
}
}
return true, nil
}
func (cmd *batchCommandExists) Execute() error {
return cmd.execute(cmd, true)
}
func (cmd *batchCommandExists) generateBatchNodes(cluster *Cluster) ([]*batchNode, error) {
return newBatchNodeListKeys(cluster, cmd.policy, cmd.keys, cmd.sequenceAP, cmd.sequenceSC, cmd.batch)
}