From 08ed9cf54665cec1e86f937f521c9a194f7fb4b0 Mon Sep 17 00:00:00 2001 From: Daniel Ponte Date: Sun, 28 Jul 2024 08:40:01 -0400 Subject: [PATCH] new learn --- pkg/gordio/database/models.go | 12 +++++------ sql/postgres/migrations/001_initial.up.sql | 25 +++++++--------------- 2 files changed, 14 insertions(+), 23 deletions(-) diff --git a/pkg/gordio/database/models.go b/pkg/gordio/database/models.go index 900e0f3..afcfc98 100644 --- a/pkg/gordio/database/models.go +++ b/pkg/gordio/database/models.go @@ -77,12 +77,12 @@ type Talkgroup struct { } type TalkgroupsLearned struct { - ID int32 `json:"id"` - SystemID int `json:"system_id"` - Tgid int `json:"tgid"` - GroupName string `json:"group_name"` - GroupTag *string `json:"group_tag"` - Ignored *bool `json:"ignored"` + ID int32 `json:"id"` + SystemID int `json:"system_id"` + Tgid int `json:"tgid"` + Name string `json:"name"` + GroupTag *string `json:"group_tag"` + Ignored *bool `json:"ignored"` } type TalkgroupsTag struct { diff --git a/sql/postgres/migrations/001_initial.up.sql b/sql/postgres/migrations/001_initial.up.sql index 84a44c3..b613c04 100644 --- a/sql/postgres/migrations/001_initial.up.sql +++ b/sql/postgres/migrations/001_initial.up.sql @@ -37,32 +37,23 @@ CREATE TABLE IF NOT EXISTS talkgroups_learned( id SERIAL PRIMARY KEY, system_id INTEGER REFERENCES systems(id) NOT NULL, tgid INTEGER NOT NULL, - group_name TEXT NOT NULL, + name TEXT NOT NULL, group_tag TEXT, ignored BOOLEAN, - UNIQUE (system_id, tgid, group_name) + UNIQUE (system_id, tgid, name) ); CREATE OR REPLACE FUNCTION learn_talkgroup() RETURNS TRIGGER AS $$ BEGIN - IF NOT EXISTS (SELECT * - FROM talkgroups - LEFT JOIN talkgroups_tags - ON talkgroups_tags.system_id = talkgroups.system_id AND talkgroups_tags.talkgroup_id = talkgroups.tgid - LEFT JOIN talkgroups_learned - ON talkgroups_learned.system_id = talkgroups.system_id AND talkgroups_learned.tgid = talkgroups.tgid - WHERE - talkgroups.system_id = NEW.system AND talkgroups.tgid = NEW.talkgroup AND - ( - talkgroups.name = NEW.tg_label - AND (NEW.tg_tag IS NULL OR (talkgroups_tags.tags @> ARRAY[NEW.tg_tag])) - AND (NEW.tg_group IS NULL OR talkgroups.tg_group = NEW.tg_group) - ) + IF NOT EXISTS ( + SELECT tg.system_id, tg.tgid, tg.name FROM talkgroups tg WHERE tg.system_id = NEW.system AND tg.tgid = NEW.talkgroup + UNION + SELECT tgl.system_id, tgl.tgid, tgl.name FROM talkgroups_learned tgl WHERE tgl.system_id = NEW.system AND tgl.tgid = NEW.talkgroup ) THEN - INSERT INTO talkgroups_learned(system_id, tgid, group_name, group_tag) VALUES( + INSERT INTO talkgroups_learned(system_id, tgid, name, group_tag) VALUES( NEW.system, NEW.talkgroup, NEW.tg_label, NEW.tg_tag - ) ON CONFLICT (system_id, tgid, group_name) DO UPDATE SET group_tag = EXCLUDED.group_tag; + ) ON CONFLICT DO NOTHING; END IF; RETURN NEW; END