1 /*
   2  * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 package jdk.incubator.http.internal.hpack;
  26 
  27 import java.nio.ByteBuffer;
  28 
  29 abstract class IndexNameValueWriter implements BinaryRepresentationWriter {
  30 
  31     private final int pattern;
  32     private final int prefix;
  33     private final IntegerWriter intWriter = new IntegerWriter();
  34     private final StringWriter nameWriter = new StringWriter();
  35     private final StringWriter valueWriter = new StringWriter();
  36 
  37     protected boolean indexedRepresentation;
  38 
  39     private static final int NEW               = 0;
  40     private static final int NAME_PART_WRITTEN = 1;
  41     private static final int VALUE_WRITTEN     = 2;
  42 
  43     private int state = NEW;
  44 
  45     protected IndexNameValueWriter(int pattern, int prefix) {
  46         this.pattern = pattern;
  47         this.prefix = prefix;
  48     }
  49 
  50     IndexNameValueWriter index(int index) {
  51         indexedRepresentation = true;
  52         intWriter.configure(index, prefix, pattern);
  53         return this;
  54     }
  55 
  56     IndexNameValueWriter name(CharSequence name, boolean useHuffman) {
  57         indexedRepresentation = false;
  58         intWriter.configure(0, prefix, pattern);
  59         nameWriter.configure(name, useHuffman);
  60         return this;
  61     }
  62 
  63     IndexNameValueWriter value(CharSequence value, boolean useHuffman) {
  64         valueWriter.configure(value, useHuffman);
  65         return this;
  66     }
  67 
  68     @Override
  69     public boolean write(HeaderTable table, ByteBuffer destination) {
  70         if (state < NAME_PART_WRITTEN) {
  71             if (indexedRepresentation) {
  72                 if (!intWriter.write(destination)) {
  73                     return false;
  74                 }
  75             } else {
  76                 if (!intWriter.write(destination) || !nameWriter.write(destination)) {
  77                     return false;
  78                 }
  79             }
  80             state = NAME_PART_WRITTEN;
  81         }
  82         if (state < VALUE_WRITTEN) {
  83             if (!valueWriter.write(destination)) {
  84                 return false;
  85             }
  86             state = VALUE_WRITTEN;
  87         }
  88         return state == VALUE_WRITTEN;
  89     }
  90 
  91     @Override
  92     public IndexNameValueWriter reset() {
  93         intWriter.reset();
  94         if (!indexedRepresentation) {
  95             nameWriter.reset();
  96         }
  97         valueWriter.reset();
  98         state = NEW;
  99         return this;
 100     }
 101 }