src/main/java/org/apache/hadoop/chukwa/extraction/engine/ChukwaRecord.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.extraction.engine;


import java.nio.charset.Charset;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

import org.apache.hadoop.record.Buffer;

public class ChukwaRecord extends ChukwaRecordJT implements Record {
  public ChukwaRecord() {
  }

  public void add(String key, String value) {
    synchronized (this) {
      if (this.mapFields == null) {
        this.mapFields = new TreeMap<String, Buffer>();
      }
    }
    this.mapFields.put(key, new Buffer(value.getBytes(Charset.forName("UTF-8"))));
  }

  public String[] getFields() {
    return this.mapFields.keySet().toArray(new String[0]);
  }

  public String getValue(String field) {
    if (this.mapFields.containsKey(field)) {
      return new String(this.mapFields.get(field).get(), Charset.forName("UTF-8"));
    } else {
      return null;
    }
  }

  public boolean containsField(String field) {
    return this.mapFields.containsKey(field);
  }

  public void removeValue(String field) {
    if (this.mapFields.containsKey(field)) {
      this.mapFields.remove(field);
    }
  }

  @Override
  public String toString() {
    Set<Map.Entry<String, Buffer>> f = this.mapFields.entrySet();
    Iterator<Map.Entry<String, Buffer>> it = f.iterator();

    Map.Entry<String, Buffer> entry = null;
    StringBuilder sb = new StringBuilder();
    sb.append("<event  ");
    StringBuilder body = new StringBuilder();

    String key = null;
    String val = null;
    boolean hasBody = false;
    String bodyVal = null;
    while (it.hasNext()) {
      entry = it.next();
      key = entry.getKey().intern();
      val = new String(entry.getValue().get(), Charset.forName("UTF-8"));
      if (key.intern() == Record.bodyField.intern()) {
        hasBody = true;
        bodyVal = val;
      } else {
        sb.append(key).append("=\"").append(val).append("\" ");
        body.append(key).append(" = ").append(val).append("<br>");
      }

    }
    if (hasBody) {
      sb.append(">").append(bodyVal);
    } else {
      sb.append(">").append(body);
    }
    sb.append("</event>");

    return sb.toString();
  }

}