handle db insert when first resource add cases

kube-tarian · May 30, 2024 · 2434d95 · 2434d95
1 parent 0f8e275
commit 2434d95
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 3 deletions.
diff --git a/capten/agent/internal/api/container_registry.go b/capten/agent/internal/api/container_registry.go
@@ -63,7 +63,7 @@ func (a *Agent) AddContainerRegistry(ctx context.Context, request *captenplugins
 		Labels:       request.Labels,
 		RegistryType: request.RegistryType,
 	}
-	if err := a.as.UpsertContainerRegistry(&ContainerRegistry); err != nil {
+	if err := a.as.AddContainerRegistry(&ContainerRegistry); err != nil {
 		a.log.Errorf("failed to store Container registry to DB, %v", err)
 		return &captenpluginspb.AddContainerRegistryResponse{
 			Status:        captenpluginspb.StatusCode_INTERNAL_ERROR,

diff --git a/capten/agent/internal/api/plugin_cloud_provider_apis.go b/capten/agent/internal/api/plugin_cloud_provider_apis.go
@@ -36,7 +36,7 @@ func (a *Agent) AddCloudProvider(ctx context.Context, request *captenpluginspb.A
 		CloudType: request.CloudType,
 		Labels:    request.Labels,
 	}
-	if err := a.as.UpsertCloudProvider(&CloudProvider); err != nil {
+	if err := a.as.AddCloudProvider(&CloudProvider); err != nil {
 		a.log.Errorf("failed to store cloud provider to DB, %v", err)
 		return &captenpluginspb.AddCloudProviderResponse{
 			Status:        captenpluginspb.StatusCode_INTERNAL_ERROR,

diff --git a/capten/agent/internal/api/plugin_git_apis.go b/capten/agent/internal/api/plugin_git_apis.go
@@ -50,7 +50,7 @@ func (a *Agent) AddGitProject(ctx context.Context, request *captenpluginspb.AddG
 		ProjectUrl: request.ProjectUrl,
 		Labels:     request.Labels,
 	}
-	if err := a.as.UpsertGitProject(&gitProject); err != nil {
+	if err := a.as.AddGitProject(&gitProject); err != nil {
 		a.log.Errorf("failed to store git project to DB, %v", err)
 		return &captenpluginspb.AddGitProjectResponse{
 			Status:        captenpluginspb.StatusCode_INTERNAL_ERROR,

diff --git a/capten/common-pkg/capten-store/cloud_provider.go b/capten/common-pkg/capten-store/cloud_provider.go
@@ -8,6 +8,16 @@ import (
 	"github.com/kube-tarian/kad/capten/common-pkg/pb/captenpluginspb"
 )
 
+func (a *Store) AddCloudProvider(config *captenpluginspb.CloudProvider) error {
+	provider := CloudProvider{
+		ID:             uuid.MustParse(config.Id),
+		CloudType:      config.CloudType,
+		Labels:         config.Labels,
+		LastUpdateTime: time.Now(),
+	}
+	return a.dbClient.Create(&provider)
+}
+
 func (a *Store) UpsertCloudProvider(config *captenpluginspb.CloudProvider) error {
 	if config.Id == "" {
 		provider := CloudProvider{

diff --git a/capten/common-pkg/capten-store/container_registry.go b/capten/common-pkg/capten-store/container_registry.go
@@ -8,6 +8,17 @@ import (
 	"github.com/kube-tarian/kad/capten/common-pkg/pb/captenpluginspb"
 )
 
+func (a *Store) AddContainerRegistry(config *captenpluginspb.ContainerRegistry) error {
+	registry := ContainerRegistry{
+		ID:             uuid.MustParse(config.Id),
+		RegistryURL:    config.RegistryUrl,
+		RegistryType:   config.RegistryType,
+		Labels:         config.Labels,
+		LastUpdateTime: time.Now(),
+	}
+	return a.dbClient.Create(&registry)
+}
+
 func (a *Store) UpsertContainerRegistry(config *captenpluginspb.ContainerRegistry) error {
 	if config.Id == "" {
 		registry := ContainerRegistry{

diff --git a/capten/common-pkg/capten-store/git_projects.go b/capten/common-pkg/capten-store/git_projects.go
@@ -8,6 +8,16 @@ import (
 	"github.com/kube-tarian/kad/capten/common-pkg/pb/captenpluginspb"
 )
 
+func (a *Store) AddGitProject(config *captenpluginspb.GitProject) error {
+	project := GitProject{
+		ID:             uuid.MustParse(config.Id),
+		ProjectURL:     config.ProjectUrl,
+		Labels:         config.Labels,
+		LastUpdateTime: time.Now(),
+	}
+	return a.dbClient.Create(&project)
+}
+
 func (a *Store) UpsertGitProject(config *captenpluginspb.GitProject) error {
 	if config.Id == "" {
 		project := GitProject{