feat: add sync asset handlers

Makefile

@@ -26,7 +26,7 @@ help:           ##@help show this help
 NAME="github.com/goto/compass"
 VERSION=$(shell git describe --always --tags 2>/dev/null)
 COVERFILE="/tmp/compass.coverprofile"
-PROTON_COMMIT := "a6b2821e8ddd1127a63d3b376f860990d58931da"
+PROTON_COMMIT := "eaca9798d1c1d7b3101ec1259c7e5fb949afba28"
 
 TOOLS_MOD_DIR = ./tools
 TOOLS_DIR = $(abspath ./.tools)

cli/server.go

@@ -132,9 +132,16 @@ func runServer(ctx context.Context, cfg *Config) error {
 		return fmt.Errorf("create new lineage repository: %w", err)
 	}
 
+	jobRepository, err := postgres.NewJobRepository(pgClient)
+	if err != nil {
+		return fmt.Errorf("create new job repository: %w", err)
+	}
+
 	wrkr, err := initAssetWorker(ctx, workermanager.Deps{
 		Config:        cfg.Worker,
 		DiscoveryRepo: discoveryRepository,
+		AssetRepo:     assetRepository,
+		JobRepo:       jobRepository,
 		Logger:        logger,
 	})
 	if err != nil {

core/asset/discovery.go

@@ -12,6 +12,10 @@ type DiscoveryRepository interface {
 	Search(ctx context.Context, cfg SearchConfig) (results []SearchResult, err error)
 	Suggest(ctx context.Context, cfg SearchConfig) (suggestions []string, err error)
 	GroupAssets(ctx context.Context, cfg GroupConfig) (results []GroupResult, err error)
+	Clone(ctx context.Context, indexName, clonedIndexName string) error
+	UpdateAlias(ctx context.Context, indexName, alias string) error
+	DeleteByIndexName(ctx context.Context, indexName string) error
+	UpdateIndexSettings(ctx context.Context, indexName string, body string) error
 }
 
 // GroupConfig represents a group query along

core/asset/service.go

@@ -24,6 +24,7 @@ type Service struct {
 type Worker interface {
 	EnqueueIndexAssetJob(ctx context.Context, ast Asset) error
 	EnqueueDeleteAssetJob(ctx context.Context, urn string) error
+	EnqueueSyncAssetJob(ctx context.Context, service string) error
 	Close() error
 }
 
@@ -229,6 +230,13 @@ func (s *Service) SuggestAssets(ctx context.Context, cfg SearchConfig) (suggesti
 	return s.discoveryRepository.Suggest(ctx, cfg)
 }
 
+func (s *Service) SyncAssets(ctx context.Context, services []string) error {
+	for _, service := range services {
+		s.worker.EnqueueSyncAssetJob(ctx, service)
+	}
+	return nil
+}
+
 func (s *Service) instrumentAssetOp(ctx context.Context, op, id string, err error) {
 	identifier := "URN"
 	if isValidUUID(id) {

core/job/job.go

@@ -0,0 +1,22 @@
+package job
+
+import (
+	"context"
+	"time"
+)
+
+type Repository interface {
+	GetSyncJobsByService(ctx context.Context, serviceName string) ([]JobsQueue, error)
+}
+
+type JobsQueue struct {
+	ID          string    `db:"id"`
+	Type        string    `db:"type"`
+	LastError   string    `db:"last_error"`
+	AttemptsDo  int32     `db:"attempts_do"`
+	Payload     []byte    `db:"payload"`
+	RunAt       time.Time `db:"run_at"`
+	CreatedAt   time.Time `db:"created_at"`
+	UpdatedAt   time.Time `db:"updated_at"`
+	LastAttempt time.Time `db:"last_attempt"`
+}

core/job/service.go

@@ -0,0 +1,18 @@
+package job
+
+import "context"
+
+func NewService(jobRepository Repository) *Service {
+	return &Service{
+		jobRepository: jobRepository,
+	}
+}
+
+type Service struct {
+	jobRepository Repository
+}
+
+// GetSyncJobsByService handles business process to get tags by its asset id
+func (s *Service) GetSyncJobsByService(ctx context.Context, serviceName string) ([]JobsQueue, error) {
+	return s.jobRepository.GetSyncJobsByService(ctx, serviceName)
+}

internal/server/v1beta1/asset.go

@@ -39,6 +39,8 @@ type AssetService interface {
 	SuggestAssets(ctx context.Context, cfg asset.SearchConfig) (suggestions []string, err error)
 
 	AddProbe(ctx context.Context, assetURN string, probe *asset.Probe) error
+
+	SyncAssets(ctx context.Context, services []string) error
 }
 
 func (server *APIServer) GetAllAssets(ctx context.Context, req *compassv1beta1.GetAllAssetsRequest) (*compassv1beta1.GetAllAssetsResponse, error) {
@@ -344,6 +346,13 @@ func (server *APIServer) CreateAssetProbe(ctx context.Context, req *compassv1bet
 	}, nil
 }
 
+func (server *APIServer) SyncAssets(ctx context.Context, req *compassv1beta1.SyncAssetsRequest) (*compassv1beta1.SyncAssetsResponse, error) {
+
+	server.assetService.SyncAssets(ctx, req.GetServices())
+
+	return nil, nil
+}
+
 func (server *APIServer) upsertAsset(
 	ctx context.Context,
 	ast asset.Asset,