mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package sqlstore
|
|
|
|
import "github.com/torkelo/grafana-pro/pkg/models"
|
|
|
|
func SaveAccount(account *models.Account) error {
|
|
var err error
|
|
|
|
sess := x.NewSession()
|
|
defer sess.Close()
|
|
|
|
if err = sess.Begin(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if account.Id == 0 {
|
|
_, err = sess.Insert(account)
|
|
} else {
|
|
_, err = sess.Id(account.Id).Update(account)
|
|
}
|
|
|
|
if err != nil {
|
|
sess.Rollback()
|
|
return err
|
|
} else if err = sess.Commit(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetAccount(id int64) (*models.Account, error) {
|
|
var err error
|
|
|
|
var account models.Account
|
|
has, err := x.Id(id).Get(&account)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
} else if has == false {
|
|
return nil, models.ErrAccountNotFound
|
|
}
|
|
|
|
if account.UsingAccountId == 0 {
|
|
account.UsingAccountId = account.Id
|
|
}
|
|
|
|
return &account, nil
|
|
}
|
|
|
|
func GetAccountByLogin(emailOrLogin string) (*models.Account, error) {
|
|
var err error
|
|
|
|
account := &models.Account{Login: emailOrLogin}
|
|
has, err := x.Get(account)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
} else if has == false {
|
|
return nil, models.ErrAccountNotFound
|
|
}
|
|
|
|
return account, nil
|
|
}
|
|
|
|
func GetCollaboratorsForAccount(accountId int64) ([]*models.CollaboratorInfo, error) {
|
|
collaborators := make([]*models.CollaboratorInfo, 0)
|
|
|
|
sess := x.Table("collaborator")
|
|
sess.Join("INNER", "account", "account.id=collaborator.account_Id")
|
|
sess.Where("collaborator.for_account_id=?", accountId)
|
|
err := sess.Find(&collaborators)
|
|
|
|
return collaborators, err
|
|
}
|
|
|
|
func AddCollaborator(collaborator *models.Collaborator) error {
|
|
var err error
|
|
|
|
sess := x.NewSession()
|
|
defer sess.Close()
|
|
|
|
if err = sess.Begin(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err = sess.Insert(collaborator); err != nil {
|
|
sess.Rollback()
|
|
return err
|
|
} else if err = sess.Commit(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetOtherAccountsFor(accountId int64) ([]*models.OtherAccount, error) {
|
|
collaborators := make([]*models.OtherAccount, 0)
|
|
sess := x.Table("collaborator")
|
|
sess.Join("INNER", "account", "collaborator.for_account_id=account.id")
|
|
sess.Where("account_id=?", accountId)
|
|
sess.Cols("collaborator.id", "collaborator.role", "account.email")
|
|
err := sess.Find(&collaborators)
|
|
return collaborators, err
|
|
}
|