Zum Debuggen habe ich das Testszenario erstellt:
Das hätten Sie in der Frage in Ihr Setup aufnehmen sollen.
-- drop schema x CASCADE;
create schema x
create table x.users(id int);
create table x.demographics (user_id int, gender text);
INSERT INTO x.users VALUES (1),(2),(3),(4),(5);
INSERT INTO x.demographics VALUES (1, 'm'),(2, 'f'),(3, 'm'),(4, 'f'),(5, 'm');
Dies funktioniert jetzt, nach einigen Korrekturen:
create type x.similarity as (
distance float,
explanation text
);
create or replace function x.similarity_gender(my_gender text, other_gender text)
returns x.similarity as $$
declare
distance float;
sim x.similarity;
begin
if my_gender is null or other_gender is null then
distance = 0.9;
elsif (my_gender = other_gender) then
distance = 0.0;
else
distance = 1.0;
end if;
sim.distance = distance;
sim.explanation = hstore('gender', cast(sim.distance as text));
return sim;
end;
$$ language plpgsql immutable;
create or replace function x.similarity(my_user_id int)
returns table(user_id int, distance float, explanation text) as $$
with factors as (
select u.id as user_id, d.gender
from x.users u
join x.demographics d on u.id = d.user_id),
my_factors as (
select f.user_id, f.gender
from factors f
where f.user_id = $1),
similarities as (
select f.user_id, x.similarity_gender(m.gender, f.gender) AS sim
from factors f, my_factors m)
select s.user_id, (s.sim).distance, (s.sim).explanation
from similarities s;
$$ language sql stable strict;
Aufruf:
test=# SELECT * FROM x.similarity(2);
user_id | distance | explanation
---------+----------+---------------
1 | 1 | "gender"=>"1"
2 | 0 | "gender"=>"0"
3 | 1 | "gender"=>"1"
4 | 0 | "gender"=>"0"
5 | 1 | "gender"=>"1"
Wichtige Punkte
- Erstellen Sie zuerst die Funktion, Sie haben die Ausführungsreihenfolge in Ihrem Setup umgekehrt
- Bei Funktionsähnlichkeit müssen Sie Spaltennamen qualifizieren, um Namenskonflikte mit gleichnamigen OUT-Parametern (
user_id
) zu vermeiden ,distance
,explanation
). - Ihre CTE-Ähnlichkeiten wurden verstümmelt. Ich habe den Funktionsaufruf similarity_gender(..) in die SELECT-Liste hochgezogen. Um nicht zweimal anzurufen, habe ich es im nächsten Schritt aufgeteilt.
- Verwenden Sie Klammern, um auf die Felder zusammengesetzter Typen zuzugreifen. Konsultieren Sie das feine Handbuch hier .
- Rückgabetyp der Funktion similarity() hatte einen Fehler:
explanation hstore
. Mussexplanation text
sein .