Merge branch 'refresh-on-demand' into 'dle-4-0'

feat: Refresh API method

See merge request postgres-ai/database-lab!1019

Author: Bogdan Tsechoev

engine/api/swagger-spec/dblab_openapi.yaml

@@ -224,6 +224,38 @@ paths:
               example:
                 code: "UNAUTHORIZED"
                 message: "Check your verification token."
+  /full-refresh:
+    post:
+      tags:
+        - Instance
+      summary: Trigger full data refresh
+      description: "Initiates a full data refresh."
+      operationId: refresh
+      parameters:
+        - name: Verification-Token
+          in: header
+          required: true
+          schema:
+            type: string
+      responses:
+        200:
+          description: Full refresh has been initiated
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/FullRefresh'
+              example:
+                status: OK
+                message: Full refresh started
+        401:
+          description: Unauthorized access
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/Error'
+              example:
+                code: "UNAUTHORIZED"
+                message: "Check your verification token."
   /snapshot:
     post:
       tags:
@@ -1876,3 +1908,12 @@ components:
           format: date-time
         message:
           type: string
+    FullRefresh:
+      type: object
+      properties:
+        status:
+          type: string
+          example: OK
+        message:
+          type: string
+          example: Full refresh started

engine/internal/srv/routes.go

@@ -950,3 +950,19 @@ func (s *Server) healthCheck(w http.ResponseWriter, _ *http.Request) {
 		return
 	}
 }
+
+func (s *Server) refresh(w http.ResponseWriter, r *http.Request) {
+	go func() {
+		if err := s.Retrieval.FullRefresh(context.Background()); err != nil {
+			log.Err("failed to initiate full refresh", err)
+		}
+	}()
+
+	if err := api.WriteJSON(w, http.StatusOK, models.Response{
+		Status:  models.ResponseOK,
+		Message: "Full refresh started",
+	}); err != nil {
+		api.SendError(w, r, err)
+		return
+	}
+}

engine/internal/srv/server.go

@@ -236,6 +236,9 @@ func (s *Server) InitHandlers() {
 	// Health check.
 	r.HandleFunc("/healthz", s.healthCheck).Methods(http.MethodGet, http.MethodPost)
 
+	// Full refresh
+	r.HandleFunc("/full-refresh", authMW.Authorized(s.refresh)).Methods(http.MethodPost)
+
 	// Show Swagger UI on index page.
 	if err := attachAPI(r); err != nil {
 		log.Err("Cannot load API description.")