integration/spark/src/test/scala/org/apache/carbondata/spark/testsuite/segment/ShowSegmentTestCase.scala - carbondata - Git at Google

 /*
  * 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.carbondata.spark.testsuite.segment

 import org.apache.spark.sql.{AnalysisException, CarbonEnv, Row}
 import org.apache.spark.sql.test.util.QueryTest
 import org.scalatest.BeforeAndAfterAll

 import org.apache.carbondata.common.exceptions.sql.MalformedCarbonCommandException
 import org.apache.carbondata.core.metadata.CarbonMetadata
 import org.apache.carbondata.core.statusmanager.SegmentStatusManager

 /**
  * Test Class for SHOW SEGMENTS command
  */
 class ShowSegmentTestCase extends QueryTest with BeforeAndAfterAll {

   test("test show segment by query, success case") {
     sql("drop table if exists source")
     sql(
       """
         |create table source (age int)
         |STORED AS carbondata
         |partitioned by (name string, class string)
         |TBLPROPERTIES('AUTO_LOAD_MERGE'='true','COMPACTION_LEVEL_THRESHOLD'='2,2')
         |""".stripMargin)
     sql("insert into source select 1, 'abc1', 'classA'")
     sql("insert into source select 2, 'abc2', 'classB'")
     sql("insert into source select 3, 'abc3', 'classA'")
     sql("insert into source select 4, 'abc4', 'classB'")
     sql("insert into source select 5, 'abc5', 'classA'")
     sql("insert into source select 6, 'abc6', 'classC'")
     sql("show segments on source").show(false)

     val df = sql(s"""show segments on source""").collect()
     // validating headers
     val header = df(0).schema
     assert(header(0).name.equalsIgnoreCase("ID"))
     assert(header(1).name.equalsIgnoreCase("Status"))
     assert(header(2).name.equalsIgnoreCase("Load Start Time"))
     assert(header(3).name.equalsIgnoreCase("Load Time Taken"))
     assert(header(4).name.equalsIgnoreCase("Partition"))
     assert(header(5).name.equalsIgnoreCase("Data Size"))
     assert(header(6).name.equalsIgnoreCase("Index Size"))
     val col = df
       .map(row => Row(row.getString(0), row.getString(1)))
       .filter(_.getString(1).equals("Success"))
       .toSeq
     assert(col.equals(Seq(Row("4.1", "Success"), Row("0.2", "Success"))))

     var rows = sql(
       """
         | show segments on source as
         | select id, status, datasize from source_segments where status = 'Success' order by dataSize
         |""".stripMargin).collect()


     assertResult("4.1")(rows(0).get(0))
     assertResult("Success")(rows(0).get(1))
     assertResult("0.2")(rows(1).get(0))
     assertResult("Success")(rows(1).get(1))

     rows = sql(
       """
         | show segments on source limit 2 as
         | select id, status, datasize from source_segments where status = 'Success' order by dataSize
         |""".stripMargin).collect()

     assertResult("4.1")(rows(0).get(0))
     assertResult("Success")(rows(0).get(1))
     assertResult(1)(rows.length)

     val tables = sql("show tables").collect()
     assert(!tables.toSeq.exists(_.get(1).equals("source_segments")))

     sql(s"""drop table source""").collect
   }

   test("Show Segments on empty table") {
     sql(s"""drop TABLE if exists source""").collect
     sql(s"""CREATE TABLE source (CUST_ID int,CUST_NAME String,ACTIVE_EMUI_VERSION string,DOB timestamp, DOJ timestamp, BIGINT_COLUMN1 bigint,BIGINT_COLUMN2 bigint,DECIMAL_COLUMN1 decimal(30,10),Double_COLUMN1 double,DECIMAL_COLUMN2 decimal(36,10), Double_COLUMN2 double,INTEGER_COLUMN1 int) STORED AS carbondata TBLPROPERTIES('table_blocksize'='1')""").collect
     checkAnswer(sql("show segments on source"), Seq.empty)
     var result = sql("show segments on source as select * from source_segments").collect()
     assertResult(0)(result.length)
     result = sql("show segments on source limit 10 as select * from source_segments").collect()
     assertResult(0)(result.length)
   }

   test("test show segments on already existing table") {
     sql("drop TABLE if exists source").collect
     sql(
       """
         |create table source (age int, name string, class string)
         |STORED AS carbondata
         |""".stripMargin)
     sql("insert into source select 1, 'abc1', 'classA'")
     sql("drop table if exists source_segments")
     sql("create table source_segments (age int)")
     val ex = intercept[MalformedCarbonCommandException](sql("show segments on source as select * from source_segments"))
     assert(ex.getMessage.contains("source_segments already exists"))
     sql("drop TABLE if exists source")
     sql("drop table if exists source_segments")
   }

   test(" test show segments by wrong query") {
     sql("drop TABLE if exists source").collect
     sql(
       """
         |create table source (age int, name string, class string)
         |STORED AS carbondata
         |""".stripMargin)
     sql("insert into source select 1, 'abc1', 'classA'")
     val ex = intercept[AnalysisException](sql("show segments on source as select dsjk from source_segments"))
     val tables = sql("show tables").collect()
     assert(!tables.toSeq.exists(_.get(1).equals("source_segments")))
     sql("drop TABLE if exists source")
   }

   //Show Segments failing if table name not in same case
   test("DataLoadManagement001_830") {
     sql(s"""drop TABLE if exists Case_ShowSegment_196""").collect
     sql(s"""CREATE TABLE Case_ShowSegment_196 (CUST_ID int,CUST_NAME String,ACTIVE_EMUI_VERSION string,DOB timestamp, DOJ timestamp, BIGINT_COLUMN1 bigint,BIGINT_COLUMN2 bigint,DECIMAL_COLUMN1 decimal(30,10),Double_COLUMN1 double,DECIMAL_COLUMN2 decimal(36,10), Double_COLUMN2 double,INTEGER_COLUMN1 int) STORED AS carbondata TBLPROPERTIES('table_blocksize'='1')""").collect
     val df = sql(s"""show segments on default.CASE_ShowSegment_196""").collect()
     val col = df.map {
       row => Row(row.getString(0), row.getString(1), row.getString(4))
     }.toSeq
     assert(col.equals(Seq()))
     sql(s"""drop table Case_ShowSegment_196""").collect
   }

   test("separate visible and invisible segments info into two files") {
     val tableName = "test_tablestatus_history"
     sql(s"drop table if exists ${tableName}")
     sql(s"create table ${tableName} (name String, age int) STORED AS carbondata "
         + "TBLPROPERTIES('AUTO_LOAD_MERGE'='true','COMPACTION_LEVEL_THRESHOLD'='2,2')")
     val carbonTable = CarbonEnv.getCarbonTable(Some("default"), tableName)(sqlContext.sparkSession)
     insertTestDataIntoTable(tableName)
     assert(sql(s"show segments on ${tableName} as select * from ${tableName}_segments").collect().length == 10)
     var detail = SegmentStatusManager.readLoadMetadata(carbonTable.getMetadataPath)
     var historyDetail = SegmentStatusManager.readLoadHistoryMetadata(carbonTable.getMetadataPath)
     assert(detail.length == 10)
     assert(historyDetail.length == 0)
     sql(s"clean files for table ${tableName}")
     assert(sql(s"show segments on ${tableName}").collect().length == 2)
     assert(sql(s"show segments on ${tableName} limit 1").collect().length == 1)
     detail = SegmentStatusManager.readLoadMetadata(carbonTable.getMetadataPath)
     historyDetail = SegmentStatusManager.readLoadHistoryMetadata(carbonTable.getMetadataPath)
     assert(detail.length == 4)
     assert(historyDetail.length == 6)
     dropTable(tableName)
   }

   test("show history segments") {
     val tableName = "test_tablestatus_history"
     sql(s"drop table if exists ${tableName}")
     sql(s"create table ${tableName} (name String, age int) STORED AS carbondata "
         + "TBLPROPERTIES('AUTO_LOAD_MERGE'='true','COMPACTION_LEVEL_THRESHOLD'='2,2')")
     val carbonTable = CarbonMetadata.getInstance().getCarbonTable("default", tableName)
     insertTestDataIntoTable(tableName)
     assert(sql(s"show segments on ${tableName} as select * from ${tableName}_segments").collect().length == 10)
     assert(sql(s"show history segments on ${tableName} as select * from ${tableName}_segments").collect().length == 10)
     sql(s"clean files for table ${tableName}")
     assert(sql(s"show segments on ${tableName} as select * from ${tableName}_segments").collect().length == 2)
     sql(s"show history segments on ${tableName} as select * from ${tableName}_segments").show(false)
     var segmentsHistoryList = sql(s"show history segments on ${tableName} as select * from ${tableName}_segments").collect()
     assert(segmentsHistoryList.length == 10)
     assertResult("0")(segmentsHistoryList(0).getString(0))
     assertResult("Compacted")(segmentsHistoryList(0).getString(1))
     assertResult("0.1")(segmentsHistoryList(0).getString(7))
     assertResult("0.2")(segmentsHistoryList(1).getString(0))
     assertResult("Success")(segmentsHistoryList(1).getString(1))
     assertResult("5")(segmentsHistoryList(2).getString(0))
     assertResult("Compacted")(segmentsHistoryList(2).getString(1))
     assertResult("4.1")(segmentsHistoryList(3).getString(0))
     assertResult("Success")(segmentsHistoryList(3).getString(1))
     assertResult("1")(segmentsHistoryList(4).getString(0))
     assertResult("Compacted")(segmentsHistoryList(4).getString(1))
     assertResult("0.1")(segmentsHistoryList(4).getString(7))
     assertResult("3")(segmentsHistoryList(7).getString(0))
     assertResult("Compacted")(segmentsHistoryList(7).getString(1))
     assertResult("2.1")(segmentsHistoryList(8).getString(0))
     assertResult("Compacted")(segmentsHistoryList(8).getString(1))
     assertResult("4")(segmentsHistoryList(9).getString(0))
     assertResult("Compacted")(segmentsHistoryList(9).getString(1))

     segmentsHistoryList = sql(s"show history segments on ${tableName} limit 2 as select * from ${tableName}_segments").collect()
     assert(segmentsHistoryList.length == 2)
     assertResult("0")(segmentsHistoryList(0).getString(0))
     assertResult("Compacted")(segmentsHistoryList(0).getString(1))
     assertResult("0.1")(segmentsHistoryList(0).getString(7))
     assertResult("0.2")(segmentsHistoryList(1).getString(0))
     assertResult("Success")(segmentsHistoryList(1).getString(1))

     assert(sql(s"show history segments on ${tableName} as select * from ${tableName}_segments limit 3").collect().length == 3)
     dropTable(tableName)
   }

   test("test for load time and format name") {
     sql("drop table if exists a")
     sql("create table a(a string) stored as carbondata")
     sql("insert into a select 'k'")
     sql("insert into a select 'j'")
     sql("insert into a select 'k'")
     val rows = sql("show segments for table a").collect()
     assert(rows.length == 3)
     assert(sql(s"show segments for table a limit 1").collect().length == 1)
     assert(rows(0).getString(3).replace("S", "").toDouble > 0)
     assert(rows(0).getString(7).equalsIgnoreCase("columnar_v3"))
     sql("drop table if exists a")
   }

   private def insertTestDataIntoTable(tableName: String) = {
     sql(s"insert into ${ tableName } select 'abc1',1")
     sql(s"insert into ${ tableName } select 'abc2',2")
     sql(s"insert into ${ tableName } select 'abc3',3")
     sql(s"insert into ${ tableName } select 'abc4',4")
     sql(s"insert into ${ tableName } select 'abc5',5")
     sql(s"insert into ${ tableName } select 'abc6',6")
   }
 }
	/*
	* 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.carbondata.spark.testsuite.segment

	import org.apache.spark.sql.{AnalysisException, CarbonEnv, Row}
	import org.apache.spark.sql.test.util.QueryTest
	import org.scalatest.BeforeAndAfterAll

	import org.apache.carbondata.common.exceptions.sql.MalformedCarbonCommandException
	import org.apache.carbondata.core.metadata.CarbonMetadata
	import org.apache.carbondata.core.statusmanager.SegmentStatusManager

	/**
	* Test Class for SHOW SEGMENTS command
	*/
	class ShowSegmentTestCase extends QueryTest with BeforeAndAfterAll {

	test("test show segment by query, success case") {
	sql("drop table if exists source")
	sql(
	"""
	\|create table source (age int)
	\|STORED AS carbondata
	\|partitioned by (name string, class string)
	\|TBLPROPERTIES('AUTO_LOAD_MERGE'='true','COMPACTION_LEVEL_THRESHOLD'='2,2')
	\|""".stripMargin)
	sql("insert into source select 1, 'abc1', 'classA'")
	sql("insert into source select 2, 'abc2', 'classB'")
	sql("insert into source select 3, 'abc3', 'classA'")
	sql("insert into source select 4, 'abc4', 'classB'")
	sql("insert into source select 5, 'abc5', 'classA'")
	sql("insert into source select 6, 'abc6', 'classC'")
	sql("show segments on source").show(false)

	val df = sql(s"""show segments on source""").collect()
	// validating headers
	val header = df(0).schema
	assert(header(0).name.equalsIgnoreCase("ID"))
	assert(header(1).name.equalsIgnoreCase("Status"))
	assert(header(2).name.equalsIgnoreCase("Load Start Time"))
	assert(header(3).name.equalsIgnoreCase("Load Time Taken"))
	assert(header(4).name.equalsIgnoreCase("Partition"))
	assert(header(5).name.equalsIgnoreCase("Data Size"))
	assert(header(6).name.equalsIgnoreCase("Index Size"))
	val col = df
	.map(row => Row(row.getString(0), row.getString(1)))
	.filter(_.getString(1).equals("Success"))
	.toSeq
	assert(col.equals(Seq(Row("4.1", "Success"), Row("0.2", "Success"))))

	var rows = sql(
	"""
	\| show segments on source as
	\| select id, status, datasize from source_segments where status = 'Success' order by dataSize
	\|""".stripMargin).collect()


	assertResult("4.1")(rows(0).get(0))
	assertResult("Success")(rows(0).get(1))
	assertResult("0.2")(rows(1).get(0))
	assertResult("Success")(rows(1).get(1))

	rows = sql(
	"""
	\| show segments on source limit 2 as
	\| select id, status, datasize from source_segments where status = 'Success' order by dataSize
	\|""".stripMargin).collect()

	assertResult("4.1")(rows(0).get(0))
	assertResult("Success")(rows(0).get(1))
	assertResult(1)(rows.length)

	val tables = sql("show tables").collect()
	assert(!tables.toSeq.exists(_.get(1).equals("source_segments")))

	sql(s"""drop table source""").collect
	}

	test("Show Segments on empty table") {
	sql(s"""drop TABLE if exists source""").collect
	sql(s"""CREATE TABLE source (CUST_ID int,CUST_NAME String,ACTIVE_EMUI_VERSION string,DOB timestamp, DOJ timestamp, BIGINT_COLUMN1 bigint,BIGINT_COLUMN2 bigint,DECIMAL_COLUMN1 decimal(30,10),Double_COLUMN1 double,DECIMAL_COLUMN2 decimal(36,10), Double_COLUMN2 double,INTEGER_COLUMN1 int) STORED AS carbondata TBLPROPERTIES('table_blocksize'='1')""").collect
	checkAnswer(sql("show segments on source"), Seq.empty)
	var result = sql("show segments on source as select * from source_segments").collect()
	assertResult(0)(result.length)
	result = sql("show segments on source limit 10 as select * from source_segments").collect()
	assertResult(0)(result.length)
	}

	test("test show segments on already existing table") {
	sql("drop TABLE if exists source").collect
	sql(
	"""
	\|create table source (age int, name string, class string)
	\|STORED AS carbondata
	\|""".stripMargin)
	sql("insert into source select 1, 'abc1', 'classA'")
	sql("drop table if exists source_segments")
	sql("create table source_segments (age int)")
	val ex = intercept[MalformedCarbonCommandException](sql("show segments on source as select * from source_segments"))
	assert(ex.getMessage.contains("source_segments already exists"))
	sql("drop TABLE if exists source")
	sql("drop table if exists source_segments")
	}

	test(" test show segments by wrong query") {
	sql("drop TABLE if exists source").collect
	sql(
	"""
	\|create table source (age int, name string, class string)
	\|STORED AS carbondata
	\|""".stripMargin)
	sql("insert into source select 1, 'abc1', 'classA'")
	val ex = intercept[AnalysisException](sql("show segments on source as select dsjk from source_segments"))
	val tables = sql("show tables").collect()
	assert(!tables.toSeq.exists(_.get(1).equals("source_segments")))
	sql("drop TABLE if exists source")
	}

	//Show Segments failing if table name not in same case
	test("DataLoadManagement001_830") {
	sql(s"""drop TABLE if exists Case_ShowSegment_196""").collect
	sql(s"""CREATE TABLE Case_ShowSegment_196 (CUST_ID int,CUST_NAME String,ACTIVE_EMUI_VERSION string,DOB timestamp, DOJ timestamp, BIGINT_COLUMN1 bigint,BIGINT_COLUMN2 bigint,DECIMAL_COLUMN1 decimal(30,10),Double_COLUMN1 double,DECIMAL_COLUMN2 decimal(36,10), Double_COLUMN2 double,INTEGER_COLUMN1 int) STORED AS carbondata TBLPROPERTIES('table_blocksize'='1')""").collect
	val df = sql(s"""show segments on default.CASE_ShowSegment_196""").collect()
	val col = df.map {
	row => Row(row.getString(0), row.getString(1), row.getString(4))
	}.toSeq
	assert(col.equals(Seq()))
	sql(s"""drop table Case_ShowSegment_196""").collect
	}

	test("separate visible and invisible segments info into two files") {
	val tableName = "test_tablestatus_history"
	sql(s"drop table if exists ${tableName}")
	sql(s"create table ${tableName} (name String, age int) STORED AS carbondata "
	+ "TBLPROPERTIES('AUTO_LOAD_MERGE'='true','COMPACTION_LEVEL_THRESHOLD'='2,2')")
	val carbonTable = CarbonEnv.getCarbonTable(Some("default"), tableName)(sqlContext.sparkSession)
	insertTestDataIntoTable(tableName)
	assert(sql(s"show segments on ${tableName} as select * from ${tableName}_segments").collect().length == 10)
	var detail = SegmentStatusManager.readLoadMetadata(carbonTable.getMetadataPath)
	var historyDetail = SegmentStatusManager.readLoadHistoryMetadata(carbonTable.getMetadataPath)
	assert(detail.length == 10)
	assert(historyDetail.length == 0)
	sql(s"clean files for table ${tableName}")
	assert(sql(s"show segments on ${tableName}").collect().length == 2)
	assert(sql(s"show segments on ${tableName} limit 1").collect().length == 1)
	detail = SegmentStatusManager.readLoadMetadata(carbonTable.getMetadataPath)
	historyDetail = SegmentStatusManager.readLoadHistoryMetadata(carbonTable.getMetadataPath)
	assert(detail.length == 4)
	assert(historyDetail.length == 6)
	dropTable(tableName)
	}

	test("show history segments") {
	val tableName = "test_tablestatus_history"
	sql(s"drop table if exists ${tableName}")
	sql(s"create table ${tableName} (name String, age int) STORED AS carbondata "
	+ "TBLPROPERTIES('AUTO_LOAD_MERGE'='true','COMPACTION_LEVEL_THRESHOLD'='2,2')")
	val carbonTable = CarbonMetadata.getInstance().getCarbonTable("default", tableName)
	insertTestDataIntoTable(tableName)
	assert(sql(s"show segments on ${tableName} as select * from ${tableName}_segments").collect().length == 10)
	assert(sql(s"show history segments on ${tableName} as select * from ${tableName}_segments").collect().length == 10)
	sql(s"clean files for table ${tableName}")
	assert(sql(s"show segments on ${tableName} as select * from ${tableName}_segments").collect().length == 2)
	sql(s"show history segments on ${tableName} as select * from ${tableName}_segments").show(false)
	var segmentsHistoryList = sql(s"show history segments on ${tableName} as select * from ${tableName}_segments").collect()
	assert(segmentsHistoryList.length == 10)
	assertResult("0")(segmentsHistoryList(0).getString(0))
	assertResult("Compacted")(segmentsHistoryList(0).getString(1))
	assertResult("0.1")(segmentsHistoryList(0).getString(7))
	assertResult("0.2")(segmentsHistoryList(1).getString(0))
	assertResult("Success")(segmentsHistoryList(1).getString(1))
	assertResult("5")(segmentsHistoryList(2).getString(0))
	assertResult("Compacted")(segmentsHistoryList(2).getString(1))
	assertResult("4.1")(segmentsHistoryList(3).getString(0))
	assertResult("Success")(segmentsHistoryList(3).getString(1))
	assertResult("1")(segmentsHistoryList(4).getString(0))
	assertResult("Compacted")(segmentsHistoryList(4).getString(1))
	assertResult("0.1")(segmentsHistoryList(4).getString(7))
	assertResult("3")(segmentsHistoryList(7).getString(0))
	assertResult("Compacted")(segmentsHistoryList(7).getString(1))
	assertResult("2.1")(segmentsHistoryList(8).getString(0))
	assertResult("Compacted")(segmentsHistoryList(8).getString(1))
	assertResult("4")(segmentsHistoryList(9).getString(0))
	assertResult("Compacted")(segmentsHistoryList(9).getString(1))

	segmentsHistoryList = sql(s"show history segments on ${tableName} limit 2 as select * from ${tableName}_segments").collect()
	assert(segmentsHistoryList.length == 2)
	assertResult("0")(segmentsHistoryList(0).getString(0))
	assertResult("Compacted")(segmentsHistoryList(0).getString(1))
	assertResult("0.1")(segmentsHistoryList(0).getString(7))
	assertResult("0.2")(segmentsHistoryList(1).getString(0))
	assertResult("Success")(segmentsHistoryList(1).getString(1))

	assert(sql(s"show history segments on ${tableName} as select * from ${tableName}_segments limit 3").collect().length == 3)
	dropTable(tableName)
	}

	test("test for load time and format name") {
	sql("drop table if exists a")
	sql("create table a(a string) stored as carbondata")
	sql("insert into a select 'k'")
	sql("insert into a select 'j'")
	sql("insert into a select 'k'")
	val rows = sql("show segments for table a").collect()
	assert(rows.length == 3)
	assert(sql(s"show segments for table a limit 1").collect().length == 1)
	assert(rows(0).getString(3).replace("S", "").toDouble > 0)
	assert(rows(0).getString(7).equalsIgnoreCase("columnar_v3"))
	sql("drop table if exists a")
	}

	private def insertTestDataIntoTable(tableName: String) = {
	sql(s"insert into ${ tableName } select 'abc1',1")
	sql(s"insert into ${ tableName } select 'abc2',2")
	sql(s"insert into ${ tableName } select 'abc3',3")
	sql(s"insert into ${ tableName } select 'abc4',4")
	sql(s"insert into ${ tableName } select 'abc5',5")
	sql(s"insert into ${ tableName } select 'abc6',6")
	}
	}