tkalkirill commented on a change in pull request #697: URL: https://github.com/apache/ignite-3/pull/697#discussion_r838325426
########## File path: modules/storage-page-memory/src/main/java/org/apache/ignite/internal/storage/pagememory/PageMemoryPartitionStorage.java ########## @@ -0,0 +1,417 @@ +/* + * 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.storage.pagememory; + +import static org.apache.ignite.internal.pagememory.PageIdAllocator.FLAG_AUX; +import static org.apache.ignite.internal.pagememory.PageIdAllocator.MAX_PARTITION_ID; +import static org.apache.ignite.internal.storage.StorageUtils.groupId; + +import java.nio.ByteBuffer; +import java.nio.file.Path; +import java.util.ArrayList; +import java.util.Collection; +import java.util.Iterator; +import java.util.List; +import java.util.NoSuchElementException; +import java.util.concurrent.CompletableFuture; +import java.util.concurrent.atomic.AtomicLong; +import java.util.function.Predicate; +import org.apache.ignite.configuration.schemas.table.TableConfiguration; +import org.apache.ignite.configuration.schemas.table.TableView; +import org.apache.ignite.internal.pagememory.tree.BplusTree; +import org.apache.ignite.internal.pagememory.tree.IgniteTree; +import org.apache.ignite.internal.pagememory.util.PageLockListenerNoOp; +import org.apache.ignite.internal.storage.DataRow; +import org.apache.ignite.internal.storage.InvokeClosure; +import org.apache.ignite.internal.storage.OperationType; +import org.apache.ignite.internal.storage.PartitionStorage; +import org.apache.ignite.internal.storage.SearchRow; +import org.apache.ignite.internal.storage.StorageException; +import org.apache.ignite.internal.storage.StorageUtils; +import org.apache.ignite.internal.util.Cursor; +import org.apache.ignite.internal.util.IgniteCursor; +import org.apache.ignite.lang.IgniteInternalCheckedException; +import org.apache.ignite.lang.IgniteInternalException; +import org.jetbrains.annotations.Nullable; + +/** + * Storage implementation based on a {@link BplusTree}. + */ +// TODO: IGNITE-16644 Support snapshots. +class PageMemoryPartitionStorage implements PartitionStorage { + private final int partId; + + private final TableTree tree; + + private final TableFreeList freeList; + + /** + * Constructor. + * + * @param partId Partition id. + * @param tableCfg – Table configuration. + * @param dataRegion – Data region for the table. + * @param freeList Table free list. + * @throws StorageException If there is an error while creating the partition storage. + */ + public PageMemoryPartitionStorage( + int partId, + TableConfiguration tableCfg, + PageMemoryDataRegion dataRegion, + TableFreeList freeList + ) throws StorageException { + assert partId >= 0 && partId < MAX_PARTITION_ID : partId; + + this.partId = partId; + + this.freeList = freeList; + + TableView tableView = tableCfg.value(); + + int grpId = groupId(tableView); + + try { + long metaPageId = dataRegion.pageMemory().allocatePage(grpId, partId, FLAG_AUX); + + tree = new TableTree( + grpId, + tableView.name(), + dataRegion.pageMemory(), + PageLockListenerNoOp.INSTANCE, + new AtomicLong(), + metaPageId, + freeList, + partId + ); + } catch (IgniteInternalCheckedException e) { + throw new StorageException("Error occurred while creating the partition storage", e); + } + } + + /** {@inheritDoc} */ + @Override + public int partitionId() { + return partId; + } + + /** {@inheritDoc} */ + @Override + public @Nullable DataRow read(SearchRow key) throws StorageException { + try { + return tree.findOne(wrap(key)); + } catch (IgniteInternalCheckedException e) { + throw new StorageException("Error reading row", e); + } + } + + /** {@inheritDoc} */ + @Override + public Collection<DataRow> readAll(List<? extends SearchRow> keys) throws StorageException { + Collection<DataRow> res = new ArrayList<>(keys.size()); + + try { + for (SearchRow key : keys) { + res.add(tree.findOne(wrap(key))); + } + } catch (IgniteInternalCheckedException e) { + throw new StorageException("Error reading rows", e); + } + + return res; + } + + /** {@inheritDoc} */ + @Override + public void write(DataRow row) throws StorageException { + try { + TableDataRow dataRow = wrap(row); + + freeList.insertDataRow(dataRow); + + tree.put(dataRow); + } catch (IgniteInternalCheckedException e) { + throw new StorageException("Error writing row", e); + } + } + + /** {@inheritDoc} */ + @Override + public void writeAll(List<? extends DataRow> rows) throws StorageException { + try { + for (DataRow row : rows) { + TableDataRow dataRow = wrap(row); + + freeList.insertDataRow(dataRow); + + tree.put(dataRow); + } + } catch (IgniteInternalCheckedException e) { + throw new StorageException("Error writing rows", e); + } + } + + /** {@inheritDoc} */ + @Override + public Collection<DataRow> insertAll(List<? extends DataRow> rows) throws StorageException { + Collection<DataRow> cantInsert = new ArrayList<>(); + + try { + for (DataRow row : rows) { + TableDataRow dataRow = wrap(row); + + if (tree.findOne(dataRow) == null) { Review comment: Tried to fix it. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
