|
17 | 17 |
|
18 | 18 | package org.apache.spark.sql.hudi.command.procedures |
19 | 19 |
|
20 | | -import com.google.common.collect.ImmutableMap |
21 | | - |
22 | | -import java.util |
23 | 20 | import java.util.Locale |
24 | 21 | import java.util.function.Supplier |
25 | 22 |
|
26 | 23 | object HoodieProcedures { |
27 | | - private val BUILDERS: util.Map[String, Supplier[ProcedureBuilder]] = initProcedureBuilders |
| 24 | + private val BUILDERS: Map[String, Supplier[ProcedureBuilder]] = initProcedureBuilders |
28 | 25 |
|
29 | 26 | def newBuilder(name: String): ProcedureBuilder = { |
30 | | - val builderSupplier: Supplier[ProcedureBuilder] = BUILDERS.get(name.toLowerCase(Locale.ROOT)) |
31 | | - if (builderSupplier != null) builderSupplier.get else null |
| 27 | + val builderSupplier = BUILDERS.get(name.toLowerCase(Locale.ROOT)) |
| 28 | + if (builderSupplier.isDefined) builderSupplier.get.get() else null |
32 | 29 | } |
33 | 30 |
|
34 | | - private def initProcedureBuilders: util.Map[String, Supplier[ProcedureBuilder]] = { |
35 | | - val mapBuilder: ImmutableMap.Builder[String, Supplier[ProcedureBuilder]] = ImmutableMap.builder() |
36 | | - mapBuilder.put(RunCompactionProcedure.NAME, RunCompactionProcedure.builder) |
37 | | - mapBuilder.put(ShowCompactionProcedure.NAME, ShowCompactionProcedure.builder) |
38 | | - mapBuilder.put(CreateSavepointProcedure.NAME, CreateSavepointProcedure.builder) |
39 | | - mapBuilder.put(DeleteSavepointProcedure.NAME, DeleteSavepointProcedure.builder) |
40 | | - mapBuilder.put(RollbackToSavepointProcedure.NAME, RollbackToSavepointProcedure.builder) |
41 | | - mapBuilder.put(RollbackToInstantTimeProcedure.NAME, RollbackToInstantTimeProcedure.builder) |
42 | | - mapBuilder.put(RunClusteringProcedure.NAME, RunClusteringProcedure.builder) |
43 | | - mapBuilder.put(ShowClusteringProcedure.NAME, ShowClusteringProcedure.builder) |
44 | | - mapBuilder.put(ShowCommitsProcedure.NAME, ShowCommitsProcedure.builder) |
45 | | - mapBuilder.put(ShowCommitsMetadataProcedure.NAME, ShowCommitsMetadataProcedure.builder) |
46 | | - mapBuilder.put(ShowArchivedCommitsProcedure.NAME, ShowArchivedCommitsProcedure.builder) |
47 | | - mapBuilder.put(ShowArchivedCommitsMetadataProcedure.NAME, ShowArchivedCommitsMetadataProcedure.builder) |
48 | | - mapBuilder.put(ShowCommitFilesProcedure.NAME, ShowCommitFilesProcedure.builder) |
49 | | - mapBuilder.put(ShowCommitPartitionsProcedure.NAME, ShowCommitPartitionsProcedure.builder) |
50 | | - mapBuilder.put(ShowCommitWriteStatsProcedure.NAME, ShowCommitWriteStatsProcedure.builder) |
51 | | - mapBuilder.put(CommitsCompareProcedure.NAME, CommitsCompareProcedure.builder) |
52 | | - mapBuilder.put(ShowSavepointsProcedure.NAME, ShowSavepointsProcedure.builder) |
53 | | - mapBuilder.put(DeleteMarkerProcedure.NAME, DeleteMarkerProcedure.builder) |
54 | | - mapBuilder.put(ShowRollbacksProcedure.NAME, ShowRollbacksProcedure.builder) |
55 | | - mapBuilder.put(ShowRollbackDetailProcedure.NAME, ShowRollbackDetailProcedure.builder) |
56 | | - mapBuilder.put(ExportInstantsProcedure.NAME, ExportInstantsProcedure.builder) |
57 | | - mapBuilder.put(ShowAllFileSystemViewProcedure.NAME, ShowAllFileSystemViewProcedure.builder) |
58 | | - mapBuilder.put(ShowLatestFileSystemViewProcedure.NAME, ShowLatestFileSystemViewProcedure.builder) |
59 | | - mapBuilder.put(ShowHoodieLogFileMetadataProcedure.NAME, ShowHoodieLogFileMetadataProcedure.builder) |
60 | | - mapBuilder.put(ShowHoodieLogFileRecordsProcedure.NAME, ShowHoodieLogFileRecordsProcedure.builder) |
61 | | - mapBuilder.put(StatsWriteAmplificationProcedure.NAME, StatsWriteAmplificationProcedure.builder) |
62 | | - mapBuilder.put(StatsFileSizeProcedure.NAME, StatsFileSizeProcedure.builder) |
63 | | - mapBuilder.put(HdfsParquetImportProcedure.NAME, HdfsParquetImportProcedure.builder) |
64 | | - mapBuilder.put(RunBootstrapProcedure.NAME, RunBootstrapProcedure.builder) |
65 | | - mapBuilder.put(ShowBootstrapMappingProcedure.NAME, ShowBootstrapMappingProcedure.builder) |
66 | | - mapBuilder.put(ShowBootstrapPartitionsProcedure.NAME, ShowBootstrapPartitionsProcedure.builder) |
67 | | - mapBuilder.put(UpgradeTableProcedure.NAME, UpgradeTableProcedure.builder) |
68 | | - mapBuilder.put(DowngradeTableProcedure.NAME, DowngradeTableProcedure.builder) |
69 | | - mapBuilder.put(ShowMetadataTableFilesProcedure.NAME, ShowMetadataTableFilesProcedure.builder) |
70 | | - mapBuilder.put(ShowMetadataTablePartitionsProcedure.NAME, ShowMetadataTablePartitionsProcedure.builder) |
71 | | - mapBuilder.put(CreateMetadataTableProcedure.NAME, CreateMetadataTableProcedure.builder) |
72 | | - mapBuilder.put(DeleteMetadataTableProcedure.NAME, DeleteMetadataTableProcedure.builder) |
73 | | - mapBuilder.put(InitMetadataTableProcedure.NAME, InitMetadataTableProcedure.builder) |
74 | | - mapBuilder.put(ShowMetadataTableStatsProcedure.NAME, ShowMetadataTableStatsProcedure.builder) |
75 | | - mapBuilder.put(ValidateMetadataTableFilesProcedure.NAME, ValidateMetadataTableFilesProcedure.builder) |
76 | | - mapBuilder.put(ShowFsPathDetailProcedure.NAME, ShowFsPathDetailProcedure.builder) |
77 | | - mapBuilder.put(CopyToTableProcedure.NAME, CopyToTableProcedure.builder) |
78 | | - mapBuilder.put(RepairAddpartitionmetaProcedure.NAME, RepairAddpartitionmetaProcedure.builder) |
79 | | - mapBuilder.put(RepairCorruptedCleanFilesProcedure.NAME, RepairCorruptedCleanFilesProcedure.builder) |
80 | | - mapBuilder.put(RepairDeduplicateProcedure.NAME, RepairDeduplicateProcedure.builder) |
81 | | - mapBuilder.put(RepairMigratePartitionMetaProcedure.NAME, RepairMigratePartitionMetaProcedure.builder) |
82 | | - mapBuilder.put(RepairOverwriteHoodiePropsProcedure.NAME, RepairOverwriteHoodiePropsProcedure.builder) |
83 | | - mapBuilder.put(RunCleanProcedure.NAME, RunCleanProcedure.builder) |
84 | | - mapBuilder.put(ValidateHoodieSyncProcedure.NAME, ValidateHoodieSyncProcedure.builder) |
85 | | - mapBuilder.put(ShowInvalidParquetProcedure.NAME, ShowInvalidParquetProcedure.builder) |
86 | | - mapBuilder.build |
| 31 | + private def initProcedureBuilders: Map[String, Supplier[ProcedureBuilder]] = { |
| 32 | + Map((RunCompactionProcedure.NAME, RunCompactionProcedure.builder) |
| 33 | + ,(ShowCompactionProcedure.NAME, ShowCompactionProcedure.builder) |
| 34 | + ,(CreateSavepointProcedure.NAME, CreateSavepointProcedure.builder) |
| 35 | + ,(DeleteSavepointProcedure.NAME, DeleteSavepointProcedure.builder) |
| 36 | + ,(RollbackToSavepointProcedure.NAME, RollbackToSavepointProcedure.builder) |
| 37 | + ,(RollbackToInstantTimeProcedure.NAME, RollbackToInstantTimeProcedure.builder) |
| 38 | + ,(RunClusteringProcedure.NAME, RunClusteringProcedure.builder) |
| 39 | + ,(ShowClusteringProcedure.NAME, ShowClusteringProcedure.builder) |
| 40 | + ,(ShowCommitsProcedure.NAME, ShowCommitsProcedure.builder) |
| 41 | + ,(ShowCommitsMetadataProcedure.NAME, ShowCommitsMetadataProcedure.builder) |
| 42 | + ,(ShowArchivedCommitsProcedure.NAME, ShowArchivedCommitsProcedure.builder) |
| 43 | + ,(ShowArchivedCommitsMetadataProcedure.NAME, ShowArchivedCommitsMetadataProcedure.builder) |
| 44 | + ,(ShowCommitFilesProcedure.NAME, ShowCommitFilesProcedure.builder) |
| 45 | + ,(ShowCommitPartitionsProcedure.NAME, ShowCommitPartitionsProcedure.builder) |
| 46 | + ,(ShowCommitWriteStatsProcedure.NAME, ShowCommitWriteStatsProcedure.builder) |
| 47 | + ,(CommitsCompareProcedure.NAME, CommitsCompareProcedure.builder) |
| 48 | + ,(ShowSavepointsProcedure.NAME, ShowSavepointsProcedure.builder) |
| 49 | + ,(DeleteMarkerProcedure.NAME, DeleteMarkerProcedure.builder) |
| 50 | + ,(ShowRollbacksProcedure.NAME, ShowRollbacksProcedure.builder) |
| 51 | + ,(ShowRollbackDetailProcedure.NAME, ShowRollbackDetailProcedure.builder) |
| 52 | + ,(ExportInstantsProcedure.NAME, ExportInstantsProcedure.builder) |
| 53 | + ,(ShowAllFileSystemViewProcedure.NAME, ShowAllFileSystemViewProcedure.builder) |
| 54 | + ,(ShowLatestFileSystemViewProcedure.NAME, ShowLatestFileSystemViewProcedure.builder) |
| 55 | + ,(ShowHoodieLogFileMetadataProcedure.NAME, ShowHoodieLogFileMetadataProcedure.builder) |
| 56 | + ,(ShowHoodieLogFileRecordsProcedure.NAME, ShowHoodieLogFileRecordsProcedure.builder) |
| 57 | + ,(StatsWriteAmplificationProcedure.NAME, StatsWriteAmplificationProcedure.builder) |
| 58 | + ,(StatsFileSizeProcedure.NAME, StatsFileSizeProcedure.builder) |
| 59 | + ,(HdfsParquetImportProcedure.NAME, HdfsParquetImportProcedure.builder) |
| 60 | + ,(RunBootstrapProcedure.NAME, RunBootstrapProcedure.builder) |
| 61 | + ,(ShowBootstrapMappingProcedure.NAME, ShowBootstrapMappingProcedure.builder) |
| 62 | + ,(ShowBootstrapPartitionsProcedure.NAME, ShowBootstrapPartitionsProcedure.builder) |
| 63 | + ,(UpgradeTableProcedure.NAME, UpgradeTableProcedure.builder) |
| 64 | + ,(DowngradeTableProcedure.NAME, DowngradeTableProcedure.builder) |
| 65 | + ,(ShowMetadataTableFilesProcedure.NAME, ShowMetadataTableFilesProcedure.builder) |
| 66 | + ,(ShowMetadataTablePartitionsProcedure.NAME, ShowMetadataTablePartitionsProcedure.builder) |
| 67 | + ,(CreateMetadataTableProcedure.NAME, CreateMetadataTableProcedure.builder) |
| 68 | + ,(DeleteMetadataTableProcedure.NAME, DeleteMetadataTableProcedure.builder) |
| 69 | + ,(InitMetadataTableProcedure.NAME, InitMetadataTableProcedure.builder) |
| 70 | + ,(ShowMetadataTableStatsProcedure.NAME, ShowMetadataTableStatsProcedure.builder) |
| 71 | + ,(ValidateMetadataTableFilesProcedure.NAME, ValidateMetadataTableFilesProcedure.builder) |
| 72 | + ,(ShowFsPathDetailProcedure.NAME, ShowFsPathDetailProcedure.builder) |
| 73 | + ,(CopyToTableProcedure.NAME, CopyToTableProcedure.builder) |
| 74 | + ,(RepairAddpartitionmetaProcedure.NAME, RepairAddpartitionmetaProcedure.builder) |
| 75 | + ,(RepairCorruptedCleanFilesProcedure.NAME, RepairCorruptedCleanFilesProcedure.builder) |
| 76 | + ,(RepairDeduplicateProcedure.NAME, RepairDeduplicateProcedure.builder) |
| 77 | + ,(RepairMigratePartitionMetaProcedure.NAME, RepairMigratePartitionMetaProcedure.builder) |
| 78 | + ,(RepairOverwriteHoodiePropsProcedure.NAME, RepairOverwriteHoodiePropsProcedure.builder) |
| 79 | + ,(RunCleanProcedure.NAME, RunCleanProcedure.builder) |
| 80 | + ,(ValidateHoodieSyncProcedure.NAME, ValidateHoodieSyncProcedure.builder) |
| 81 | + ,(ShowInvalidParquetProcedure.NAME, ShowInvalidParquetProcedure.builder) |
| 82 | + ) |
87 | 83 | } |
88 | 84 | } |
0 commit comments