/* Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */

#ifndef h2_req_shed_h
#define h2_req_shed_h

struct h2_req_engine;
struct h2_task;

typedef struct h2_ngn_shed h2_ngn_shed;
struct h2_ngn_shed {
    conn_rec *c;
    apr_pool_t *pool;
    apr_hash_t *ngns;
    void *user_ctx;
    
    unsigned int aborted : 1;
    
    int default_capacity;
    apr_size_t req_buffer_size; /* preferred buffer size for responses */
};

const char *h2_req_engine_get_id(h2_req_engine *engine);
int h2_req_engine_is_shutdown(h2_req_engine *engine);

void h2_req_engine_out_consumed(h2_req_engine *engine, conn_rec *c, 
                                apr_off_t bytes);

typedef apr_status_t h2_shed_ngn_init(h2_req_engine *engine, 
                                      const char *id, 
                                      const char *type,
                                      apr_pool_t *pool, 
                                      apr_size_t req_buffer_size,
                                      request_rec *r,
                                      h2_output_consumed **pconsumed,
                                      void **pbaton);

h2_ngn_shed *h2_ngn_shed_create(apr_pool_t *pool, conn_rec *c,
                                int default_capactiy, 
                                apr_size_t req_buffer_size); 

void h2_ngn_shed_destroy(h2_ngn_shed *shed);

void h2_ngn_shed_set_ctx(h2_ngn_shed *shed, void *user_ctx);
void *h2_ngn_shed_get_ctx(h2_ngn_shed *shed);

h2_ngn_shed *h2_ngn_shed_get_shed(struct h2_req_engine *ngn);

void h2_ngn_shed_abort(h2_ngn_shed *shed);

apr_status_t h2_ngn_shed_push_request(h2_ngn_shed *shed, const char *ngn_type, 
                                      request_rec *r, 
                                      h2_shed_ngn_init *init_cb);

apr_status_t h2_ngn_shed_pull_request(h2_ngn_shed *shed, h2_req_engine *pub_ngn, 
                                      int capacity, 
                                      int want_shutdown, request_rec **pr);

apr_status_t h2_ngn_shed_done_task(h2_ngn_shed *shed, 
                                   struct h2_req_engine *ngn, 
                                   struct h2_task *task);

void h2_ngn_shed_done_ngn(h2_ngn_shed *shed, struct h2_req_engine *ngn);


#endif /* h2_req_shed_h */