package org.apache.helix.controller.stages;
   
   /*
    * 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.
   */
  
  import java.util.Collections;
  import java.util.HashMap;
  import java.util.Map;
  
  import org.apache.helix.HelixDataAccessor;
  import org.apache.helix.HelixProperty;
  import org.apache.helix.PropertyType;
  import org.apache.helix.PropertyKey.Builder;
  import org.apache.helix.model.AlertStatus;
  import org.apache.helix.model.Alerts;
  import org.apache.helix.model.HealthStat;
  import org.apache.helix.model.LiveInstance;
  import org.apache.helix.model.PersistentStats;
  
  
  public class HealthDataCache
  {
    Map<String, LiveInstance> _liveInstanceMap;
  
    Map<String, Map<String, HealthStat>> _healthStatMap;
    HealthStat _globalStats; // DON'T THINK I WILL USE THIS ANYMORE
    PersistentStats _persistentStats;
    Alerts _alerts;
    AlertStatus _alertStatus;
  
    public HealthStat getGlobalStats()
    {
      return _globalStats;
    }
  
    public PersistentStats getPersistentStats()
    {
      return _persistentStats;
    }
  
    public Alerts getAlerts()
    {
      return _alerts;
    }
  
    public AlertStatus getAlertStatus()
    {
      return _alertStatus;
    }
  
    public Map<String, HealthStat> getHealthStats(String instanceName)
    {
      Map<String, HealthStat> map = _healthStatMap.get(instanceName);
      if (map != null)
      {
        return map;
      } else
      {
        return Collections.emptyMap();
      }
    }
  
    public Map<String, LiveInstance> getLiveInstances()
    {
      return _liveInstanceMap;
    }
  
    public boolean refresh(HelixDataAccessor accessor)
    {
      Builder keyBuilder = accessor.keyBuilder();
      _liveInstanceMap = accessor.getChildValuesMap(keyBuilder.liveInstances());
  
      Map<String, Map<String, HealthStat>> hsMap = new HashMap<String, Map<String, HealthStat>>();
  
      for (String instanceName : _liveInstanceMap.keySet())
      {
        // xxx clearly getting znodes for the instance here...so get the
        // timestamp!
  
        Map<String, HealthStat> childValuesMap = accessor
            .getChildValuesMap(keyBuilder.healthReports(instanceName));
        hsMap.put(instanceName, childValuesMap);
      }
     _healthStatMap = Collections.unmodifiableMap(hsMap);
     _persistentStats = accessor.getProperty(keyBuilder.persistantStat());
     _alerts = accessor.getProperty(keyBuilder.alerts());
     _alertStatus = accessor.getProperty(keyBuilder.alertStatus());
 
     return true;
 
   }
 
 }