Method TouchVCSProviderUser

backend/store/vcs_provider_user.go:49–133

TouchVCSProviderUser refreshes an active VCS provider user, or inserts/reactivates one when doing so would not exceed the active user limit.

(ctx context.Context, workspace string, user *VCSProviderUserMessage, activeWindow time.Duration, limit int)

// TouchVCSProviderUser refreshes an active VCS provider user, or inserts/reactivates
// one when doing so would not exceed the active user limit.
func (s *Store) TouchVCSProviderUser(ctx context.Context, workspace string, user *VCSProviderUserMessage, activeWindow time.Duration, limit int) (bool, error) {
	payload := user.Payload
	if payload == nil {
		payload = &storepb.VCSProviderUserPayload{}
	}
	payloadBytes, err := protojson.Marshal(payload)
	if err != nil {
		return false, errors.Wrapf(err, "failed to marshal VCS provider user payload")
	}

	vcsType := user.VCSType.String()
	var refreshed bool
	if err := s.GetDB().QueryRowContext(ctx, `
		UPDATE vcs_provider_user
		SET last_seen_at = now(), payload = $4
		WHERE workspace = $1 AND vcs_type = $2 AND user_id = $3
			AND last_seen_at >= now() - make_interval(secs => $5)
		RETURNING true
	`, workspace, vcsType, user.UserID, payloadBytes, activeWindow.Seconds()).Scan(&refreshed); err != nil {
		if err != sql.ErrNoRows {
			return false, errors.Wrapf(err, "failed to update active VCS provider user")
		}
	} else {
		return true, nil
	}

	tx, err := s.GetDB().BeginTx(ctx, nil)
	if err != nil {
		return false, errors.Wrapf(err, "failed to begin transaction")
	}
	defer tx.Rollback()

	if err := AcquireAdvisoryXactLockWithStringKey(ctx, tx, AdvisoryLockKeyVCSProviderUser, workspace); err != nil {
		return false, errors.Wrapf(err, "failed to acquire VCS provider user lock")
	}

	var active bool
	if err := tx.QueryRowContext(ctx, `
		SELECT last_seen_at >= now() - make_interval(secs => $4)
		FROM vcs_provider_user
		WHERE workspace = $1 AND vcs_type = $2 AND user_id = $3
	`, workspace, vcsType, user.UserID, activeWindow.Seconds()).Scan(&active); err != nil && err != sql.ErrNoRows {
		return false, errors.Wrapf(err, "failed to get VCS provider user")
	}

	if active {
		if _, err := tx.ExecContext(ctx, `
			UPDATE vcs_provider_user
			SET last_seen_at = now(), payload = $4
			WHERE workspace = $1 AND vcs_type = $2 AND user_id = $3
		`, workspace, vcsType, user.UserID, payloadBytes); err != nil {
			return false, errors.Wrapf(err, "failed to update VCS provider user")
		}
		if err := tx.Commit(); err != nil {
			return false, errors.Wrapf(err, "failed to commit transaction")
		}
		return true, nil
	}