1
0
mirror of https://github.com/bitwarden/server.git synced 2025-07-01 08:02:49 -05:00

[AC-1682] mySql scripts formatting

This commit is contained in:
Rui Tome
2023-12-19 15:46:23 +00:00
parent 5f1f7d6f23
commit fc8606acab
2 changed files with 35 additions and 58 deletions

View File

@ -6,20 +6,22 @@ WHERE `AccessAll` = 1;
-- Step 2: Update existing rows in `CollectionGroup`
UPDATE `CollectionGroups` CG
INNER JOIN `Collection` C ON CG.`CollectionId` = C.`Id`
INNER JOIN TempGroup TG ON CG.`GroupId` = TG.`GroupId`
SET
CG.`ReadOnly` = 0,
CG.`HidePasswords` = 0,
CG.`Manage` = 0
INNER JOIN `Collection` C ON CG.`CollectionId` = C.`Id`
INNER JOIN TempGroup TG ON CG.`GroupId` = TG.`GroupId`
SET
CG.`ReadOnly` = 0,
CG.`HidePasswords` = 0,
CG.`Manage` = 0
WHERE C.`OrganizationId` = TG.`OrganizationId`;
-- Step 3: Insert new rows into `CollectionGroup`
INSERT INTO `CollectionGroups` (`CollectionId`, `GroupId`, `ReadOnly`, `HidePasswords`, `Manage`)
SELECT C.`Id`, TG.`GroupId`, 0, 0, 0
FROM `Collection` C
INNER JOIN TempGroup TG ON C.`OrganizationId` = TG.`OrganizationId`
LEFT JOIN `CollectionGroups` CG ON CG.`CollectionId` = C.`Id` AND CG.`GroupId` = TG.`GroupId`
INNER JOIN TempGroup TG
ON C.`OrganizationId` = TG.`OrganizationId`
LEFT JOIN `CollectionGroups` CG
ON CG.`CollectionId` = C.`Id` AND CG.`GroupId` = TG.`GroupId`
WHERE CG.`CollectionId` IS NULL;
-- Step 4: Drop the temporary table

View File

@ -1,51 +1,26 @@
-- Step 1: Create a temporary table with an additional column for batch processing, update 50 k at a time
CREATE TEMPORARY TABLE TempOrgUser AS
SELECT OrganizationUser.Id AS OrganizationUserId, OrganizationUser.OrganizationId,
CAST(ROW_NUMBER() OVER(ORDER BY OrganizationUser.Id) / 50000 AS SIGNED) AS Batch
FROM OrganizationUser
WHERE AccessAll = 1;
-- Update existing rows in CollectionUsers
UPDATE CollectionUsers AS target
INNER JOIN (
SELECT C.Id AS CollectionId, T.OrganizationUserId
FROM Collection C
INNER JOIN OrganizationUser T ON C.OrganizationId = T.OrganizationId
WHERE T.AccessAll = 1
) AS source
ON target.CollectionId = source.CollectionId AND target.OrganizationUserId = source.OrganizationUserId
SET
target.ReadOnly = 0,
target.HidePasswords = 0,
target.Manage = 0;
-- Step 2: Get the maximum batch number
SET @MaxBatch := (SELECT MAX(Batch) FROM TempOrgUser);
SET @CurrentBatch := 0;
-- Create the stored procedure
DROP PROCEDURE IF EXISTS ProcessBatches;
CREATE PROCEDURE ProcessBatches(INOUT currentBatch INT, IN maxBatch INT)
BEGIN
WHILE currentBatch <= maxBatch DO
-- Update existing rows in CollectionUsers
UPDATE CollectionUsers AS target
INNER JOIN (
SELECT C.Id AS CollectionId, T.OrganizationUserId
FROM Collection C
INNER JOIN TempOrgUser T ON C.OrganizationId = T.OrganizationId AND T.Batch = currentBatch
) AS source
ON target.CollectionId = source.CollectionId AND target.OrganizationUserId = source.OrganizationUserId
SET
target.ReadOnly = 0,
target.HidePasswords = 0,
target.Manage = 0;
-- Insert new rows into CollectionUsers
INSERT INTO CollectionUsers (CollectionId, OrganizationUserId, ReadOnly, HidePasswords, Manage)
SELECT source.CollectionId, source.OrganizationUserId, 0, 0, 0
FROM (
SELECT C.Id AS CollectionId, T.OrganizationUserId
FROM Collection C
INNER JOIN TempOrgUser T ON C.OrganizationId = T.OrganizationId AND T.Batch = currentBatch
) AS source
LEFT JOIN CollectionUsers AS target
ON target.CollectionId = source.CollectionId AND target.OrganizationUserId = source.OrganizationUserId
WHERE target.CollectionId IS NULL;
-- Move to the next batch
SET currentBatch := currentBatch + 1;
END WHILE;
END;
-- Step 3: Process each batch by calling the stored procedure
CALL ProcessBatches(@CurrentBatch, @MaxBatch);
-- Step 4: Drop the temporary table
DROP TEMPORARY TABLE IF EXISTS TempOrgUser;
-- Insert new rows into CollectionUsers
INSERT INTO CollectionUsers (CollectionId, OrganizationUserId, ReadOnly, HidePasswords, Manage)
SELECT source.CollectionId, source.OrganizationUserId, 0, 0, 0
FROM (
SELECT C.Id AS CollectionId, T.OrganizationUserId
FROM Collection C
INNER JOIN OrganizationUser T ON C.OrganizationId = T.OrganizationId
WHERE T.AccessAll = 1
) AS source
LEFT JOIN CollectionUsers AS target
ON target.CollectionId = source.CollectionId AND target.OrganizationUserId = source.OrganizationUserId
WHERE target.CollectionId IS NULL;