How to filter SQL results in a has-many-through relation

后端 未结 13 1460
有刺的猬
有刺的猬 2020-11-21 05:17

Assuming I have the tables student, club, and student_club:

student {
    id
    name
}
club {
    id
    name
}
stude         


        
相关标签:
13条回答
  • 2020-11-21 05:57

    Another CTE. It looks clean, but it will probably generate the same plan as a groupby in a normal subquery.

    WITH two AS (
        SELECT student_id FROM tmp.student_club
        WHERE club_id IN (30,50)
        GROUP BY student_id
        HAVING COUNT(*) > 1
        )
    SELECT st.* FROM tmp.student st
    JOIN two ON (two.student_id=st.id)
        ;
    

    For those who want to test, a copy of my generate testdata thingy:

    DROP SCHEMA tmp CASCADE;
    CREATE SCHEMA tmp;
    
    CREATE TABLE tmp.student
        ( id INTEGER NOT NULL PRIMARY KEY
        , sname VARCHAR
        );
    
    CREATE TABLE tmp.club
        ( id INTEGER NOT NULL PRIMARY KEY
        , cname VARCHAR
        );
    
    CREATE TABLE tmp.student_club
        ( student_id INTEGER NOT NULL  REFERENCES tmp.student(id)
        , club_id INTEGER NOT NULL  REFERENCES tmp.club(id)
        );
    
    INSERT INTO tmp.student(id)
        SELECT generate_series(1,1000)
        ;
    
    INSERT INTO tmp.club(id)
        SELECT generate_series(1,100)
        ;
    
    INSERT INTO tmp.student_club(student_id,club_id)
        SELECT st.id  , cl.id
        FROM tmp.student st, tmp.club cl
        ;
    
    DELETE FROM tmp.student_club
    WHERE random() < 0.8
        ;
    
    UPDATE tmp.student SET sname = 'Student#' || id::text ;
    UPDATE tmp.club SET cname = 'Soccer' WHERE id = 30;
    UPDATE tmp.club SET cname = 'Baseball' WHERE id = 50;
    
    ALTER TABLE tmp.student_club
        ADD PRIMARY KEY (student_id,club_id)
        ;
    
    0 讨论(0)
  • 2020-11-21 05:57
    WITH RECURSIVE two AS
        ( SELECT 1::integer AS level
        , student_id
        FROM tmp.student_club sc0
        WHERE sc0.club_id = 30
        UNION
        SELECT 1+two.level AS level
        , sc1.student_id
        FROM tmp.student_club sc1
        JOIN two ON (two.student_id = sc1.student_id)
        WHERE sc1.club_id = 50
        AND two.level=1
        )
    SELECT st.* FROM tmp.student st
    JOIN two ON (two.student_id=st.id)
    WHERE two.level> 1
    
        ;
    

    This seems to perform reasonably well, since the CTE-scan avoids the need for two separate subqueries.

    There is always a reason to misuse recursive queries!

    (BTW: mysql does not seem to have recursive queries)

    0 讨论(0)
  • 2020-11-21 05:57
    SELECT s.stud_id, s.name
    FROM   student s,
    (
    select x.stud_id from 
    student_club x 
    JOIN   student_club y ON x.stud_id = y.stud_id
    WHERE  x.club_id = 30
    AND    y.club_id = 50
    ) tmp_tbl
    where tmp_tbl.stud_id = s.stud_id
    ;
    

    Use of fastest variant (Mr. Sean in Mr. Brandstetter chart). May be variant with only one join to only the student_club matrix has the right to live. So, the longest query will have only two columns to calculate, idea is to make the query thin.

    0 讨论(0)
  • 2020-11-21 05:59

    I was curious. And as we all know, curiosity has a reputation for killing cats.

    So, which is the fastest way to skin a cat?

    The precise cat-skinning environment for this test:

    • PostgreSQL 9.0 on Debian Squeeze with decent RAM and settings.
    • 6.000 students, 24.000 club memberships (data copied from a similar database with real life data.)
    • Slight diversion from the naming schema in the question: student.id is student.stud_id and club.id is club.club_id here.
    • I named the queries after their author in this thread, with an index where there are two.
    • I ran all queries a couple of times to populate the cache, then I picked the best of 5 with EXPLAIN ANALYZE.
    • Relevant indexes (should be the optimum - as long as we lack fore-knowledge which clubs will be queried):

      ALTER TABLE student ADD CONSTRAINT student_pkey PRIMARY KEY(stud_id );
      ALTER TABLE student_club ADD CONSTRAINT sc_pkey PRIMARY KEY(stud_id, club_id);
      ALTER TABLE club       ADD CONSTRAINT club_pkey PRIMARY KEY(club_id );
      CREATE INDEX sc_club_id_idx ON student_club (club_id);
      

      club_pkey is not required by most queries here.
      Primary keys implement unique indexes automatically In PostgreSQL.
      The last index is to make up for this known shortcoming of multi-column indexes on PostgreSQL:

    A multicolumn B-tree index can be used with query conditions that involve any subset of the index's columns, but the index is most efficient when there are constraints on the leading (leftmost) columns.

    Results:

    Total runtimes from EXPLAIN ANALYZE.

    1) Martin 2: 44.594 ms

    SELECT s.stud_id, s.name
    FROM   student s
    JOIN   student_club sc USING (stud_id)
    WHERE  sc.club_id IN (30, 50)
    GROUP  BY 1,2
    HAVING COUNT(*) > 1;
    

    2) Erwin 1: 33.217 ms

    SELECT s.stud_id, s.name
    FROM   student s
    JOIN   (
       SELECT stud_id
       FROM   student_club
       WHERE  club_id IN (30, 50)
       GROUP  BY 1
       HAVING COUNT(*) > 1
       ) sc USING (stud_id);
    

    3) Martin 1: 31.735 ms

    SELECT s.stud_id, s.name
       FROM   student s
       WHERE  student_id IN (
       SELECT student_id
       FROM   student_club
       WHERE  club_id = 30
       INTERSECT
       SELECT stud_id
       FROM   student_club
       WHERE  club_id = 50);
    

    4) Derek: 2.287 ms

    SELECT s.stud_id,  s.name
    FROM   student s
    WHERE  s.stud_id IN (SELECT stud_id FROM student_club WHERE club_id = 30)
    AND    s.stud_id IN (SELECT stud_id FROM student_club WHERE club_id = 50);
    

    5) Erwin 2: 2.181 ms

    SELECT s.stud_id,  s.name
    FROM   student s
    WHERE  EXISTS (SELECT * FROM student_club
                   WHERE  stud_id = s.stud_id AND club_id = 30)
    AND    EXISTS (SELECT * FROM student_club
                   WHERE  stud_id = s.stud_id AND club_id = 50);
    

    6) Sean: 2.043 ms

    SELECT s.stud_id, s.name
    FROM   student s
    JOIN   student_club x ON s.stud_id = x.stud_id
    JOIN   student_club y ON s.stud_id = y.stud_id
    WHERE  x.club_id = 30
    AND    y.club_id = 50;
    

    The last three perform pretty much the same. 4) and 5) result in the same query plan.

    Late Additions:

    Fancy SQL, but the performance can't keep up.

    7) ypercube 1: 148.649 ms

    SELECT s.stud_id,  s.name
    FROM   student AS s
    WHERE  NOT EXISTS (
       SELECT *
       FROM   club AS c 
       WHERE  c.club_id IN (30, 50)
       AND    NOT EXISTS (
          SELECT *
          FROM   student_club AS sc 
          WHERE  sc.stud_id = s.stud_id
          AND    sc.club_id = c.club_id  
          )
       );
    

    8) ypercube 2: 147.497 ms

    SELECT s.stud_id,  s.name
    FROM   student AS s
    WHERE  NOT EXISTS (
       SELECT *
       FROM  (
          SELECT 30 AS club_id  
          UNION  ALL
          SELECT 50
          ) AS c
       WHERE NOT EXISTS (
          SELECT *
          FROM   student_club AS sc 
          WHERE  sc.stud_id = s.stud_id
          AND    sc.club_id = c.club_id  
          )
       );
    

    As expected, those two perform almost the same. Query plan results in table scans, the planner doesn't find a way to use the indexes here.


    9) wildplasser 1: 49.849 ms

    WITH RECURSIVE two AS (
       SELECT 1::int AS level
            , stud_id
       FROM   student_club sc1
       WHERE  sc1.club_id = 30
       UNION
       SELECT two.level + 1 AS level
            , sc2.stud_id
       FROM   student_club sc2
       JOIN   two USING (stud_id)
       WHERE  sc2.club_id = 50
       AND    two.level = 1
       )
    SELECT s.stud_id, s.student
    FROM   student s
    JOIN   two USING (studid)
    WHERE  two.level > 1;
    

    Fancy SQL, decent performance for a CTE. Very exotic query plan.
    Again, would be interesting how 9.1 handles this. I am going to upgrade the db cluster used here to 9.1 soon. Maybe I'll rerun the whole shebang ...


    10) wildplasser 2: 36.986 ms

    WITH sc AS (
       SELECT stud_id
       FROM   student_club
       WHERE  club_id IN (30,50)
       GROUP  BY stud_id
       HAVING COUNT(*) > 1
       )
    SELECT s.*
    FROM   student s
    JOIN   sc USING (stud_id);
    

    CTE variant of query 2). Surprisingly, it can result in a slightly different query plan with the exact same data. I found a sequential scan on student, where the subquery-variant used the index.


    11) ypercube 3: 101.482 ms

    Another late addition @ypercube. It is positively amazing, how many ways there are.

    SELECT s.stud_id, s.student
    FROM   student s
    JOIN   student_club sc USING (stud_id)
    WHERE  sc.club_id = 10                 -- member in 1st club ...
    AND    NOT EXISTS (
       SELECT *
       FROM  (SELECT 14 AS club_id) AS c  -- can't be excluded for missing the 2nd
       WHERE  NOT EXISTS (
          SELECT *
          FROM   student_club AS d
          WHERE  d.stud_id = sc.stud_id
          AND    d.club_id = c.club_id
          )
       )
    

    12) erwin 3: 2.377 ms

    @ypercube's 11) is actually just the mind-twisting reverse approach of this simpler variant, that was also still missing. Performs almost as fast as the top cats.

    SELECT s.*
    FROM   student s
    JOIN   student_club x USING (stud_id)
    WHERE  sc.club_id = 10                 -- member in 1st club ...
    AND    EXISTS (                        -- ... and membership in 2nd exists
       SELECT *
       FROM   student_club AS y
       WHERE  y.stud_id = s.stud_id
       AND    y.club_id = 14
       )
    

    13) erwin 4: 2.375 ms

    Hard to believe, but here's another, genuinely new variant. I see potential for more than two memberships, but it also ranks among the top cats with just two.

    SELECT s.*
    FROM   student AS s
    WHERE  EXISTS (
       SELECT *
       FROM   student_club AS x
       JOIN   student_club AS y USING (stud_id)
       WHERE  x.stud_id = s.stud_id
       AND    x.club_id = 14
       AND    y.club_id = 10
       )
    

    Dynamic number of club memberships

    In other words: varying number of filters. This question asked for exactly two club memberships. But many use cases have to prepare for a varying number.

    Detailed discussion in this related later answer:

    • Using same column multiple times in WHERE clause
    0 讨论(0)
  • 2020-11-21 06:00
    select *
    from student
    where id in (select student_id from student_club where club_id = 30)
    and id in (select student_id from student_club where club_id = 50)
    
    0 讨论(0)
  • 2020-11-21 06:01

    Since noone has added this (classic) version:

    SELECT s.*
    FROM student AS s
    WHERE NOT EXISTS
          ( SELECT *
            FROM club AS c 
            WHERE c.id IN (30, 50)
              AND NOT EXISTS
                  ( SELECT *
                    FROM student_club AS sc 
                    WHERE sc.student_id = s.id
                      AND sc.club_id = c.id  
                  )
          )
    

    or similar:

    SELECT s.*
    FROM student AS s
    WHERE NOT EXISTS
          ( SELECT *
            FROM
              ( SELECT 30 AS club_id  
              UNION ALL
                SELECT 50
              ) AS c
            WHERE NOT EXISTS
                  ( SELECT *
                    FROM student_club AS sc 
                    WHERE sc.student_id = s.id
                      AND sc.club_id = c.club_id  
                  )
          )
    

    One more try with a slightly different approach. Inspired by an article in Explain Extended: Multiple attributes in a EAV table: GROUP BY vs. NOT EXISTS:

    SELECT s.*
    FROM student_club AS sc
      JOIN student AS s
        ON s.student_id = sc.student_id
    WHERE sc.club_id = 50                      --- one option here
      AND NOT EXISTS
          ( SELECT *
            FROM
              ( SELECT 30 AS club_id           --- all the rest in here
                                               --- as in previous query
              ) AS c
            WHERE NOT EXISTS
                  ( SELECT *
                    FROM student_club AS scc 
                    WHERE scc.student_id = sc.id
                      AND scc.club_id = c.club_id  
                  )
          )
    

    Another approach:

    SELECT s.stud_id
    FROM   student s
    
    EXCEPT
    
    SELECT stud_id
    FROM 
      ( SELECT s.stud_id, c.club_id
        FROM student s 
          CROSS JOIN (VALUES (30),(50)) c (club_id)
      EXCEPT
        SELECT stud_id, club_id
        FROM student_club
        WHERE club_id IN (30, 50)   -- optional. Not needed but may affect performance
      ) x ;   
    
    0 讨论(0)
提交回复
热议问题