src/main/java/org/apache/hadoop/chukwa/rest/bean/WidgetBean.java

/*
 * 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.
 */

package org.apache.hadoop.chukwa.rest.bean;

import java.text.ParseException;
import java.util.Collection;

import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.bind.annotation.XmlType;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;

import org.apache.hadoop.chukwa.datastore.WidgetStore;
import org.apache.hadoop.chukwa.util.ExceptionUtil;

@XmlRootElement
@XmlType(propOrder={"id", "title", "version", "categories", "url", "description","refresh","parameters"})
public class WidgetBean {
  private String id;
  private String title;
  private String version;
  private String categories;
  private String url;
  private String description;
  private int refresh;
  private ParametersBean[] parameters;
  private static Log log = LogFactory.getLog(WidgetBean.class);

  public WidgetBean() {
    
  }
  
  public WidgetBean(JSONObject json) throws ParseException {
    try {
      this.id=(String) json.get("id");
      this.title=(String) json.get("title");
      this.version=(String) json.get("version");
      this.categories=(String) json.get("categories");
      this.url=(String) json.get("url");
      this.description=(String) json.get("description");
      if(json.get("refresh").getClass().getName().equals("String")) {
        int refresh = Integer.parseInt((String) json.get("refresh"));
        this.refresh = refresh;
      } else if(json.get("refresh").getClass().getName().equals("Long")) {
        this.refresh = ((Long) json.get("refresh")).intValue();
      }
      try {
        int size = ((JSONArray) json.get("parameters")).size();
        ParametersBean[] list = new ParametersBean[size];
        for(int i=0;i<size;i++) {
          JSONArray jsonArray = (JSONArray) json.get("parameters");
          list[i] = new ParametersBean((JSONObject) jsonArray.get(i));
        }
        this.parameters=list;
      } catch (Exception e) {
        this.parameters=null;
      }
    } catch (Exception e) {
      log.error(ExceptionUtil.getStackTrace(e));
      throw new ParseException(ExceptionUtil.getStackTrace(e), 0);
    }
  }
  
  @XmlElement
  public String getId() {
    return id;
  }

  @XmlElement
  public String getTitle() {
    return title;
  }
  
  @XmlElement
  public String getVersion() {
    return version;
  }
  
  @XmlElement
  public String getCategories() {
    return categories;
  }
  
  @XmlElement
  public String getUrl() {
    return url;
  }
  
  @XmlElement
  public String getDescription() {
    return description;
  }
  
  @XmlElement
  public int getRefresh() {
    return refresh;
  }
  
  @XmlElement
  public ParametersBean[] getParameters() {
    return parameters.clone();
  }
  
  public void setId(String id) {
    this.id=id;
  }
  
  public void setUrl(String url) {
    this.url=url;
  }
  
  public void setTitle(String title) {
    this.title=title;
  }
  
  public void setDescription(String description) {
    this.description=description;
  }
  
  public void setVersion(String version) {
    this.version=version;
  }
  
  public void setCategories(String categories) {
    this.categories=categories;
  }
  
  public void setRefresh(int refresh) {
    this.refresh=refresh;
  }
  
  public void setParameters(ParametersBean[] p) {
    this.parameters=p.clone();
  }
  
  public void update() {
    try {
      WidgetBean widget = WidgetStore.list().get(this.id);
      if(widget!=null) {
        if(widget.getVersion().intern()!=this.version.intern()) {
          this.categories=widget.getCategories();
          this.title=widget.getTitle();
          this.version=widget.getVersion();
          this.url=widget.getUrl();
          this.description=widget.getDescription();
          ParametersBean[] plist = widget.getParameters();
          for(int i=0;i<this.parameters.length;i++) {
            Collection<String> value = this.parameters[i].getValue();
            for(int j=0;j<plist.length;j++) {
              if(plist[j].getName().intern()==this.parameters[i].getName().intern()) {
                plist[j].setValue(value);
              }
            }
          }
          this.parameters=plist;
        }
      } else {
        log.info("Widget "+this.id+" is deprecated.");
      }
    } catch (IllegalAccessException e) {
      log.error("Unable to update widget: "+this.id+" "+ExceptionUtil.getStackTrace(e));
    }
    
  }
  
  public JSONObject deserialize() {
    JSONObject json = new JSONObject();
    try {
      json.put("id", this.id);
      json.put("title", this.title);
      json.put("description", this.description);
      json.put("version", this.version);
      json.put("categories", this.categories);
      json.put("refresh", this.refresh);
      json.put("url", this.url);
      JSONArray ja = new JSONArray();
      if(this.parameters!=null) {
        for(int i=0;i<this.parameters.length;i++) {
          ja.add(this.parameters[i].deserialize());
        }
      }
      json.put("parameters", (JSONArray) ja);
    } catch (Exception e) {
      log.error(ExceptionUtil.getStackTrace(e));
    }
    return json;    
  }
}