| /* |
| * 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.iotdb.db.storageengine.dataregion; |
| |
| import org.apache.iotdb.commons.conf.IoTDBConstant; |
| import org.apache.iotdb.commons.exception.MetadataException; |
| import org.apache.iotdb.db.exception.StorageEngineException; |
| import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileManager; |
| import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResource; |
| import org.apache.iotdb.db.storageengine.dataregion.tsfile.generator.TsFileNameGenerator; |
| import org.apache.iotdb.db.utils.constant.TestConstant; |
| import org.apache.iotdb.tsfile.exception.write.WriteProcessException; |
| |
| import org.apache.commons.io.FileUtils; |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.util.ArrayList; |
| import java.util.Iterator; |
| import java.util.List; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertTrue; |
| |
| public class TsFileManagerTest { |
| |
| File tempSGDir; |
| private TsFileManager tsFileManager; |
| private List<TsFileResource> seqResources; |
| private List<TsFileResource> unseqResources; |
| |
| @Before |
| public void setUp() throws IOException, WriteProcessException, MetadataException { |
| tempSGDir = new File(TestConstant.BASE_OUTPUT_PATH.concat("tempSG")); |
| tempSGDir.mkdirs(); |
| tsFileManager = new TsFileManager("test", "0", tempSGDir.getAbsolutePath()); |
| seqResources = new ArrayList<>(); |
| for (int i = 0; i < 5; i++) { |
| TsFileResource resource = generateTsFileResource(i); |
| seqResources.add(resource); |
| } |
| unseqResources = new ArrayList<>(); |
| for (int i = 6; i < 10; i++) { |
| TsFileResource resource = generateTsFileResource(i); |
| unseqResources.add(resource); |
| } |
| } |
| |
| @After |
| public void tearDown() throws IOException, StorageEngineException { |
| FileUtils.deleteDirectory(tempSGDir); |
| } |
| |
| private TsFileResource generateTsFileResource(int id) { |
| File file = |
| new File( |
| TsFileNameGenerator.generateNewTsFilePath( |
| TestConstant.BASE_OUTPUT_PATH, id, id, id, id)); |
| return new TsFileResource(file); |
| } |
| |
| /** just compaction once */ |
| @Test |
| public void testAddRemoveAndIterator() { |
| for (TsFileResource tsFileResource : seqResources) { |
| tsFileManager.add(tsFileResource, true); |
| } |
| tsFileManager.addAll(unseqResources, false); |
| assertEquals(5, tsFileManager.getTsFileList(true).size()); |
| assertEquals(4, tsFileManager.getTsFileList(false).size()); |
| assertEquals(5, tsFileManager.size(true)); |
| assertEquals(4, tsFileManager.size(false)); |
| assertTrue(tsFileManager.contains(seqResources.get(0), true)); |
| assertFalse( |
| tsFileManager.contains( |
| new TsFileResource( |
| new File( |
| TestConstant.getTestTsFileDir("root.compactionTest", 0, 0) |
| .concat( |
| 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))), |
| false)); |
| assertFalse( |
| tsFileManager.contains( |
| new TsFileResource( |
| new File( |
| TestConstant.getTestTsFileDir("root.compactionTest", 0, 0) |
| .concat( |
| 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))), |
| false)); |
| assertFalse(tsFileManager.isEmpty(true)); |
| assertFalse(tsFileManager.isEmpty(false)); |
| tsFileManager.remove(tsFileManager.getTsFileList(true).get(0), true); |
| tsFileManager.remove(tsFileManager.getTsFileList(false).get(0), false); |
| assertEquals(4, tsFileManager.getTsFileList(true).size()); |
| tsFileManager.removeAll(tsFileManager.getTsFileList(false), false); |
| assertEquals(0, tsFileManager.getTsFileList(false).size()); |
| long count = 0; |
| Iterator<TsFileResource> iterator = tsFileManager.getIterator(true); |
| while (iterator.hasNext()) { |
| iterator.next(); |
| count++; |
| } |
| assertEquals(4, count); |
| tsFileManager.removeAll(tsFileManager.getTsFileList(true), true); |
| assertEquals(0, tsFileManager.getTsFileList(true).size()); |
| assertTrue(tsFileManager.isEmpty(true)); |
| assertTrue(tsFileManager.isEmpty(false)); |
| tsFileManager.add( |
| new TsFileResource( |
| new File( |
| TestConstant.getTestTsFileDir("root.compactionTest", 0, 0) |
| .concat( |
| 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))), |
| true); |
| tsFileManager.add( |
| new TsFileResource( |
| new File( |
| TestConstant.getTestTsFileDir("root.compactionTest", 0, 0) |
| .concat( |
| 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))), |
| false); |
| assertEquals(1, tsFileManager.size(true)); |
| assertEquals(1, tsFileManager.size(false)); |
| tsFileManager.clear(); |
| assertEquals(0, tsFileManager.size(true)); |
| assertEquals(0, tsFileManager.size(false)); |
| } |
| |
| @Test |
| public void testIteratorRemove() { |
| for (TsFileResource tsFileResource : seqResources) { |
| tsFileManager.add(tsFileResource, true); |
| } |
| tsFileManager.addAll(seqResources, false); |
| assertEquals(5, tsFileManager.getTsFileList(true).size()); |
| |
| Iterator<TsFileResource> tsFileResourceIterator = tsFileManager.getIterator(true); |
| tsFileResourceIterator.next(); |
| try { |
| tsFileResourceIterator.remove(); |
| } catch (UnsupportedOperationException e) { |
| // pass |
| } |
| assertEquals(5, tsFileManager.getTsFileList(true).size()); |
| |
| TsFileResource tsFileResource1 = |
| new TsFileResource( |
| new File( |
| TestConstant.getTestTsFileDir("root.compactionTest", 0, 0) |
| .concat( |
| 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 10 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 1 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))); |
| TsFileResource tsFileResource2 = |
| new TsFileResource( |
| new File( |
| TestConstant.BASE_OUTPUT_PATH.concat( |
| 11 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 11 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 1 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))); |
| tsFileManager.add(tsFileResource1, true); |
| tsFileManager.add(tsFileResource2, true); |
| TsFileResource tsFileResource3 = |
| new TsFileResource( |
| new File( |
| TestConstant.BASE_OUTPUT_PATH.concat( |
| 12 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 12 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 2 |
| + IoTDBConstant.FILE_NAME_SEPARATOR |
| + 0 |
| + ".tsfile"))); |
| tsFileManager.add(tsFileResource3, true); |
| Iterator<TsFileResource> tsFileResourceIterator2 = tsFileManager.getIterator(true); |
| int count = 0; |
| while (tsFileResourceIterator2.hasNext()) { |
| count++; |
| tsFileResourceIterator2.next(); |
| } |
| assertEquals(8, count); |
| } |
| } |