| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 org.apache.ignite.internal.table.distributed.command; |
| |
| import java.util.HashSet; |
| import java.util.Set; |
| import org.apache.ignite.internal.schema.BinaryRow; |
| import org.apache.ignite.raft.client.ReadCommand; |
| import org.jetbrains.annotations.NotNull; |
| |
| /** |
| * This is a command for the batch get operation. |
| */ |
| public class GetAllCommand implements ReadCommand { |
| /** Binary key rows. */ |
| private transient Set<BinaryRow> keyRows; |
| |
| /* |
| * Row bytes. |
| * It is a temporary solution, before network have not implement correct serialization BinaryRow. |
| * TODO: Remove the field after (IGNITE-14793). |
| */ |
| private byte[] keyRowsBytes; |
| |
| /** |
| * Creates a new instance of GetAllCommand with the given keys to be got. |
| * The {@code keyRows} should not be {@code null} or empty. |
| * |
| * @param keyRows Binary key rows. |
| */ |
| public GetAllCommand(@NotNull Set<BinaryRow> keyRows) { |
| assert keyRows != null && !keyRows.isEmpty(); |
| |
| this.keyRows = keyRows; |
| |
| CommandUtils.rowsToBytes(keyRows, bytes -> keyRowsBytes = bytes); |
| } |
| |
| /** |
| * Gets a set of binary key rows to be got. |
| * |
| * @return Binary keys. |
| */ |
| public Set<BinaryRow> getKeyRows() { |
| if (keyRows == null && keyRowsBytes != null) { |
| keyRows = new HashSet<>(); |
| |
| CommandUtils.readRows(keyRowsBytes, keyRows::add); |
| } |
| |
| return keyRows; |
| } |
| } |