/*
* 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.io;
import java.io.File;
import java.io.IOException;
import java.util.Map;
import java.util.SortedMap;
import java.util.Set;
import java.util.TreeMap;
import org.apache.cassandra.config.DatabaseDescriptor;
import org.apache.cassandra.db.Column;
import org.apache.cassandra.db.ColumnFamily;
import org.apache.cassandra.db.Table;
import org.apache.cassandra.service.StorageService;
import org.apache.cassandra.io.util.DataOutputBuffer;
/**
* TODO: These methods imitate Memtable.writeSortedKeys to some degree, but
* because it is so monolithic, we can't reuse much.
*/
public class SSTableUtils
{
// first configured table and cf
public static String TABLENAME;
public static String CFNAME;
static
{
try
{
TABLENAME = DatabaseDescriptor.getTables().iterator().next();
CFNAME = Table.open(TABLENAME).getColumnFamilies().iterator().next();
}
catch(IOException e)
{
throw new RuntimeException(e);
}
}
public static File tempSSTableFile(String tablename, String cfname) throws IOException
{
File tempdir = File.createTempFile(tablename, cfname);
if(!tempdir.delete() || !tempdir.mkdir())
throw new IOException("Temporary directory creation failed.");
tempdir.deleteOnExit();
File tabledir = new File(tempdir, tablename);
tabledir.mkdir();
tabledir.deleteOnExit();
return File.createTempFile(cfname + "-",
"-" + SSTable.TEMPFILE_MARKER + "-Data.db",
tabledir);
}
public static SSTableReader writeSSTable(Set<String> keys) throws IOException
{
TreeMap<String, ColumnFamily> map = new TreeMap<String, ColumnFamily>();
for (String key : keys)
{
ColumnFamily cf = ColumnFamily.create(TABLENAME, CFNAME);
cf.addColumn(new Column(key.getBytes(), key.getBytes(), 0));
map.put(key, cf);
}
return writeSSTable(map);
}
public static SSTableReader writeSSTable(SortedMap<String, ColumnFamily> entries) throws IOException
{
TreeMap<String, byte[]> map = new TreeMap<String, byte[]>();
for (Map.Entry<String, ColumnFamily> entry : entries.entrySet())
{
DataOutputBuffer buffer = new DataOutputBuffer();
ColumnFamily.serializer().serializeWithIndexes(entry.getValue(), buffer);
map.put(entry.getKey(), buffer.getData());
}
return writeRawSSTable(TABLENAME, CFNAME, map);
}
public static SSTableReader writeRawSSTable(String tablename, String cfname, SortedMap<String, byte[]> entries) throws IOException
{
File f = tempSSTableFile(tablename, cfname);
SSTableWriter writer = new SSTableWriter(f.getAbsolutePath(), entries.size(), StorageService.getPartitioner());
for (Map.Entry<String, byte[]> entry : entries.entrySet())
writer.append(writer.partitioner.decorateKey(entry.getKey()),
entry.getValue());
new File(writer.indexFilename()).deleteOnExit();
new File(writer.filterFilename()).deleteOnExit();
return writer.closeAndOpenReader();
}
}