Add EventIPBlocklisted

mtglib/events.go

@@ -31,3 +31,12 @@ type EventConcurrencyLimited struct {
 func (e EventConcurrencyLimited) StreamID() string {
 	return ""
 }
+
+type EventIPBlocklisted struct {
+	CreatedAt time.Time
+	RemoteIP  net.IP
+}
+
+func (e EventIPBlocklisted) StreamID() string {
+	return ""
+}

mtglib/events_test.go

@@ -36,6 +36,10 @@ func (suite *EventsTestSuite) TestEventConcurrencyLimited() {
 	suite.Empty(mtglib.EventConcurrencyLimited{}.StreamID())
 }
 
+func (suite *EventsTestSuite) TestEventIPBlocklisted() {
+	suite.Empty(mtglib.EventIPBlocklisted{}.StreamID())
+}
+
 func TestEvents(t *testing.T) {
 	t.Parallel()
 	suite.Run(t, &EventsTestSuite{})

mtglib/proxy.go

@@ -52,6 +52,16 @@ func (p *Proxy) Serve(listener net.Listener) error {
 			return fmt.Errorf("cannot accept a new connection: %w", err)
 		}
 
+		if addr := conn.RemoteAddr().(*net.TCPAddr).IP; p.ipBlocklist.Contains(addr) {
+			conn.Close()
+			p.eventStream.Send(p.ctx, EventIPBlocklisted{
+				CreatedAt: time.Now(),
+				RemoteIP:  addr,
+			})
+
+			continue
+		}
+
 		err = p.workerPool.Invoke(conn)
 
 		switch {