1616import com .park .utmstack .service .application_modules .UtmModuleGroupService ;
1717import com .park .utmstack .service .collectors .CollectorOpsService ;
1818import com .park .utmstack .service .collectors .UtmCollectorService ;
19- import com .park .utmstack .service .dto .collectors .dto .CollectorConfigKeysDTO ;
19+ import com .park .utmstack .service .dto .collectors .dto .CollectorConfigDTO ;
2020import com .park .utmstack .service .dto .collectors .dto .CollectorDTO ;
2121import com .park .utmstack .service .dto .collectors .CollectorModuleEnum ;
2222import com .park .utmstack .service .dto .collectors .dto .ListCollectorsResponseDTO ;
2323import com .park .utmstack .service .dto .network_scan .AssetGroupDTO ;
2424import com .park .utmstack .service .validators .collector .CollectorValidatorService ;
2525import com .park .utmstack .util .UtilResponse ;
26- import com .park .utmstack .web .rest .application_modules .UtmModuleGroupConfigurationResource ;
2726import com .park .utmstack .web .rest .errors .BadRequestAlertException ;
2827import com .park .utmstack .web .rest .errors .InternalServerErrorException ;
2928import com .park .utmstack .web .rest .network_scan .UtmNetworkScanResource ;
@@ -94,9 +93,7 @@ public UtmCollectorResource(CollectorOpsService collectorService,
9493 * persist the configurations.
9594 */
9695 @ PostMapping ("/collector-config" )
97- public ResponseEntity <Void > upsertCollectorConfig (
98- @ Valid @ RequestBody CollectorConfigKeysDTO collectorConfig ,
99- CollectorDTO collectorDTO ) {
96+ public ResponseEntity <Void > upsertCollectorConfig (@ Valid @ RequestBody CollectorConfigDTO collectorConfig ) {
10097 final String ctx = CLASSNAME + ".upsertCollectorConfig" ;
10198
10299 try {
@@ -112,11 +109,11 @@ public ResponseEntity<Void> upsertCollectorConfig(
112109
113110 // Get the actual configuration just in case of error when updating local db.
114111 CollectorConfig cacheConfig = collectorService .getCollectorConfig (
115- ConfigRequest .newBuilder ().setModule (CollectorModule .valueOf (collectorDTO .getModule ().toString ())).build (),
116- AuthResponse .newBuilder ().setId (collectorDTO . getId ()).setKey (collectorDTO .getCollectorKey ()).build ());
112+ ConfigRequest .newBuilder ().setModule (CollectorModule .valueOf (collectorConfig . getCollector () .getModule ().toString ())).build (),
113+ AuthResponse .newBuilder ().setId (collectorConfig . getCollector (). getId ()).setKey (collectorConfig . getCollector () .getCollectorKey ()).build ());
117114 // Map the configurations to gRPC CollectorConfig and try to insert/update the collector config
118115 collectorService .upsertCollectorConfig (collectorService .mapToCollectorConfig (
119- collectorService .mapPasswordConfiguration (collectorConfig .getKeys ()), collectorDTO ));
116+ collectorService .mapPasswordConfiguration (collectorConfig .getKeys ()), collectorConfig . getCollector () ));
120117 // If the update is fine via gRPC, then update the configurations in local db.
121118 try {
122119 moduleGroupConfigurationService .updateConfigurationKeys (collectorConfig .getModuleId (), collectorConfig .getKeys ());
0 commit comments