Repository: ambari Updated Branches: refs/heads/branch-1.7.0 995e0889b -> 41e4a8e90
AMBARI-7950. NPE after upgrade to 1.6.0 to 1.7.0.(vbrodetskyi) Project: http://git-wip-us.apache.org/repos/asf/ambari/repo Commit: http://git-wip-us.apache.org/repos/asf/ambari/commit/41e4a8e9 Tree: http://git-wip-us.apache.org/repos/asf/ambari/tree/41e4a8e9 Diff: http://git-wip-us.apache.org/repos/asf/ambari/diff/41e4a8e9 Branch: refs/heads/branch-1.7.0 Commit: 41e4a8e900aba32ea5f556459175ce2a7eb99829 Parents: 995e088 Author: Vitaly Brodetskyi <[email protected]> Authored: Fri Oct 24 17:56:55 2014 +0300 Committer: Vitaly Brodetskyi <[email protected]> Committed: Fri Oct 24 17:56:55 2014 +0300 ---------------------------------------------------------------------- .../server/upgrade/UpgradeCatalog170.java | 21 +++++++------------- .../server/upgrade/UpgradeCatalog170Test.java | 6 +++--- 2 files changed, 10 insertions(+), 17 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/ambari/blob/41e4a8e9/ambari-server/src/main/java/org/apache/ambari/server/upgrade/UpgradeCatalog170.java ---------------------------------------------------------------------- diff --git a/ambari-server/src/main/java/org/apache/ambari/server/upgrade/UpgradeCatalog170.java b/ambari-server/src/main/java/org/apache/ambari/server/upgrade/UpgradeCatalog170.java index 00b52dc..c756a29 100644 --- a/ambari-server/src/main/java/org/apache/ambari/server/upgrade/UpgradeCatalog170.java +++ b/ambari-server/src/main/java/org/apache/ambari/server/upgrade/UpgradeCatalog170.java @@ -652,23 +652,16 @@ public class UpgradeCatalog170 extends AbstractUpgradeCatalog { } private void removeMapred2Log4jConfig() { - ConfigHelper configHelper = injector.getInstance(ConfigHelper.class); - - AmbariManagementController ambariManagementController = injector.getInstance( - AmbariManagementController.class); - Clusters clusters = ambariManagementController.getClusters(); - if (clusters == null) { - return; - } - Map<String, Cluster> clusterMap = clusters.getClusters(); + final ClusterDAO clusterDAO = injector.getInstance(ClusterDAO.class); - if (clusterMap != null && !clusterMap.isEmpty()) { - for (final Cluster cluster : clusterMap.values()) { - Config config = cluster.getDesiredConfigByType(Configuration.MAPREDUCE2_LOG4J_CONFIG_TAG); - if (config != null) { - configHelper.removeConfigsByType(cluster, Configuration.MAPREDUCE2_LOG4J_CONFIG_TAG); + List<ClusterEntity> clusters = clusterDAO.findAll(); + for (ClusterEntity cluster : clusters) { + for (ClusterConfigMappingEntity configMapping : cluster.getConfigMappingEntities()) { + if (configMapping.getType().equals(Configuration.MAPREDUCE2_LOG4J_CONFIG_TAG)) { + configMapping.setSelected(0); } } + clusterDAO.merge(cluster); } } http://git-wip-us.apache.org/repos/asf/ambari/blob/41e4a8e9/ambari-server/src/test/java/org/apache/ambari/server/upgrade/UpgradeCatalog170Test.java ---------------------------------------------------------------------- diff --git a/ambari-server/src/test/java/org/apache/ambari/server/upgrade/UpgradeCatalog170Test.java b/ambari-server/src/test/java/org/apache/ambari/server/upgrade/UpgradeCatalog170Test.java index fbb3eca..acda6a6 100644 --- a/ambari-server/src/test/java/org/apache/ambari/server/upgrade/UpgradeCatalog170Test.java +++ b/ambari-server/src/test/java/org/apache/ambari/server/upgrade/UpgradeCatalog170Test.java @@ -515,7 +515,7 @@ public class UpgradeCatalog170Test { expect(amc.getClusters()).andReturn(clusters).anyTimes(); expect(clusters.getClusters()).andReturn(clustersMap).anyTimes(); expect(clusters.getClusterById(1L)).andReturn(clustersMap.values().iterator().next()).anyTimes(); - expect(clusters.getClusters()).andReturn(clustersMap).times(2); + expect(clusters.getClusters()).andReturn(clustersMap).times(1); expect(cluster.getDesiredConfigByType("global")).andReturn(config).anyTimes(); expect(cluster.getDesiredConfigByType("oozie-log4j")).andReturn(config).anyTimes(); expect(cluster.getClusterId()).andReturn(1L); @@ -568,7 +568,7 @@ public class UpgradeCatalog170Test { expect(configGroupConfigMappingDAO.findAll()).andReturn(configGroupConfigMappingEntities).once(); expect(userDAO.findAll()).andReturn(Collections.<UserEntity> emptyList()).times(1); expect(userDAO.findAll()).andReturn(Arrays.asList(userEntity1, userEntity2)).times(1); - expect(clusterDAO.findAll()).andReturn(Collections.<ClusterEntity> emptyList()).times(1); + expect(clusterDAO.findAll()).andReturn(Collections.<ClusterEntity> emptyList()).times(2); String yarnConfig = String.format("{'%s':'%s', '%s':'%s'}", YARN_TIMELINE_SERVICE_WEBAPP_ADDRESS_PROPERTY, "timeline:8081", @@ -594,7 +594,7 @@ public class UpgradeCatalog170Test { ClusterStateEntity clusterStateEntity = createNiceMock(ClusterStateEntity.class); expect(clusterEntity.getClusterId()).andReturn(1L).anyTimes(); - expect(clusterEntity.getConfigMappingEntities()).andReturn(Collections.singleton(configMappingEntity)); + expect(clusterEntity.getConfigMappingEntities()).andReturn(Collections.singleton(configMappingEntity)).times(2); expect(clusterEntity.getClusterStateEntity()).andReturn(clusterStateEntity).anyTimes(); expect(clusterStateEntity.getCurrentStackVersion()).andReturn(CLUSTER_STATE_STACK_HDP_2_1); expect(configMappingEntity.getType()).andReturn(YARN_SITE).anyTimes();
