Til fejlretning oprettede jeg testscenariet:
Du skulle have inkluderet det i din opsætning i spørgsmålet.
-- 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');
Dette virker nu, efter nogle rettelser:
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;
Ring til:
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"
Vigtige punkter
- Opret funktionen først, du har vendt eksekveringssekvens i din opsætning
- I funktionslighed skal du kvalificere kolonnenavne for at undgå navnekonflikter med OUT-parametre af samme navn (
user_id
,distance
,explanation
). - Dine CTE-ligheder blev ødelagt. Jeg trak funktionskaldet similarity_gender(..) op i SELECT-listen. For ikke at ringe to gange, deler jeg det op i næste trin.
- Brug parentes til at få adgang til felterne med sammensatte typer. Se den fine manual her .
- Returtype af funktion similarity() havde en fejl:
explanation hstore
. Skal væreexplanation text
.