diff --git a/src/main/scala/com/cloudera/sa/examples/tablestats/model/TopNList.scala b/src/main/scala/com/cloudera/sa/examples/tablestats/model/TopNList.scala index 3913e39..caf0584 100644 --- a/src/main/scala/com/cloudera/sa/examples/tablestats/model/TopNList.scala +++ b/src/main/scala/com/cloudera/sa/examples/tablestats/model/TopNList.scala @@ -11,13 +11,14 @@ class TopNList(val maxSize:Int) extends Serializable { var lowestValue = Long.MaxValue def add(newValue:Any, newCount:Long): Unit = { - if (topNCountsForColumnArray.length < maxSize -1) { + if (topNCountsForColumnArray.length < maxSize - 1) { + topNCountsForColumnArray += ((newValue, newCount)) + } else if (topNCountsForColumnArray.length == maxSize - 1) { topNCountsForColumnArray += ((newValue, newCount)) - } else if (topNCountsForColumnArray.length == maxSize) { updateLowestValue } else { if (newCount > lowestValue) { - topNCountsForColumnArray.insert(lowestColumnCountIndex, (newValue, newCount)) + topNCountsForColumnArray.update(lowestColumnCountIndex, (newValue, newCount)) updateLowestValue } }