b6781d3de0
Replace the many-to-many junction table with a direct parent_third_party_id foreign key on third_parties. Each sub-third-party now belongs to exactly one parent, making duplicates across parents independent entities. Replace the firstLevel boolean with an integer level field (1 = direct, 2+ = parent level + 1) to support arbitrary nesting depth. Remove the createThirdPartyThirdPartyMapping and deleteThirdPartyThirdPartyMapping mutations, the CLI link/unlink commands, and the corresponding MCP tools. Creating a child third party now just requires passing parentThirdPartyId on the existing createThirdParty mutation. The frontend walks the parentThirdParty chain to build display names like "Name (Ancestor1/Ancestor2)" and shows clickable ancestor links on the detail page. Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
128 lines
3.0 KiB
Go
128 lines
3.0 KiB
Go
// Copyright (c) 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"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"go.gearno.de/kit/pg"
|
|
)
|
|
|
|
func (v *ThirdParty) LoadNextPendingVettingForUpdateSkipLocked(
|
|
ctx context.Context,
|
|
tx pg.Tx,
|
|
) error {
|
|
q := `
|
|
SELECT
|
|
id,
|
|
organization_id,
|
|
parent_third_party_id,
|
|
common_third_party_id,
|
|
name,
|
|
description,
|
|
category,
|
|
headquarter_address,
|
|
legal_name,
|
|
website_url,
|
|
privacy_policy_url,
|
|
service_level_agreement_url,
|
|
data_processing_agreement_url,
|
|
business_associate_agreement_url,
|
|
subprocessors_list_url,
|
|
certifications,
|
|
countries,
|
|
business_owner_profile_id,
|
|
security_owner_profile_id,
|
|
status_page_url,
|
|
terms_of_service_url,
|
|
security_page_url,
|
|
trust_page_url,
|
|
show_on_trust_center,
|
|
level,
|
|
vetting_status,
|
|
vetting_website_url,
|
|
vetting_procedure,
|
|
vetting_processing_started_at,
|
|
vetting_error_message,
|
|
created_at,
|
|
updated_at
|
|
FROM
|
|
third_parties
|
|
WHERE
|
|
vetting_status = @vetting_status
|
|
AND vetting_website_url IS NOT NULL
|
|
ORDER BY
|
|
created_at ASC
|
|
LIMIT 1
|
|
FOR UPDATE SKIP LOCKED;
|
|
`
|
|
|
|
rows, err := tx.Query(
|
|
ctx,
|
|
q,
|
|
pgx.StrictNamedArgs{"vetting_status": ThirdPartyVettingStatusPending},
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot query third party vetting queue: %w", err)
|
|
}
|
|
|
|
thirdParty, err := pgx.CollectExactlyOneRow(rows, pgx.RowToStructByName[ThirdParty])
|
|
if err != nil {
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
|
return ErrResourceNotFound
|
|
}
|
|
|
|
return fmt.Errorf("cannot collect third party: %w", err)
|
|
}
|
|
|
|
*v = thirdParty
|
|
|
|
return nil
|
|
}
|
|
|
|
func ResetStaleVettingProcessing(
|
|
ctx context.Context,
|
|
conn pg.Querier,
|
|
staleAfter time.Duration,
|
|
) error {
|
|
q := `
|
|
UPDATE third_parties
|
|
SET
|
|
vetting_status = @pending_status,
|
|
vetting_processing_started_at = NULL,
|
|
updated_at = @now
|
|
WHERE
|
|
vetting_status = @processing_status
|
|
AND vetting_processing_started_at < @stale_before;
|
|
`
|
|
|
|
_, err := conn.Exec(
|
|
ctx,
|
|
q,
|
|
pgx.StrictNamedArgs{
|
|
"pending_status": ThirdPartyVettingStatusPending,
|
|
"processing_status": ThirdPartyVettingStatusProcessing,
|
|
"now": time.Now(),
|
|
"stale_before": time.Now().Add(-staleAfter),
|
|
},
|
|
)
|
|
|
|
return err
|
|
}
|