set introspection status for upload repos

content-services · Jul 12, 2024 · 2209358 · 2209358
1 parent 610fdc9
commit 2209358
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 4 deletions.
diff --git a/api/docs.go b/api/docs.go
@@ -393,7 +393,7 @@ const docTemplate = `{
                     },
                     {
                         "type": "string",
-                        "description": "A comma separated list of statuses to control api response. Statuses can include ` + "`" + `pending` + "`" + `, ` + "`" + `valid` + "`" + `, ` + "`" + `invalid` + "`" + `, ` + "`" + `unavailable` + "`" + `.",
+                        "description": "A comma separated list of statuses to control api response. Statuses can include ` + "`" + `Pending` + "`" + `, ` + "`" + `Valid` + "`" + `, ` + "`" + `Invalid` + "`" + `, ` + "`" + `Unavailable` + "`" + `.",
                         "name": "status",
                         "in": "query"
                     },

diff --git a/api/openapi.json b/api/openapi.json
@@ -1782,7 +1782,7 @@
                         }
                     },
                     {
-                        "description": "A comma separated list of statuses to control api response. Statuses can include `pending`, `valid`, `invalid`, `unavailable`.",
+                        "description": "A comma separated list of statuses to control api response. Statuses can include `Pending`, `Valid`, `Invalid`, `Unavailable`.",
                         "in": "query",
                         "name": "status",
                         "schema": {

diff --git a/pkg/dao/repository_configs_test.go b/pkg/dao/repository_configs_test.go
@@ -126,6 +126,8 @@ func (suite *RepositoryConfigSuite) TestCreateUpload() {
 	assert.NoError(suite.T(), err)
 	assert.True(suite.T(), repo.UUID != "")
 	assert.NotEqual(suite.T(), repo.UUID, repo2.UUID)
+	assert.NotEmpty(suite.T(), repo.LastIntrospectionTime)
+	assert.NotEmpty(suite.T(), repo.LastIntrospectionStatus)
 }
 
 func (suite *RepositoryConfigSuite) TestCreateUploadNoSnap() {

diff --git a/pkg/handler/repositories.go b/pkg/handler/repositories.go
@@ -77,7 +77,7 @@ func getAccountIdOrgId(c echo.Context) (string, string) {
 // @Param		 url query string false "A comma separated list of URLs to control api response."
 // @Param		 uuid query string false "A comma separated list of UUIDs to control api response."
 // @Param		 sort_by query string false "Sort the response data based on specific repository parameters. Sort criteria can include `name`, `url`, `status`, and `package_count`."
-// @Param        status query string false "A comma separated list of statuses to control api response. Statuses can include `pending`, `valid`, `invalid`, `unavailable`."
+// @Param        status query string false "A comma separated list of statuses to control api response. Statuses can include `Pending`, `Valid`, `Invalid`, `Unavailable`."
 // @Param		 origin query string false "A comma separated list of origins to filter api response. Origins can include `red_hat` and `external`."
 // @Param		 content_type query string false "content type of a repository to filter on (rpm)"
 // @Accept       json
@@ -529,7 +529,7 @@ func (rh *RepositoryHandler) introspect(c echo.Context) error {
 
 	var repoUpdate dao.RepositoryUpdate
 	count := 0
-	lastIntrospectionStatus := "Pending"
+	lastIntrospectionStatus := config.StatusPending
 	if req.ResetCount {
 		repoUpdate = dao.RepositoryUpdate{
 			UUID:                      repo.UUID,

diff --git a/pkg/models/repository.go b/pkg/models/repository.go
@@ -38,6 +38,13 @@ func (r *Repository) BeforeCreate(tx *gorm.DB) (err error) {
 		return err
 	}
 	r.URL = CleanupURL(r.URL)
+	if r.Origin == config.OriginUpload {
+		r.LastIntrospectionStatus = config.StatusValid
+		time := time.Now()
+		r.LastIntrospectionTime = &time
+		r.LastIntrospectionSuccessTime = &time
+		r.LastIntrospectionUpdateTime = &time
+	}
 	return nil
 }