mirror of
https://github.com/grafana/grafana.git
synced 2024-11-29 20:24:18 -06:00
6c43eb0b4d
* Split Create User * Use new create user and User from package user * Add service to wire * Making create user work * Replace user from user pkg * One more * Move Insert to orguser Service/Store * Remove unnecessary conversion * Cleaunp * Fix Get User and add fakes * Fixing get org id for user logic, adding fakes and other adjustments * Add some tests for ourguser service and store * Fix insert org logic * Add comment about deprecation * Fix after merge with main * Move orguser service/store to org service/store * Remove orguser from wire * Unimplement new Create user and use User from pkg user * Fix wire generation * Fix lint * Fix lint - use only User and CrateUserCommand from user pkg * Remove User and CreateUserCommand from models * Fix lint 2
107 lines
2.5 KiB
Go
107 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 {
|
|
sessionLogger.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()
|
|
}
|