1 /*
   2  * Copyright (c) 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 import java.util.Iterator;
  29 
  30 import static java.util.Objects.requireNonNull;
  31 
  32 final class BulkSizeUpdateWriter implements BinaryRepresentationWriter {
  33 
  34     private final SizeUpdateWriter writer = new SizeUpdateWriter();
  35     private Iterator<Integer> maxSizes;
  36     private boolean writing;
  37     private boolean configured;
  38 
  39     BulkSizeUpdateWriter maxHeaderTableSizes(Iterable<Integer> sizes) {
  40         if (configured) {
  41             throw new IllegalStateException("Already configured");
  42         }
  43         requireNonNull(sizes, "sizes");
  44         maxSizes = sizes.iterator();
  45         configured = true;
  46         return this;
  47     }
  48 
  49     @Override
  50     public boolean write(HeaderTable table, ByteBuffer destination) {
  51         if (!configured) {
  52             throw new IllegalStateException("Configure first");
  53         }
  54         while (true) {
  55             if (writing) {
  56                 if (!writer.write(table, destination)) {
  57                     return false;
  58                 }
  59                 writing = false;
  60             } else if (maxSizes.hasNext()) {
  61                 writing = true;
  62                 writer.reset();
  63                 writer.maxHeaderTableSize(maxSizes.next());
  64             } else {
  65                 configured = false;
  66                 return true;
  67             }
  68         }
  69     }
  70 
  71     @Override
  72     public BulkSizeUpdateWriter reset() {
  73         maxSizes = null;
  74         writing = false;
  75         configured = false;
  76         return this;
  77     }
  78 }