Add some logging to enter/exit of some functions

This commit is contained in:
Jeff Mitchell 2016-06-24 16:10:27 -04:00
parent 185bcb7b48
commit 0802497c8a
2 changed files with 13 additions and 2 deletions

View File

@ -3,6 +3,7 @@ package postgresql
import (
"database/sql"
"fmt"
"log"
"strings"
"sync"
@ -11,10 +12,10 @@ import (
)
func Factory(conf *logical.BackendConfig) (logical.Backend, error) {
return Backend().Setup(conf)
return Backend(conf).Setup(conf)
}
func Backend() *backend {
func Backend(conf *logical.BackendConfig) *backend {
var b backend
b.Backend = &framework.Backend{
Help: strings.TrimSpace(backendHelp),
@ -34,6 +35,7 @@ func Backend() *backend {
Clean: b.ResetDB,
}
b.logger = conf.Logger
return &b
}
@ -42,10 +44,14 @@ type backend struct {
db *sql.DB
lock sync.Mutex
logger *log.Logger
}
// DB returns the database connection.
func (b *backend) DB(s logical.Storage) (*sql.DB, error) {
b.logger.Println("[WARN] postgres/db: enter")
defer b.logger.Println("[WARN] postgres/db: exit")
b.lock.Lock()
defer b.lock.Unlock()
@ -100,6 +106,9 @@ func (b *backend) DB(s logical.Storage) (*sql.DB, error) {
// ResetDB forces a connection next time DB() is called.
func (b *backend) ResetDB() {
b.logger.Println("[WARN] postgres/resetdb: enter")
defer b.logger.Println("[WARN] postgres/resetdb: exit")
b.lock.Lock()
defer b.lock.Unlock()

View File

@ -31,6 +31,8 @@ func pathRoleCreate(b *backend) *framework.Path {
func (b *backend) pathRoleCreateRead(
req *logical.Request, data *framework.FieldData) (*logical.Response, error) {
b.logger.Println("[WARN] postgres/pathRoleCreateRead: enter")
defer b.logger.Println("[WARN] postgres/pathRoleCreateRead: exit")
name := data.Get("name").(string)
// Get the role