/*
 * 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.
 */

/*-------------------------------------------------------------------------
 *
 * cdbdatalocality.h
 *	  Manages data locality.
 *
 *-------------------------------------------------------------------------
 */
#ifndef CDBDATALOCALITY_H
#define CDBDATALOCALITY_H

#include "postgres.h"

#include "catalog/gp_policy.h"
#include "nodes/parsenodes.h"
#include "executor/execdesc.h"
#include "catalog/pg_exttable.h"
#include "magma/cwrapper/magma-client-c.h"
#include "access/extprotocol.h"


/*
 * structure containing information about data residence
 * at the host.
 */
typedef struct SplitAllocResult
{
  QueryResource *resource;
  List *alloc_results;
  int planner_segments;
  List *relsType;// relation type after datalocality changing
  StringInfo datalocalityInfo;
  double datalocalityTime;
  char *hiveUrl;
} SplitAllocResult;

/*
 * structure containing all relation range table entries.
 */
typedef struct udf_collector_context {
	bool udf_exist;
} udf_collector_context;

/*
 * structure containing rel and type when execution
 */
typedef struct CurrentRelType {
	Oid relid;
	bool isHash;
	uint16_t range_num;
} CurrentRelType;

/*
 * structure for virtual segment.
 */
typedef struct VirtualSegmentNode
{
	NodeTag type;
	char *hostname;
} VirtualSegmentNode;

typedef struct blocklocation_file{
	BlockLocation *locations;
	int block_num;
	char *file_uri;
}blocklocation_file;

/*
 * calculate_planner_segment_num: based on the parse tree,
 * we calculate the appropriate planner segment_num.
 */
SplitAllocResult * calculate_planner_segment_num(PlannedStmt *plannedstmt, Query *query,
		QueryResourceLife resourceLife, int fixedVsegNum);

/*
 * set_magma_range_rg_map: set up the range and replica maps.
 */
void set_magma_range_vseg_map(List *SegFileSplitMaps, int nvseg);

/*
 * find the range and virtual segment map by nvseg
 */
void get_magma_range_vseg_map(int **map, int *nmap, int nvseg);

/*
 * get filesplits of magma table for insert
 */
List* get_magma_scansplits(List *all_relids);
void fetch_magma_result_splits_from_plan(List **alloc_result, PlannedStmt* plannedstmt, int vsegNum);
void build_magma_scansplits_for_result_relations(List **alloc_result, List *relOids, int vsegNum);

/*
 * udf_collector_walker: the routine to file udfs.
 */
bool udf_collector_walker(Node *node,	udf_collector_context *context);

/*
 * find_udf: collect all udf, and store them into the udf_collector_context.
 */
void find_udf(Query *query, udf_collector_context *context);


/* used for magma analyze*/
Oid LookupCustomProtocolBlockLocationFunc(char *protoname);
Oid LookupCustomProtocolTableSizeFunc(char *protoname);
Oid LookupCustomProtocolDatabaseSizeFunc(char *protoname);

void InvokeMagmaProtocolBlockLocation(ExtTableEntry *ext_entry,
                                             Oid    procOid,
                                             char  *dbname,
                                             char  *schemaname,
                                             char  *tablemame,
                                             MagmaSnapshot *snapshot,
                                             bool useClientCacheDirectly,
                                             ExtProtocolBlockLocationData **bldata);

void InvokeMagmaProtocolTableSize(ExtTableEntry *ext_entry,
                                             Oid procOid,
                                             char *dbname,
                                             char *schemaname,
                                             char *tablename,
                                             MagmaSnapshot *snapshot,
                                             ExtProtocolTableSizeData **tsdata);

void InvokeMagmaProtocolDatabaseSize(Oid procOid,
                                     const char *dbname,
                                     MagmaSnapshot *snapshot,
                                     ExtProtocolDatabaseSizeData **dbsdata);

bool dataStoredInMagma(Relation rel);

bool dataStoredInMagmaByOid(Oid relid);

bool dataStoredInHive(Relation rel);

void fetchUrlStoredInHive(Oid relid, char **hiveUrl);

void fetchDistributionPolicy(Oid relid, int32 *n_dist_keys,
                             int16 **dist_keys);

List *magma_build_range_to_rg_map(List *splits, uint32 *range_to_rg_map);

void magma_build_rg_to_url_map(List *splits, List *rg, uint16 *rgIds,
                                      char **rgUrls);

char *search_hostname_by_ipaddr(const char *ipaddr);

FILE *fp;
FILE *fpaoseg;
FILE *fpsegnum;
FILE *fpratio;
#endif /* CDBDATALOCALITY_H */