cassandra-pr mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From dineshjoshi <...@git.apache.org>
Subject [GitHub] cassandra pull request #239: [CASSANDRA-14556] Optimize Streaming
Date Wed, 25 Jul 2018 22:09:06 GMT
Github user dineshjoshi commented on a diff in the pull request:

    https://github.com/apache/cassandra/pull/239#discussion_r205277660
  
    --- Diff: src/java/org/apache/cassandra/db/streaming/CassandraBlockStreamReader.java ---
    @@ -0,0 +1,177 @@
    +/*
    + * 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.cassandra.db.streaming;
    +
    +import java.io.File;
    +import java.io.IOException;
    +import java.util.Set;
    +
    +import com.google.common.base.Throwables;
    +import org.slf4j.Logger;
    +import org.slf4j.LoggerFactory;
    +
    +import org.apache.cassandra.db.ColumnFamilyStore;
    +import org.apache.cassandra.db.DecoratedKey;
    +import org.apache.cassandra.db.Directories;
    +import org.apache.cassandra.db.SerializationHeader;
    +import org.apache.cassandra.db.lifecycle.LifecycleTransaction;
    +import org.apache.cassandra.io.sstable.Component;
    +import org.apache.cassandra.io.sstable.Descriptor;
    +import org.apache.cassandra.io.sstable.SSTableMultiWriter;
    +import org.apache.cassandra.io.sstable.format.SSTableFormat;
    +import org.apache.cassandra.io.sstable.format.Version;
    +import org.apache.cassandra.io.sstable.format.big.BigTableBlockWriter;
    +import org.apache.cassandra.io.util.DataInputPlus;
    +import org.apache.cassandra.schema.TableId;
    +import org.apache.cassandra.streaming.ProgressInfo;
    +import org.apache.cassandra.streaming.StreamReceiver;
    +import org.apache.cassandra.streaming.StreamSession;
    +import org.apache.cassandra.streaming.messages.StreamMessageHeader;
    +import org.apache.cassandra.utils.FBUtilities;
    +
    +/**
    + * CassandraBlockStreamReader reads SSTable off the wire and writes it to disk.
    + */
    +public class CassandraBlockStreamReader implements IStreamReader
    +{
    +    private static final Logger logger = LoggerFactory.getLogger(CassandraBlockStreamReader.class);
    +    private final TableId tableId;
    +    private final StreamSession session;
    +    private final int sstableLevel;
    +    private final SerializationHeader.Component header;
    +    private final int fileSeqNum;
    +    private final ComponentManifest manifest;
    +    private final SSTableFormat.Type format;
    +    private final Version version;
    +    private final DecoratedKey firstKey;
    +
    +    public CassandraBlockStreamReader(StreamMessageHeader header, CassandraStreamHeader
streamHeader, StreamSession session)
    +    {
    +        if (session.getPendingRepair() != null)
    +        {
    +            // we should only ever be streaming pending repair
    +            // sstables if the session has a pending repair id
    +            if (!session.getPendingRepair().equals(header.pendingRepair))
    +                throw new IllegalStateException(String.format("Stream Session & SSTable
({}) pendingRepair UUID mismatch.",
    +                                                              header.tableId));
    +        }
    +        this.session = session;
    +        this.tableId = header.tableId;
    +        this.manifest = streamHeader.componentManifest;
    +        this.sstableLevel = streamHeader.sstableLevel;
    +        this.header = streamHeader.header;
    +        this.format = streamHeader.format;
    +        this.fileSeqNum = header.sequenceNumber;
    +        this.version = streamHeader.version;
    +        this.firstKey = streamHeader.firstKey;
    +    }
    +
    +    /**
    +     * @param inputPlus where this reads data from
    +     * @return SSTable transferred
    +     * @throws IOException if reading the remote sstable fails. Will throw an RTE if
local write fails.
    +     */
    +    @SuppressWarnings("resource") // input needs to remain open, streams on top of it
can't be closed
    +    @Override
    +    public SSTableMultiWriter read(DataInputPlus inputPlus) throws IOException
    +    {
    +        long totalSize = manifest.getTotalSize();
    +
    +        ColumnFamilyStore cfs = ColumnFamilyStore.getIfExists(tableId);
    +
    +        if (cfs == null)
    +        {
    +            // schema was dropped during streaming
    +            throw new IOException("CF " + tableId + " was dropped during streaming");
    +        }
    +
    +        logger.debug("[Stream #{}] Start receiving file #{} from {}, size = {}, ks =
'{}', table = '{}'.",
    +                     session.planId(), fileSeqNum, session.peer, totalSize, cfs.keyspace.getName(),
cfs.getTableName());
    --- End diff --
    
    Cherry picked.


---

---------------------------------------------------------------------
To unsubscribe, e-mail: pr-unsubscribe@cassandra.apache.org
For additional commands, e-mail: pr-help@cassandra.apache.org


Mime
View raw message