platform/packages/security/example/rate_limit_redis.dart
2022-04-23 12:21:39 +08:00

40 lines
1.5 KiB
Dart

import 'package:angel3_redis/angel3_redis.dart';
import 'package:angel3_framework/angel3_framework.dart';
import 'package:angel3_production/angel3_production.dart';
import 'package:angel3_security/angel3_security.dart';
import 'package:resp_client/resp_client.dart';
import 'package:resp_client/resp_commands.dart';
import 'package:resp_client/resp_server.dart';
// We run this through angel_production, so that we can have
// multiple instances, all using the same Redis queue.
void main(List<String> args) =>
Runner('rate_limit_redis', configureServer).run(args);
void configureServer(Angel app) async {
// Create a simple rate limiter that limits users to 10
// queries per 30 seconds.
//
// In this case, we rate limit users by IP address.
//
// Our Redis store will be used to manage windows.
var connection = await connectSocket('localhost');
var client = RespClient(connection);
var service = RedisService(RespCommandsTier2(client),
prefix: 'rate_limit_redis_example');
var rateLimiter = ServiceRateLimiter(
10, Duration(seconds: 30), service, (req, res) => req.ip);
// `RateLimiter.handleRequest` is a middleware, and can be used anywhere
// a middleware can be used. In this case, we apply the rate limiter to
// *all* incoming requests.
app.fallback(rateLimiter.handleRequest);
// Basic routes.
app
..get('/', (req, res) {
var instance = req.container!.make<InstanceInfo>();
res.writeln('This is instance ${instance.id}.');
})
..fallback((req, res) => throw AngelHttpException.notFound());
}