Add the task service for scan task.

This commit is contained in:
lishuang
2020-07-11 20:46:47 +08:00
parent 7fe673068b
commit 7f394b327b
5 changed files with 184 additions and 17 deletions

View File

@ -15,6 +15,7 @@ type PreferenceController struct {
preferenceDao *PreferenceDao
matterDao *MatterDao
preferenceService *PreferenceService
taskService *TaskService
}
func (this *PreferenceController) Init() {
@ -34,6 +35,11 @@ func (this *PreferenceController) Init() {
this.preferenceService = b
}
b = core.CONTEXT.GetBean(this.taskService)
if b, ok := b.(*TaskService); ok {
this.taskService = b
}
}
func (this *PreferenceController) RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request) {
@ -175,6 +181,9 @@ func (this *PreferenceController) EditScanConfig(writer http.ResponseWriter, req
preference = this.preferenceService.Save(preference)
//reinit the scan task.
this.taskService.InitScanTask()
return this.Success(preference)
}