Sqlserver
 sql >> Base de données >  >> RDS >> Sqlserver

Regrouper tous les enregistrements associés dans une relation plusieurs à plusieurs, composants connectés au graphique SQL

J'ai pensé à utiliser CTE récursif , mais, pour autant que je sache, il n'est pas possible dans SQL Server d'utiliser UNION pour connecter un membre d'ancrage et un membre récursif de CTE récursif (je pense que c'est possible de le faire dans PostgreSQL), il n'est donc pas possible d'éliminer les doublons.

declare @i int

with cte as (
     select
         GroupID,
         row_number() over(order by Company) as rn
     from Table1
)
update cte set GroupID = rn

select @i = @@rowcount

-- while some rows updated
while @i > 0
begin
    update T1 set
        GroupID = T2.GroupID
    from Table1 as T1
        inner join (
            select T2.Company, min(T2.GroupID) as GroupID
            from Table1 as T2
            group by T2.Company
        ) as T2 on T2.Company = T1.Company
    where T1.GroupID > T2.GroupID

    select @i = @@rowcount

    update T1 set
        GroupID = T2.GroupID
    from Table1 as T1
        inner join (
            select T2.Publisher, min(T2.GroupID) as GroupID
            from Table1 as T2
            group by T2.Publisher
        ) as T2 on T2.Publisher = T1.Publisher
    where T1.GroupID > T2.GroupID

    -- will be > 0 if any rows updated
    select @i = @i + @@rowcount
end

;with cte as (
     select
         GroupID,
         dense_rank() over(order by GroupID) as rn
     from Table1
)
update cte set GroupID = rn

démo de violon SQL

J'ai également essayé un premier algorithme de recherche en largeur. Je pensais que cela pourrait être plus rapide (c'est mieux en termes de complexité), donc je vais fournir une solution ici. J'ai trouvé que ce n'est pas plus rapide que l'approche SQL, cependant :

declare @Company nvarchar(2), @Publisher nvarchar(2), @GroupID int

declare @Queue table (
    Company nvarchar(2), Publisher nvarchar(2), ID int identity(1, 1),
    primary key(Company, Publisher)
)

select @GroupID = 0

while 1 = 1
begin
    select top 1 @Company = Company, @Publisher = Publisher
    from Table1
    where GroupID is null

    if @@rowcount = 0 break

    select @GroupID = @GroupID + 1

    insert into @Queue(Company, Publisher)
    select @Company, @Publisher

    while 1 = 1
    begin
        select top 1 @Company = Company, @Publisher = Publisher
        from @Queue
        order by ID asc

        if @@rowcount = 0 break

        update Table1 set
            GroupID = @GroupID
        where Company = @Company and Publisher = @Publisher

        delete from @Queue where Company = @Company and Publisher = @Publisher

        ;with cte as (
            select Company, Publisher from Table1 where Company = @Company and GroupID is null
            union all
            select Company, Publisher from Table1 where Publisher = @Publisher and GroupID is null
        )
        insert into @Queue(Company, Publisher)
        select distinct c.Company, c.Publisher
        from cte as c
        where not exists (select * from @Queue as q where q.Company = c.Company and q.Publisher = c.Publisher)
   end
end

démo de violon SQL

J'ai testé ma version et celle de Gordon Linoff pour vérifier ses performances. Il semble que le CTE soit bien pire, je ne pouvais pas attendre qu'il soit terminé sur plus de 1000 lignes.

Voici démo de violon SQL avec des données aléatoires. Mes résultats étaient :
128 lignes :
ma solution RBAR :190 ms
ma solution SQL :27ms
Solution de Gordon Linoff :958 ms
256 lignes :
ma solution RBAR :560 ms
ma solution SQL :1226 ms
Solution de Gordon Linoff :45371ms

Il s'agit de données aléatoires, les résultats peuvent donc ne pas être très cohérents. Je pense que le timing pourrait être modifié par les index, mais ne pense pas que cela puisse changer une image entière.

ancien version - en utilisant une table temporaire, en calculant simplement GroupID sans toucher à la table initiale :

declare @i int

-- creating table to gather all possible GroupID for each row
create table #Temp
(
    Company varchar(1), Publisher varchar(1), GroupID varchar(1),
    primary key (Company, Publisher, GroupID)
)

-- initializing it with data
insert into #Temp (Company, Publisher, GroupID)
select Company, Publisher, Company
from Table1

select @i = @@rowcount

-- while some rows inserted into #Temp
while @i > 0
begin
    -- expand #Temp in both directions
    ;with cte as (
        select
            T2.Company, T1.Publisher,
            T1.GroupID as GroupID1, T2.GroupID as GroupID2
        from #Temp as T1
            inner join #Temp as T2 on T2.Company = T1.Company
        union
        select
            T1.Company, T2.Publisher,
            T1.GroupID as GroupID1, T2.GroupID as GroupID2
        from #Temp as T1
            inner join #Temp as T2 on T2.Publisher = T1.Publisher        
    ), cte2 as (
        select
            Company, Publisher,
            case when GroupID1 < GroupID2 then GroupID1 else GroupID2 end as GroupID
        from cte
    )
    insert into #Temp
    select Company, Publisher, GroupID
    from cte2
    -- don't insert duplicates
    except
    select Company, Publisher, GroupID
    from #Temp

    -- will be > 0 if any row inserted
    select @i = @@rowcount
end

select
    Company, Publisher,
    dense_rank() over(order by min(GroupID)) as GroupID
from #Temp
group by Company, Publisher

=> exemple de violon sql