core/src/main/java/org/apache/hadoop/chukwa/rest/bean/ViewBean.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.nio.charset.Charset;
import java.text.ParseException;

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.json.simple.parser.JSONParser;
import org.apache.hadoop.chukwa.util.ExceptionUtil;

@XmlRootElement
@XmlType
public class ViewBean {
  private PagesBean[] pages;
  private String description;
  private String owner;
  
  private String name;
  private String permissionType;
  private static Log log = LogFactory.getLog(ViewBean.class);

  public ViewBean() {  
  }
  
  public ViewBean(byte[] buffer) throws ParseException {
    JSONParser parser = new JSONParser();
    try {
      JSONObject json = (JSONObject) parser.parse(new String(buffer, Charset.forName("UTF-8")));
      if(json.containsKey("description")) {
        this.description = (String) json.get("description");
      } else {
        this.description = "";
      }
      this.owner= (String) json.get("owner");
      this.name= (String) json.get("name");
      this.permissionType= (String) json.get("permissionType");
      int size = ((JSONArray) json.get("pages")).size();
      PagesBean[] pages = new PagesBean[size];
      JSONArray pagesArray = (JSONArray) json.get("pages");
      for(int i=0;i<size;i++) {
        pages[i] = new PagesBean((JSONObject) pagesArray.get(i));
      }
      this.pages=pages;
    } catch (Exception e) {
      log.error(ExceptionUtil.getStackTrace(e));
      throw new ParseException(ExceptionUtil.getStackTrace(e), 0);
    }
  }
  
  @XmlElement
  public PagesBean[] getPages() {
    return pages.clone();
  }
  
  @XmlElement
  public int getPagesCount() {
    return pages.length;  
  }
  
  @XmlElement
  public String getDescription() {
    return this.description;
  }
  
  @XmlElement
  public String getOwner() {
    return this.owner;
  }
  
  @XmlElement
  public String getName() {
    return this.name;
  }
  
  @XmlElement
  public String getPermissionType() {
    return this.permissionType;
  }
  
  public void setPages(PagesBean[] pages) {
    this.pages = (PagesBean[]) pages.clone();
  }
  
  public void setDescription(String description) {
    this.description = description;
  }
  
  public void setOwner(String owner) {
    this.owner = owner;
  }
  
  public void setName(String name) {
    this.name = name;
  }
  
  public void setPermissionType(String permissionType) {
    this.permissionType = permissionType;
  }
  
  public void update() {
    if(this.pages!=null) {
      for(PagesBean page : pages) {
        page.update();
      }
    }
  }
  
  @SuppressWarnings("unchecked")
  public JSONObject deserialize() {
    update();
    JSONObject view = new JSONObject();
    try {
      view.put("name", this.name);
      view.put("owner", this.owner);
      view.put("permissionType", this.permissionType);
      view.put("description", this.description);
      JSONArray ja = new JSONArray();
      for(int i=0;i<this.pages.length;i++) {
        ja.add(this.pages[i].deserialize());
      }
      view.put("pages", (JSONArray) ja);
    } catch (Exception e){
      log.error(ExceptionUtil.getStackTrace(e));
    }
    return view;
  }
}