Commitea9d6fc

authored and

committed

Add handling for api delays in case of rate limit

1 parent7a3b3e5 commitea9d6fcCopy full SHA for ea9d6fc

File tree

+55

-0

lines changed

+55

-0

lines changed

Lines changed: 38 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,38 @@`
	`1`	`+packageorg.lowcoder.api.framework.filter;`
	`2`	`+`
	`3`	`+importorg.lowcoder.infra.config.repository.ServerConfigRepository;`
	`4`	`+importorg.springframework.beans.factory.annotation.Autowired;`
	`5`	`+importorg.springframework.core.Ordered;`
	`6`	`+importorg.springframework.stereotype.Component;`
	`7`	`+importorg.springframework.web.server.ServerWebExchange;`
	`8`	`+importorg.springframework.web.server.WebFilter;`
	`9`	`+importorg.springframework.web.server.WebFilterChain;`
	`10`	`+importreactor.core.publisher.Mono;`
	`11`	`+`
	`12`	`+importjava.time.Duration;`
	`13`	`+`
	`14`	`+importstaticorg.lowcoder.api.framework.filter.FilterOrder.API_DELAY_FILTER;`
	`15`	`+`
	`16`	`+@Component`
	`17`	`+publicclassAPIDelayFilterimplementsWebFilter,Ordered {`
	`18`	`+`
	`19`	`+@Autowired`
	`20`	`+privateServerConfigRepositoryserverConfigRepository;`
	`21`	`+`
	`22`	`+@Override`
	`23`	`+publicintgetOrder() {`
	`24`	`+returnAPI_DELAY_FILTER.getOrder();`
	`25`	`+ }`
	`26`	`+`
	`27`	`+@Override`
	`28`	`+publicMono<Void>filter(ServerWebExchangeexchange,WebFilterChainchain) {`
	`29`	`+returnserverConfigRepository.findByKey("isRateLimited")`
	`30`	`+ .map(serverConfig -> {`
	`31`	`+if(serverConfig.getValue() !=null &&Boolean.parseBoolean(serverConfig.getValue().toString())) {`
	`32`	`+returnMono.delay(Duration.ofSeconds(5)).block();`
	`33`	`+ }else {`
	`34`	`+returnMono.empty();`
	`35`	`+ }`
	`36`	`+ }).then(chain.filter(exchange));`
	`37`	`+ }`
	`38`	`+}`

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -10,6 +10,8 @@ public enum FilterOrder {`
`10`	`10`	`REQUEST_COST(BEFORE_PROXY_CHAIN),`
`11`	`11`	`THROTTLING(BEFORE_PROXY_CHAIN),`
`12`	`12`
	`13`	`+API_DELAY_FILTER(BEFORE_PROXY_CHAIN),`
	`14`	`+`
`13`	`15`	`// WEB_FILTER_CHAIN_PROXY here`
`14`	`16`
`15`	`17`	`USER_BAN(AFTER_PROXY_CHAIN),`

Lines changed: 15 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -5,9 +5,11 @@`
`5`	`5`	`importjava.util.function.Consumer;`
`6`	`6`
`7`	`7`	`importorg.lowcoder.api.framework.plugin.endpoint.PluginEndpointHandler;`
	`8`	`+importorg.lowcoder.infra.config.repository.ServerConfigRepository;`
`8`	`9`	`importorg.lowcoder.plugin.api.LowcoderServices;`
`9`	`10`	`importorg.lowcoder.plugin.api.PluginEndpoint;`
`10`	`11`	`importorg.lowcoder.plugin.api.event.LowcoderEvent;`
	`12`	`+importorg.springframework.beans.factory.annotation.Autowired;`
`11`	`13`	`importorg.springframework.context.event.EventListener;`
`12`	`14`	`importorg.springframework.stereotype.Component;`
`13`	`15`
`@@ -18,6 +20,9 @@`
`18`	`20`	`publicclassSharedPluginServicesimplementsLowcoderServices`
`19`	`21`	`{`
`20`	`22`	`privatefinalPluginEndpointHandlerpluginEndpointHandler;`
	`23`	`+`
	`24`	`+@Autowired`
	`25`	`+privateServerConfigRepositoryserverConfigRepository;`
`21`	`26`
`22`	`27`	`privateList<Consumer<LowcoderEvent>>eventListeners =newLinkedList<>();`
`23`	`28`
`@@ -41,4 +46,14 @@ public void registerEndpoints(String urlPrefix, List<PluginEndpoint> endpoints)`
`41`	`46`	`{`
`42`	`47`	`pluginEndpointHandler.registerEndpoints(urlPrefix,endpoints);`
`43`	`48`	`}`
	`49`	`+`
	`50`	`+@Override`
	`51`	`+publicvoidsetConfig(Stringkey,Objectvalue) {`
	`52`	`+serverConfigRepository.upsert(key,value).block();`
	`53`	`+}`
	`54`	`+`
	`55`	`+@Override`
	`56`	`+publicObjectgetConfig(Stringkey) {`
	`57`	`+returnserverConfigRepository.findByKey(key).block();`
	`58`	`+}`
`44`	`59`	`}`

Comments

(0)