Files
probo/pkg/coredata/processing_activity_third_party.go
Sacha Al Himdani 9ac71f948f Update contact email to hello@probo.com
Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
2026-06-09 16:45:23 +02:00

121 lines
3.6 KiB
Go

// Copyright (c) 2025-2026 Probo Inc <hello@probo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package coredata
import (
"context"
"fmt"
"time"
"github.com/jackc/pgx/v5"
"go.gearno.de/kit/pg"
"go.probo.inc/probo/pkg/gid"
)
type (
ProcessingActivityThirdParty struct {
ProcessingActivityID gid.GID `db:"processing_activity_id"`
ThirdPartyID gid.GID `db:"third_party_id"`
TenantID gid.TenantID `db:"tenant_id"`
CreatedAt time.Time `db:"created_at"`
}
ProcessingActivityThirdParties []*ProcessingActivityThirdParty
)
func (pav ProcessingActivityThirdParties) Merge(
ctx context.Context,
conn pg.Querier,
scope Scoper,
processingActivityID gid.GID,
organizationID gid.GID,
thirdPartyIDs []gid.GID,
) error {
q := `
WITH third_party_ids AS (
SELECT
unnest(@third_party_ids::text[]) AS third_party_id,
@tenant_id AS tenant_id,
@processing_activity_id AS processing_activity_id,
@organization_id AS organization_id,
@created_at::timestamptz AS created_at
)
MERGE INTO processing_activity_third_parties AS tgt
USING third_party_ids AS src
ON tgt.tenant_id = src.tenant_id
AND tgt.processing_activity_id = src.processing_activity_id
AND tgt.third_party_id = src.third_party_id
WHEN NOT MATCHED
THEN INSERT (tenant_id, processing_activity_id, third_party_id, organization_id, created_at)
VALUES (src.tenant_id, src.processing_activity_id, src.third_party_id, src.organization_id, src.created_at)
WHEN NOT MATCHED BY SOURCE
AND tgt.tenant_id = @tenant_id AND tgt.processing_activity_id = @processing_activity_id
THEN DELETE
`
args := pgx.StrictNamedArgs{
"tenant_id": scope.GetTenantID(),
"processing_activity_id": processingActivityID,
"organization_id": organizationID,
"created_at": time.Now(),
"third_party_ids": thirdPartyIDs,
}
_, err := conn.Exec(ctx, q, args)
if err != nil {
return fmt.Errorf("cannot merge processing activity thirdParties: %w", err)
}
return nil
}
func (pav ProcessingActivityThirdParties) Insert(
ctx context.Context,
conn pg.Tx,
scope Scoper,
processingActivityID gid.GID,
organizationID gid.GID,
thirdPartyIDs []gid.GID,
) error {
q := `
WITH third_party_ids AS (
SELECT unnest(@third_party_ids::text[]) AS third_party_id
)
INSERT INTO processing_activity_third_parties (tenant_id, processing_activity_id, third_party_id, organization_id, created_at)
SELECT
@tenant_id AS tenant_id,
@processing_activity_id AS processing_activity_id,
third_party_id,
@organization_id AS organization_id,
@created_at AS created_at
FROM third_party_ids
`
args := pgx.StrictNamedArgs{
"tenant_id": scope.GetTenantID(),
"processing_activity_id": processingActivityID,
"organization_id": organizationID,
"created_at": time.Now(),
"third_party_ids": thirdPartyIDs,
}
_, err := conn.Exec(ctx, q, args)
if err != nil {
return fmt.Errorf("cannot insert processing activity thirdParties: %w", err)
}
return nil
}