Commit my code

Author: kovmir

pg_sqlwall--1.0.sql

@@ -3,7 +3,7 @@
 -- complain if script is sourced in psql, rather than via CREATE EXTENSION
 \echo Use "CREATE EXTENSION pg_show_plans" to load this file. \quit
 
-CREATE FUNCTION my_function()
+CREATE FUNCTION my_function(integer, text)
 RETURNS cstring
 AS 'MODULE_PATHNAME'
 LANGUAGE C;

pg_sqlwall.c

@@ -2,10 +2,12 @@
 #include "fmgr.h"
 
 #include "funcapi.h"
+#include "utils/builtins.h"
 #include "miscadmin.h"
 #include "storage/ipc.h"
 #include "storage/lwlock.h"
 #include "storage/shmem.h"
+#include "utils/elog.h"
 
 /* Constants and Macros */
 PG_MODULE_MAGIC;
@@ -116,5 +118,34 @@ pgsw_shmem_startup(void)
 Datum
 my_function(PG_FUNCTION_ARGS)
 {
-	PG_RETURN_CSTRING("Hello from my_function()!");
+	const Oid   user_id  = PG_GETARG_INT32(0);
+	const char* cmd_type = text_to_cstring(PG_GETARG_TEXT_PP(1));
+
+	pgswHashKey    hash_key;
+	pgswHashEntry *hash_entry;
+
+	pgswHashEntry *test_entry;
+
+	hash_key.userId = user_id;
+	hash_entry = (pgswHashEntry *)hash_search(pgsw_hash,
+	                                 &hash_key,
+	                                 HASH_ENTER_NULL,
+	                                 NULL);
+	hash_entry->denyCmd[CMD_DELETE] = true;
+	hash_entry->denyCmd[CMD_UNKNOWN] = true;
+	hash_entry->denyCmd[CMD_SELECT] = false;
+
+	test_entry = (pgswHashEntry *)hash_search(pgsw_hash,
+	                                 &hash_key,
+	                                 HASH_ENTER_NULL,
+	                                 NULL);
+	if (test_entry->denyCmd[CMD_DELETE] && test_entry->denyCmd[CMD_UNKNOWN])
+	{
+		elog(NOTICE, "everything is fine!");
+	}
+	if (!test_entry->denyCmd[CMD_SELECT])
+	{
+		elog(NOTICE, "DEFINITELY OK!");
+	}
+	PG_RETURN_CSTRING(cmd_type);
 }