add pg_cfm to pg_visibility. Delete it after debugging

alubennikova · alubennikova · commit c15789e3948e · 2017-01-19T14:16:40.000+03:00
diff --git a/contrib/pg_visibility/pg_visibility--1.1.sql b/contrib/pg_visibility/pg_visibility--1.1.sql
@@ -64,6 +64,14 @@ AS 'MODULE_PATHNAME', 'pg_truncate_visibility_map'
 LANGUAGE C STRICT
 PARALLEL UNSAFE;  -- let's not make this any more dangerous
 
+-- Show visibility map and page-level visibility information.
+CREATE FUNCTION pg_cfm(regclass, segno int,
+					   physSize OUT int, virtSize OUT int,
+					   usedSize OUT int, generation OUT bigint)
+RETURNS record
+AS 'MODULE_PATHNAME', 'pg_cfm'
+LANGUAGE C STRICT;
+
 -- Don't want these to be available to public.
 REVOKE ALL ON FUNCTION pg_visibility_map(regclass, bigint) FROM PUBLIC;
 REVOKE ALL ON FUNCTION pg_visibility(regclass, bigint) FROM PUBLIC;
diff --git a/contrib/pg_visibility/pg_visibility.c b/contrib/pg_visibility/pg_visibility.c
@@ -20,6 +20,16 @@
 #include "storage/procarray.h"
 #include "storage/smgr.h"
 #include "utils/rel.h"
+#include "storage/cfs.h"
+#ifndef WIN32
+#include <sys/mman.h>
+#endif
+#include <sys/file.h>
+#include <sys/param.h>
+#include <sys/stat.h>
+#include <limits.h>
+#include <unistd.h>
+#include <fcntl.h>
 
 PG_MODULE_MAGIC;
 
@@ -747,3 +757,53 @@ tuple_all_visible(HeapTuple tup, TransactionId OldestXmin, Buffer buffer)
 
 	return true;
 }
+
+PG_FUNCTION_INFO_V1(pg_cfm);
+
+Datum
+pg_cfm(PG_FUNCTION_ARGS)
+{
+	Oid			relid = PG_GETARG_OID(0);
+	int32		segno = PG_GETARG_INT32(1);
+	Relation	rel = relation_open(relid, AccessShareLock);
+	char* path = relpathbackend(rel->rd_node, rel->rd_backend, MAIN_FORKNUM);
+	char* map_path = (char*)palloc(strlen(path) + 16);
+	int md;
+	FileMap* map;
+	TupleDesc tupdesc;
+	Datum		values[4];
+	bool		nulls[4];
+
+	if (segno == 0)
+		sprintf(map_path, "%s.cfm", path);
+	else
+		sprintf(map_path, "%s.%u.cfm", path, segno);
+
+	md = open(map_path, O_RDWR|PG_BINARY, 0);
+
+	map = cfs_mmap(md);
+	if (map == MAP_FAILED)
+		elog(ERROR, "pg_cfm failed to read map file %s: %m", map_path);
+
+	tupdesc = CreateTemplateTupleDesc(4, false);
+	TupleDescInitEntry(tupdesc, (AttrNumber) 1, "physSize", INT4OID, -1, 0);
+	TupleDescInitEntry(tupdesc, (AttrNumber) 2, "virtSize", INT4OID, -1, 0);
+	TupleDescInitEntry(tupdesc, (AttrNumber) 3, "usedSize", INT4OID, -1, 0);
+	TupleDescInitEntry(tupdesc, (AttrNumber) 4, "generation", INT8OID, -1, 0);
+	tupdesc = BlessTupleDesc(tupdesc);
+
+	MemSet(nulls, 0, sizeof(nulls));
+	values[0] = UInt32GetDatum(pg_atomic_read_u32(&map->physSize));
+	values[1] = UInt32GetDatum(pg_atomic_read_u32(&map->virtSize));
+	values[2] = UInt32GetDatum(pg_atomic_read_u32(&map->usedSize));
+	values[3] = UInt64GetDatum(map->generation);
+
+
+	if (cfs_munmap(map) < 0)
+		elog(ERROR, "pg_cfm failed to unmap file %s: %m", map_path);
+	if (close(md) < 0)
+		elog(ERROR, "pg_cfm failed to close file %s: %m", map_path);
+	relation_close(rel, AccessShareLock);
+
+	PG_RETURN_DATUM(HeapTupleGetDatum(heap_form_tuple(tupdesc, values, nulls)));
+}
diff --git a/src/backend/storage/file/cfs.c b/src/backend/storage/file/cfs.c
@@ -842,7 +842,7 @@ static bool cfs_gc_file(char* map_path)
 		pg_atomic_write_u32(&map->usedSize, newSize);
 		pg_atomic_write_u32(&map->physSize, newSize);
 		map->generation += 1; /* force all backends to reopen the file */
-		
+
 		/* Before removing backup files and releasing locks
 		 * we need to flush updated map file */
 		if (cfs_msync(map) < 0)
@@ -906,7 +906,7 @@ static bool cfs_gc_file(char* map_path)
 	else if (cfs_state->max_iterations == 1)
 		elog(LOG, "%d: file %.*s: physical size %d, logical size %d, used %d, compression ratio %f",
 			 MyProcPid, suf, map_path, physSize, virtSize, usedSize, (double)virtSize/physSize);
-	
+
 	if (cfs_munmap(map) < 0)
 	{
 		elog(LOG, "CFS failed to unmap file %s: %m", map_path);
@@ -1204,7 +1204,7 @@ Datum cfs_compression_ratio(PG_FUNCTION_ARGS)
 
 			virtSize += pg_atomic_read_u32(&map->virtSize);
 			physSize += pg_atomic_read_u32(&map->physSize);
-			
+
 			if (cfs_munmap(map) < 0)
 				elog(LOG, "CFS failed to unmap file %s: %m", map_path);
 			if (close(md) < 0)
@@ -1255,7 +1255,7 @@ Datum cfs_fragmentation(PG_FUNCTION_ARGS)
 			}
 			usedSize += pg_atomic_read_u32(&map->usedSize);
 			physSize += pg_atomic_read_u32(&map->physSize);
-			
+
 			if (cfs_munmap(map) < 0)
 				elog(LOG, "CFS failed to unmap file %s: %m", map_path);
 			if (close(md) < 0)
diff --git a/src/backend/storage/file/fd.c b/src/backend/storage/file/fd.c
@@ -1665,6 +1665,7 @@ FileLock(File file)
 
 	cfs_lock_file(vfdP->map, vfdP->fileName); /* protect file from GC */
 	
+	/* Reopen file, because it was rewritten by gc */
 	if (vfdP->generation != vfdP->map->generation)
 	{
 		close(vfdP->fd);
diff --git a/src/include/storage/cfs.h b/src/include/storage/cfs.h
@@ -35,7 +35,9 @@
 #define CFS_COMPRESSOR ZLIB_COMPRESSOR
 #endif
 
-/* Encryption related variables*/
+/* Encryption related variables.
+ * TODO We're going to change this algorithm till PGPROEE2_0
+ */
 #define CFS_RC4_DROP_N 3072 
 #define CFS_CIPHER_KEY_SIZE 256
 

Original file line number	Diff line number	Diff line change
`@@ -1665,6 +1665,7 @@ FileLock(File file)`
`1665`	`1665`
`1666`	`1666`	`cfs_lock_file(vfdP->map, vfdP->fileName); /* protect file from GC */`
`1667`	`1667`
	`1668`	`+ /* Reopen file, because it was rewritten by gc */`
`1668`	`1669`	`if (vfdP->generation != vfdP->map->generation)`
`1669`	`1670`	`{`
`1670`	`1671`	`close(vfdP->fd);`