StructTypeID.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:5k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.record.meta;
  19. import java.io.IOException;
  20. import java.util.*;
  21. import org.apache.hadoop.record.RecordInput;
  22. import org.apache.hadoop.record.RecordOutput;
  23. /** 
  24.  * Represents typeID for a struct 
  25.  */
  26. public class StructTypeID extends TypeID {
  27.   private ArrayList<FieldTypeInfo> typeInfos = new ArrayList<FieldTypeInfo>();
  28.   
  29.   StructTypeID() {
  30.     super(RIOType.STRUCT);
  31.   }
  32.   
  33.   /**
  34.    * Create a StructTypeID based on the RecordTypeInfo of some record
  35.    */
  36.   public StructTypeID(RecordTypeInfo rti) {
  37.     super(RIOType.STRUCT);
  38.     typeInfos.addAll(rti.getFieldTypeInfos());
  39.   }
  40.   void add (FieldTypeInfo ti) {
  41.     typeInfos.add(ti);
  42.   }
  43.   
  44.   public Collection<FieldTypeInfo> getFieldTypeInfos() {
  45.     return typeInfos;
  46.   }
  47.   
  48.   /* 
  49.    * return the StructTypeiD, if any, of the given field 
  50.    */
  51.   StructTypeID findStruct(String name) {
  52.     // walk through the list, searching. Not the most efficient way, but this
  53.     // in intended to be used rarely, so we keep it simple. 
  54.     // As an optimization, we can keep a hashmap of record name to its RTI, for later.
  55.     for (FieldTypeInfo ti : typeInfos) {
  56.       if ((0 == ti.getFieldID().compareTo(name)) && (ti.getTypeID().getTypeVal() == RIOType.STRUCT)) {
  57.         return (StructTypeID) ti.getTypeID();
  58.       }
  59.     }
  60.     return null;
  61.   }
  62.   
  63.   void write(RecordOutput rout, String tag) throws IOException {
  64.     rout.writeByte(typeVal, tag);
  65.     writeRest(rout, tag);
  66.   }
  67.   /* 
  68.    * Writes rest of the struct (excluding type value).
  69.    * As an optimization, this method is directly called by RTI 
  70.    * for the top level record so that we don't write out the byte
  71.    * indicating that this is a struct (since top level records are
  72.    * always structs).
  73.    */
  74.   void writeRest(RecordOutput rout, String tag) throws IOException {
  75.     rout.writeInt(typeInfos.size(), tag);
  76.     for (FieldTypeInfo ti : typeInfos) {
  77.       ti.write(rout, tag);
  78.     }
  79.   }
  80.   /* 
  81.    * deserialize ourselves. Called by RTI. 
  82.    */
  83.   void read(RecordInput rin, String tag) throws IOException {
  84.     // number of elements
  85.     int numElems = rin.readInt(tag);
  86.     for (int i=0; i<numElems; i++) {
  87.       typeInfos.add(genericReadTypeInfo(rin, tag));
  88.     }
  89.   }
  90.   
  91.   // generic reader: reads the next TypeInfo object from stream and returns it
  92.   private FieldTypeInfo genericReadTypeInfo(RecordInput rin, String tag) throws IOException {
  93.     String fieldName = rin.readString(tag);
  94.     TypeID id = genericReadTypeID(rin, tag);
  95.     return new FieldTypeInfo(fieldName, id);
  96.   }
  97.   
  98.   // generic reader: reads the next TypeID object from stream and returns it
  99.   private TypeID genericReadTypeID(RecordInput rin, String tag) throws IOException {
  100.     byte typeVal = rin.readByte(tag);
  101.     switch (typeVal) {
  102.     case TypeID.RIOType.BOOL: 
  103.       return TypeID.BoolTypeID;
  104.     case TypeID.RIOType.BUFFER: 
  105.       return TypeID.BufferTypeID;
  106.     case TypeID.RIOType.BYTE:
  107.       return TypeID.ByteTypeID;
  108.     case TypeID.RIOType.DOUBLE:
  109.       return TypeID.DoubleTypeID;
  110.     case TypeID.RIOType.FLOAT:
  111.       return TypeID.FloatTypeID;
  112.     case TypeID.RIOType.INT: 
  113.       return TypeID.IntTypeID;
  114.     case TypeID.RIOType.LONG:
  115.       return TypeID.LongTypeID;
  116.     case TypeID.RIOType.MAP:
  117.     {
  118.       TypeID tIDKey = genericReadTypeID(rin, tag);
  119.       TypeID tIDValue = genericReadTypeID(rin, tag);
  120.       return new MapTypeID(tIDKey, tIDValue);
  121.     }
  122.     case TypeID.RIOType.STRING: 
  123.       return TypeID.StringTypeID;
  124.     case TypeID.RIOType.STRUCT: 
  125.     {
  126.       StructTypeID stID = new StructTypeID();
  127.       int numElems = rin.readInt(tag);
  128.       for (int i=0; i<numElems; i++) {
  129.         stID.add(genericReadTypeInfo(rin, tag));
  130.       }
  131.       return stID;
  132.     }
  133.     case TypeID.RIOType.VECTOR: 
  134.     {
  135.       TypeID tID = genericReadTypeID(rin, tag);
  136.       return new VectorTypeID(tID);
  137.     }
  138.     default:
  139.       // shouldn't be here
  140.       throw new IOException("Unknown type read");
  141.     }
  142.   }
  143. }