vendor dependencies with dep

vendor/github.com/go-pg/pg/hook.go

@@ -0,0 +1,142 @@
+package pg
+
+import (
+	"fmt"
+	"runtime"
+	"strings"
+	"time"
+
+	"github.com/go-pg/pg/orm"
+)
+
+type dummyDB struct {
+	orm.DB
+}
+
+var _ orm.DB = dummyDB{}
+
+func (dummyDB) FormatQuery(dst []byte, query string, params ...interface{}) []byte {
+	return append(dst, query...)
+}
+
+type QueryProcessedEvent struct {
+	StartTime time.Time
+	Func      string
+	File      string
+	Line      int
+
+	DB     orm.DB
+	Query  interface{}
+	Params []interface{}
+	Result orm.Result
+	Error  error
+}
+
+func (ev *QueryProcessedEvent) UnformattedQuery() (string, error) {
+	b, err := queryString(ev.Query)
+	if err != nil {
+		return "", err
+	}
+	return string(b), nil
+}
+
+func (ev *QueryProcessedEvent) FormattedQuery() (string, error) {
+	b, err := appendQuery(nil, ev.DB, ev.Query, ev.Params...)
+	if err != nil {
+		return "", err
+	}
+	return string(b), nil
+}
+
+func queryString(query interface{}) ([]byte, error) {
+	switch query := query.(type) {
+	case orm.QueryAppender:
+		query = query.Copy()
+		query.Query().DB(dummyDB{})
+		return query.AppendQuery(nil)
+	case string:
+		return dummyDB{}.FormatQuery(nil, query), nil
+	default:
+		return nil, fmt.Errorf("pg: can't append %T", query)
+	}
+}
+
+type queryProcessedHook func(*QueryProcessedEvent)
+
+// OnQueryProcessed calls the fn with QueryProcessedEvent
+// when query is processed.
+func (db *DB) OnQueryProcessed(fn func(*QueryProcessedEvent)) {
+	db.queryProcessedHooks = append(db.queryProcessedHooks, fn)
+}
+
+func (db *DB) queryProcessed(
+	ormDB orm.DB,
+	start time.Time,
+	query interface{},
+	params []interface{},
+	res orm.Result,
+	err error,
+) {
+	if len(db.queryProcessedHooks) == 0 {
+		return
+	}
+
+	funcName, file, line := fileLine(2)
+	event := &QueryProcessedEvent{
+		StartTime: start,
+		Func:      funcName,
+		File:      file,
+		Line:      line,
+
+		DB:     ormDB,
+		Query:  query,
+		Params: params,
+		Result: res,
+		Error:  err,
+	}
+	for _, hook := range db.queryProcessedHooks {
+		hook(event)
+	}
+}
+
+const packageName = "github.com/go-pg/pg"
+
+func fileLine(depth int) (string, string, int) {
+	for i := depth; ; i++ {
+		pc, file, line, ok := runtime.Caller(i)
+		if !ok {
+			break
+		}
+		if strings.Contains(file, packageName) {
+			continue
+		}
+		_, funcName := packageFuncName(pc)
+		return funcName, file, line
+	}
+	return "", "", 0
+}
+
+func packageFuncName(pc uintptr) (string, string) {
+	f := runtime.FuncForPC(pc)
+	if f == nil {
+		return "", ""
+	}
+
+	packageName := ""
+	funcName := f.Name()
+
+	if ind := strings.LastIndex(funcName, "/"); ind > 0 {
+		packageName += funcName[:ind+1]
+		funcName = funcName[ind+1:]
+	}
+	if ind := strings.Index(funcName, "."); ind > 0 {
+		packageName += funcName[:ind]
+		funcName = funcName[ind+1:]
+	}
+
+	return packageName, funcName
+}
+
+func copyQueryProcessedHooks(s []queryProcessedHook) []queryProcessedHook {
+	return s[:len(s):len(s)]
+}