Function main

cmd/database/database.go:19–53

()

var conf = config.New()

func main() {
	fmt.Println("[Database service]")

	// Postgres connection
	connPG, err := sql.Open("postgres", conf.PostgresURL+"?sslmode=disable")
	if err != nil {
		log.Fatalf("postgres connection: %s", err)
	}
	defer connPG.Close()

	// The table is not created when deployed on AWS RDS.
	_, err = connPG.Exec("create table if not exists messages (id serial primary key, message text not null, created timestamp not null)")
	if err != nil {
		log.Fatalf("create table: %s", err)
	}

	// RabbitMQ connection
	connMQ, err := rabbit.GetConn(conf.RabbitURL)
	if err != nil {
		log.Fatalf("rabbit connection: %s", err)
	}
	defer connMQ.Close()

	err = connMQ.DeclareTopicExchange(conf.Exchange)
	if err != nil {
		log.Fatalf("declare exchange: %s", err)
	}

	// Start a Rabbit consumer with a message processing handler.
	connMQ.StartConsumer(conf.Exchange, conf.QueueDB, conf.KeyDB, func(d amqp.Delivery) bool {
		return insertToDB(d, connPG)
	})

	select {}
}

// insertToDB inserts a Rabbit message into a Postgres database.
func insertToDB(d amqp.Delivery, c *sql.DB) bool {