From a2be5112ba26dabbb932f4eef336d7bab768a6de Mon Sep 17 00:00:00 2001 From: Emile Date: Mon, 21 Oct 2019 12:47:46 +0200 Subject: create table if not exist yet function --- src/db.go | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'src') diff --git a/src/db.go b/src/db.go index 3ec9b71..75cc93a 100644 --- a/src/db.go +++ b/src/db.go @@ -145,3 +145,25 @@ func dbDeleteChallengeByUUID(uuid string) error { return nil } + +func dbCreateTableIfNotExist(db *sql.DB) error { + log.Println("Creating a table in case it doesn't exist") + query := `CREATE TABLE public.challenges +( + uuid uuid NOT NULL DEFAULT uuid_generate_v4(), + name character varying COLLATE pg_catalog."default" NOT NULL, + description character varying COLLATE pg_catalog."default", + flag character varying COLLATE pg_catalog."default", + container character varying COLLATE pg_catalog."default", + category character varying COLLATE pg_catalog."default", + points integer, + static boolean NOT NULL, + CONSTRAINT challenges_pkey PRIMARY KEY (uuid) +)` + + _, err := db.Exec(query) + if err != nil { + return err + } + return nil +} -- cgit 1.4.1