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 BonkFieldThriftCodec
          implements ThriftCodec<BonkField>
  {
      private final ThriftType type;
  
      public BonkFieldThriftCodec(ThriftType type)
      {
          this.type = type;
      }
  
      @Override
      public ThriftType getType()
      {
          return type;
      }
  
      @Override
      public BonkField read(TProtocolReader protocol)
              throws Exception
      {
          ProtocolReader reader = new ProtocolReader(protocol);
  
          String message = null;
          int type = 0;
  
          reader.readStructBegin();
  
          while (reader.nextField()) {
              switch (reader.getFieldId()) {
                  case 1:
                      message = reader.readStringField();
                      break;
                  case 2:
                      type = reader.readI32Field();
                      break;
                  default:
                      reader.skipFieldData();
              }
          }
          reader.readStructEnd();
  
          BonkField bonkField = new BonkField();
          if (message != null) {
              bonkField.message = message;
          }
          bonkField.type = type;
  
          return bonkField;
      }
  
      @Override
      public void write(BonkField value, TProtocolWriter protocol)
              throws Exception
      {
          ProtocolWriter writer = new ProtocolWriter(protocol);
  
          writer.writeStructBegin("bonk");
  
          String message = value.message;
          if (message != null) {
              writer.writeStringField("message", (short) 1, message);
          }
  
          writer.writeI32Field("type", (short) 2, value.type);
          writer.writeStructEnd();
      }
  }
  