mirror of
https://github.com/grafana/grafana.git
synced 2025-02-14 17:43:35 -06:00
862a6a2fa6
Introduces a FromContext method on the log.Logger interface that allows contextual key/value pairs to be attached, e.g. per request, so that any logger using this API will automatically get the per request context attached. The proposal makes the traceID available for contextual logger , if available, and would allow logs originating from a certain HTTP request to be correlated with traceID. In addition, when tracing not enabled, skip adding traceID=00000000000000000000000000000000 to logs.
108 lines
2.5 KiB
Go
108 lines
2.5 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
|
|
"xorm.io/xorm"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
)
|
|
|
|
var sessionLogger = log.New("sqlstore.session")
|
|
|
|
type DBSession struct {
|
|
*xorm.Session
|
|
transactionOpen bool
|
|
events []interface{}
|
|
}
|
|
|
|
type DBTransactionFunc func(sess *DBSession) error
|
|
|
|
func (sess *DBSession) publishAfterCommit(msg interface{}) {
|
|
sess.events = append(sess.events, msg)
|
|
}
|
|
|
|
func (sess *DBSession) PublishAfterCommit(msg interface{}) {
|
|
sess.events = append(sess.events, msg)
|
|
}
|
|
|
|
// NewSession returns a new DBSession
|
|
func (ss *SQLStore) NewSession(ctx context.Context) *DBSession {
|
|
sess := &DBSession{Session: ss.engine.NewSession()}
|
|
sess.Session = sess.Session.Context(ctx)
|
|
return sess
|
|
}
|
|
|
|
func (ss *SQLStore) newSession(ctx context.Context) *DBSession {
|
|
sess := &DBSession{Session: ss.engine.NewSession()}
|
|
sess.Session = sess.Session.Context(ctx)
|
|
|
|
return sess
|
|
}
|
|
|
|
func startSessionOrUseExisting(ctx context.Context, engine *xorm.Engine, beginTran bool) (*DBSession, bool, error) {
|
|
value := ctx.Value(ContextSessionKey{})
|
|
var sess *DBSession
|
|
sess, ok := value.(*DBSession)
|
|
|
|
if ok {
|
|
ctxLogger := sessionLogger.FromContext(ctx)
|
|
ctxLogger.Debug("reusing existing session", "transaction", sess.transactionOpen)
|
|
sess.Session = sess.Session.Context(ctx)
|
|
return sess, false, nil
|
|
}
|
|
|
|
newSess := &DBSession{Session: engine.NewSession(), transactionOpen: beginTran}
|
|
if beginTran {
|
|
err := newSess.Begin()
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
}
|
|
|
|
newSess.Session = newSess.Session.Context(ctx)
|
|
return newSess, true, nil
|
|
}
|
|
|
|
// WithDbSession calls the callback with a session.
|
|
func (ss *SQLStore) WithDbSession(ctx context.Context, callback DBTransactionFunc) error {
|
|
return withDbSession(ctx, ss.engine, callback)
|
|
}
|
|
|
|
func withDbSession(ctx context.Context, engine *xorm.Engine, callback DBTransactionFunc) error {
|
|
sess, isNew, err := startSessionOrUseExisting(ctx, engine, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if isNew {
|
|
defer sess.Close()
|
|
}
|
|
return callback(sess)
|
|
}
|
|
|
|
func (sess *DBSession) InsertId(bean interface{}) (int64, error) {
|
|
table := sess.DB().Mapper.Obj2Table(getTypeName(bean))
|
|
|
|
if err := dialect.PreInsertId(table, sess.Session); err != nil {
|
|
return 0, err
|
|
}
|
|
id, err := sess.Session.InsertOne(bean)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if err := dialect.PostInsertId(table, sess.Session); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return id, nil
|
|
}
|
|
|
|
func getTypeName(bean interface{}) (res string) {
|
|
t := reflect.TypeOf(bean)
|
|
for t.Kind() == reflect.Ptr {
|
|
t = t.Elem()
|
|
}
|
|
return t.Name()
|
|
}
|