feat: adding configurable parameter for time in the listener (#13)

examples/base/main.go

@@ -22,7 +22,9 @@ func main() {
 		TableName: "messenger_messages",
 	}
 
-	err := gosumer.Listen(database, processMessage, Message{})
+	sec := 5
+
+	err := gosumer.Listen(database, processMessage, Message{}, sec)
 
 	if err != nil {
 		log.Fatal(err)

pgsql_transport.go

@@ -31,7 +31,7 @@ func (database PgDatabase) connect() error {
 	return nil
 }
 
-func (database PgDatabase) listen(fn process, message any) error {
+func (database PgDatabase) listen(fn process, message any, sec int) error {
 	err := database.connect()
 
 	if err != nil {
@@ -40,7 +40,7 @@ func (database PgDatabase) listen(fn process, message any) error {
 
 	defer pool.Close()
 
-	database.listenEvery(5, fn, message)
+	database.listenEvery(sec, fn, message)
 
 	log.Printf("Successfully connected to the database!")
 

pgsql_transport_test.go

@@ -85,7 +85,7 @@ func TestPgListen(t *testing.T) {
 	defer pool.Close()
 
 	go func() {
-		err := database.listen(processMessage, Message{})
+		err := database.listen(processMessage, Message{}, 5)
 		if err != nil {
 			t.Errorf("Expected no error, got %v", err)
 		}

transport.go

@@ -6,7 +6,7 @@ import (
 
 type Transport interface {
 	connect() error
-	listen(fn process, message any) error
+	listen(fn process, message any, sec int) error
 }
 
 func formatMessage(message string, msg any) (any, error) {
@@ -17,8 +17,8 @@ func formatMessage(message string, msg any) (any, error) {
 	return msg, nil
 }
 
-func Listen(transport Transport, fn process, message any) error {
-	err := transport.listen(fn, message)
+func Listen(transport Transport, fn process, message any, sec int) error {
+	err := transport.listen(fn, message, sec)
 	if err != nil {
 		return err
 	}

transport_test.go

@@ -31,7 +31,7 @@ func (transport TestTransport) connect() error {
 	return nil
 }
 
-func (transport TestTransport) listen(fn process, message any) error {
+func (transport TestTransport) listen(fn process, message any, sec int) error {
 	go fn(message, make(chan error))
 
 	return nil
@@ -51,7 +51,7 @@ type Message struct {
 func TestListen(t *testing.T) {
 	transport := TestTransport{}
 
-	err := Listen(transport, processMessage, Message{})
+	err := Listen(transport, processMessage, Message{}, 5)
 	if err != nil {
 		t.Errorf("Expected no error, got %v", err)
 	}