Golang : Pass database connection to function called from another package and HTTP Handler
For this tutorial, we will explore how to pass database connection to external function called from another package. To pass a database connection is simple, just treat the database connection as a variable and at the function located in package, use pointer to point to the database connection variable. This tutorial will also cover how to pass database connection to HTTP Handler function as well.
Here we go!
package main
import (
"database/sql"
"fmt"
// without the underscore _, you will get imported but not
// used error message
_ "github.com/go-sql-driver/mysql"
"mypackage/myfunction"
"os"
)
func main() {
conn, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database")
if err != nil {
fmt.Println(err)
os.Exit(1)
}
// Pass conn (sql.DB type variable) to external function
myfunction.PrintTitles(conn)
conn.Close()
}
and then in the external function, remember to use *sql.DB
package myfunction
import (
"database/sql"
"fmt"
_ "github.com/go-sql-driver/mysql"
"os"
)
func PrintTitles(conn *sql.DB) {
// use your own select statement
// this is just an example statement
statement, err := conn.Prepare("select blogtitle from blogs limit 10")
if err != nil {
fmt.Println(err)
os.Exit(1)
}
rows, err := statement.Query() // execute our select statement
if err != nil {
fmt.Println(err)
os.Exit(1)
}
for rows.Next() {
var blogtitle string
rows.Scan(&blogtitle)
fmt.Println("Title :", blogtitle)
}
}
and if you are calling your external function with HTTP Handler, such as :
mx := mux.NewRouter()
mx.HandleFunc("/", PrintTitles(conn))
then you need to wrap your function with a closure. So, instead of :
func PrintTitles(conn *sql.DB) (w http.ResponseWriter, r *http.Request) {
...
}
wrap(close) it with a anonymous function
func PrintTitles(conn *sql.DB) func(w http.ResponseWriter, r *http.Request) {
return func(w http.ResponseWriter, r *http.Request) {
...
// can access conn variable now
}
}
Happy coding!
References :
https://www.socketloop.com/tutorials/golang-call-function-from-another-package
https://www.socketloop.com/tutorials/golang-connect-to-database-mysql-mariadb-server
See also : Golang : Call function from another package
By Adam Ng
IF you gain some knowledge or the information here solved your programming problem. Please consider donating to the less fortunate or some charities that you like. Apart from donation, planting trees, volunteering or reducing your carbon footprint will be great too.
Advertisement
Tutorials
+3.9k Unix/Linux : How to pipe/save output of a command to file?
+15.6k Golang : Capture stdout of a child process and act according to the result
+9.3k JavaScript/JQuery : Detect or intercept enter key pressed example
+10.7k Golang : Split strings into command line arguments
+18.3k Golang : How to Set or Add Header http.ResponseWriter?
+12.7k Golang : convert rune to unicode hexadecimal value and back to rune character
+4.6k Golang : Get S3 or CloudFront object or file information
+25.8k Golang : Convert CSV data to JSON format and save to file
+7.8k Golang : Heap sort example
+8.6k Golang : Eroding and dilating image with OpenCV example
+10.6k Golang : Calculations using complex numbers example
+13.2k Golang : How to shuffle elements in array or slice?