fix: unable to monitor data source status issue

crawlab-team · Jun 3, 2023 · 9c40ef4 · 9c40ef4
1 parent 68a112d
commit 9c40ef4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/controllers/data_source.go b/controllers/data_source.go
@@ -69,6 +69,23 @@ func (ctr *dataSourceController) Post(c *gin.Context) {
 	HandleSuccess(c)
 }
 
+func (ctr *dataSourceController) Put(c *gin.Context) {
+	// data source
+	var _ds models.DataSource
+	if err := c.ShouldBindJSON(&_ds); err != nil {
+		HandleErrorBadRequest(c, err)
+		return
+	}
+
+	if err := delegate.NewModelDelegate(&_ds).Save(); err != nil {
+		HandleErrorInternalServerError(c, err)
+		return
+	}
+
+	// check data source status
+	go func() { _ = ctr.ctx.dsSvc.CheckStatus(_ds.Id) }()
+}
+
 type dataSourceContext struct {
 	dsSvc interfaces.DataSourceService
 }

diff --git a/ds/service.go b/ds/service.go
@@ -288,6 +288,9 @@ func NewDataSourceService(opts ...DataSourceServiceOption) (svc2 interfaces.Data
 	// initialize
 	svc.Init()
 
+	// start
+	svc.Start()
+
 	return svc, nil
 }