Skip to content

Commit

Permalink
add definition of hawk properties (sampling and subset)
Browse files Browse the repository at this point in the history
  • Loading branch information
Dmitriusan committed Dec 4, 2023
1 parent 377be1f commit 46d13e5
Show file tree
Hide file tree
Showing 2 changed files with 43 additions and 1 deletion.
39 changes: 39 additions & 0 deletions hawk/src/main/java/io/irw/hawk/configuration/HawkProperties.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
package io.irw.hawk.configuration;

import io.irw.hawk.dto.merchandise.ProductVariantEnum;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import java.util.List;

@Component
@ConfigurationProperties(prefix = "hawk")
@Data
public class HawkProperties {

private boolean ai;

private Sampling sampling;

private Subset subset;

@Data
public static class Sampling {
private boolean enabled;
private double rate;

}

@Data
public static class Subset {
private boolean enabled;
private Products products;
}

@Data
public static class Products {
private List<ProductVariantEnum> variants;
}

}
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
package io.irw.hawk.scraper.service.scrape;

import io.irw.hawk.configuration.HawkProperties;
import io.irw.hawk.dto.merchandise.ProductVariantEnum;
import java.util.Optional;
import java.util.Queue;
Expand All @@ -14,9 +15,11 @@
public class ScrapeTargetProviderService {
ProductCatalogService productCatalogService;
Queue<ProductVariantEnum> searchTargets;
HawkProperties hawkProperties;

public ScrapeTargetProviderService(ProductCatalogService productCatalogService) {
public ScrapeTargetProviderService(ProductCatalogService productCatalogService, HawkProperties hawkProperties) {
this.productCatalogService = productCatalogService;
this.hawkProperties = hawkProperties;
searchTargets = new java.util.LinkedList<>(productCatalogService.getProducts());
}

Expand Down

0 comments on commit 46d13e5

Please sign in to comment.