소스 검색

feat: add func

许家凯 4 년 전
부모
커밋
f47f0ca89e
1개의 변경된 파일7개의 추가작업 그리고 0개의 파일을 삭제
  1. 7 0
      src/main/scala/com/winhc/bigdata/spark/utils/LoggingUtils.scala

+ 7 - 0
src/main/scala/com/winhc/bigdata/spark/utils/LoggingUtils.scala

@@ -116,6 +116,13 @@ trait LoggingUtils extends Logging {
   }
 
 
+  def dropAllPartitions(tableName: String): Unit = {
+    for (elem <- getPartitions(tableName)) {
+      sql(s"ALTER TABLE $tableName DROP IF EXISTS PARTITION(ds='$elem')")
+    }
+  }
+
+
   def getPartitions(t: String): Seq[String] = {
     val sql_s = s"show partitions " + t
     sql(sql_s).collect.toList.map(r => r.getString(0)).map(r => {