content: /*
   * Copyright (C) 2012 Facebook, Inc.
   *
   * Licensed 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 io.airlift.drift.codec;
  
  import io.airlift.drift.codec.internal.ProtocolReader;
  import io.airlift.drift.codec.internal.ProtocolWriter;
  import io.airlift.drift.codec.metadata.ThriftType;
  import io.airlift.drift.protocol.TProtocolReader;
  import io.airlift.drift.protocol.TProtocolWriter;
  
  public class OneOfEverythingThriftCodec
          implements ThriftCodec<OneOfEverything>
  {
      private final ThriftType type;
      private final ThriftCodec<BonkField> aStructCodec;
      private final ThriftCodec<UnionField> aUnionCodec;
      private final ThriftCodec<Fruit> aFruitCodec;
  
      public OneOfEverythingThriftCodec(ThriftType type,
              ThriftCodec<BonkField> aStructCodec,
              ThriftCodec<UnionField> aUnionCodec,
              ThriftCodec<Fruit> aFruitCodec)
      {
          this.type = type;
          this.aStructCodec = aStructCodec;
          this.aUnionCodec = aUnionCodec;
          this.aFruitCodec = aFruitCodec;
      }
  
      @Override
      public ThriftType getType()
      {
          return type;
      }
  
      @Override
      public OneOfEverything read(TProtocolReader protocol)
              throws Exception
      {
          ProtocolReader reader = new ProtocolReader(protocol);
  
          boolean aBoolean = false;
          byte aByte = 0;
          short aShort = 0;
          int aInt = 0;
          long aLong = 0;
          double aDouble = 0;
          String aString = null;
          BonkField aStruct = null;
          Fruit aEnum = null;
          UnionField aUnion = null;
  
          reader.readStructBegin();
  
          while (reader.nextField()) {
              switch (reader.getFieldId()) {
                  case 1:
                      aBoolean = reader.readBoolField();
                      break;
                  case 2:
                      aByte = reader.readByteField();
                      break;
                  case 3:
                      aShort = reader.readI16Field();
                      break;
                  case 4:
                      aInt = reader.readI32Field();
                      break;
                  case 5:
                      aLong = reader.readI64Field();
                      break;
                  case 6:
                      aDouble = reader.readDoubleField();
                      break;
                  case 7:
                      aString = reader.readStringField();
                      break;
                  case 8:
                      aStruct = reader.readStructField(aStructCodec);
                      break;
                  case 9:
                      aEnum = reader.readEnumField(aFruitCodec);
                      break;
                  case 60:
                      aUnion = reader.readStructField(aUnionCodec);
                      break;
                  default:
                      reader.skipFieldData();
              }
          }
          reader.readStructEnd();
  
          OneOfEverything oneOfEverything = new OneOfEverything();
          oneOfEverything.aBoolean = aBoolean;
          oneOfEverything.aByte = aByte;
          oneOfEverything.aShort = aShort;
          oneOfEverything.aInt = aInt;
          oneOfEverything.aLong = aLong;
          oneOfEverything.aDouble = aDouble;
          oneOfEverything.aString = aString;
          oneOfEverything.aStruct = aStruct;
          oneOfEverything.aEnum = aEnum;
          oneOfEverything.aUnion = aUnion;
  
          return oneOfEverything;
      }
  
      @Override
      public void write(OneOfEverything oneOfEverything, TProtocolWriter protocol)
              throws Exception
      {
          ProtocolWriter writer = new ProtocolWriter(protocol);
  
          writer.writeStructBegin("OneOfEverything");
          writer.writeBoolField("aBoolean", (short) 1, oneOfEverything.aBoolean);
          writer.writeByteField("aByte", (short) 2, oneOfEverything.aByte);
          writer.writeI16Field("aShort", (short) 3, oneOfEverything.aShort);
          writer.writeI32Field("aInt", (short) 4, oneOfEverything.aInt);
          writer.writeI64Field("aLong", (short) 5, oneOfEverything.aLong);
          writer.writeDoubleField("aDouble", (short) 6, oneOfEverything.aDouble);
          writer.writeStringField("aString", (short) 7, oneOfEverything.aString);
          writer.writeStructField("aStruct", (short) 8, aStructCodec, oneOfEverything.aStruct);
          writer.writeEnumField("aEnum", (short) 9, aFruitCodec, oneOfEverything.aEnum);
          writer.writeStructField("aUnion", (short) 61, aUnionCodec, oneOfEverything.aUnion);
          writer.writeStructEnd();
      }
  }
  