Add explicit GRPC call to clear caches/cleanup. Print cache stats.

ConductorOne · Nov 8, 2024 · 93550b1 · 93550b1
1 parent 82ace9a
commit 93550b1
Show file tree

Hide file tree

Showing 11 changed files with 688 additions and 192 deletions.
diff --git a/pb/c1/connector/v2/connector.pb.go b/pb/c1/connector/v2/connector.pb.go
diff --git a/pb/c1/connector/v2/connector.pb.validate.go b/pb/c1/connector/v2/connector.pb.validate.go
diff --git a/pb/c1/connector/v2/connector_grpc.pb.go b/pb/c1/connector/v2/connector_grpc.pb.go
diff --git a/pkg/connectorbuilder/connectorbuilder.go b/pkg/connectorbuilder/connectorbuilder.go
@@ -406,13 +406,6 @@ func (b *builderImpl) ListResourceTypes(
 	tt := tasks.ListResourceTypesType
 	var out []*v2.ResourceType
 
-	l := ctxzap.Extract(ctx)
-	// Clear all http caches at the start of a sync. This must be run in the child process, which is why it's in this function and not in syncer.go
-	err := uhttp.ClearCaches(ctx)
-	if err != nil {
-		l.Warn("error clearing http caches", zap.Error(err))
-	}
-
 	for _, rb := range b.resourceBuilders {
 		out = append(out, rb.ResourceType(ctx))
 	}
@@ -800,6 +793,17 @@ func (b *builderImpl) RotateCredential(ctx context.Context, request *v2.RotateCr
 	}, nil
 }
 
+func (b *builderImpl) Cleanup(ctx context.Context, request *v2.ConnectorServiceCleanupRequest) (*v2.ConnectorServiceCleanupResponse, error) {
+	l := ctxzap.Extract(ctx)
+	// Clear all http caches at the end of a sync. This must be run in the child process, which is why it's in this function and not in syncer.go
+	err := uhttp.ClearCaches(ctx)
+	if err != nil {
+		l.Warn("error clearing http caches", zap.Error(err))
+	}
+	resp := &v2.ConnectorServiceCleanupResponse{}
+	return resp, err
+}
+
 func (b *builderImpl) CreateAccount(ctx context.Context, request *v2.CreateAccountRequest) (*v2.CreateAccountResponse, error) {
 	start := b.nowFunc()
 	tt := tasks.CreateAccountType