Fix error: relation public.sr_plans does not exist

Artur Zakirov · Artur Zakirov · commit 9531f664691e · 2017-01-26T19:02:10.000+03:00
diff --git a/Makefile b/Makefile
@@ -23,4 +23,4 @@ endif
 
 genparser:
 #	test -d sr_plan_env ||  
-	python gen_parser.py nodes.h `pg_config --includedir-server`
+	python gen_parser.py nodes.h `$(PG_CONFIG) --includedir-server`
diff --git a/sr_plan.c b/sr_plan.c
@@ -60,6 +60,7 @@ PlannedStmt *sr_planner(Query *parse,
 	bool find_ok = false;
 	LOCKMODE heap_lock = AccessShareLock;
 	Oid query_index_rel_oid;
+	Oid			sr_plans_oid;
 	IndexScanDesc query_index_scan;
 	ScanKeyData key;
 
@@ -81,7 +82,14 @@ PlannedStmt *sr_planner(Query *parse,
 	sr_query_walker((Query *)parse, NULL);
 
 	sr_plans_table_rv = makeRangeVar("public", "sr_plans", -1);
-	sr_plans_heap = heap_openrv(sr_plans_table_rv, heap_lock);
+	/* First check existance of "sr_plans" table */
+	sr_plans_oid = RangeVarGetRelid(sr_plans_table_rv, heap_lock, true);
+	if (!OidIsValid(sr_plans_oid))
+		/* Just call standard_planner() if table doesn't exist. */
+		return standard_planner(parse, cursorOptions, boundParams);
+
+	/* Table "sr_plans" exists */
+	sr_plans_heap = heap_open(sr_plans_oid, NoLock);
 
 #if PG_VERSION_NUM >= 90600
 	query_index_rel_oid = DatumGetObjectId(DirectFunctionCall1(to_regclass, PointerGetDatum(cstring_to_text("sr_plans_query_hash_idx"))));
@@ -90,6 +98,7 @@ PlannedStmt *sr_planner(Query *parse,
 #endif
 	if (query_index_rel_oid == InvalidOid)
 	{
+		heap_close(sr_plans_heap, heap_lock);
 		elog(WARNING, "Not found sr_plans_query_hash_idx index");
 		return standard_planner(parse, cursorOptions, boundParams);
 	}
