package services import ( "errors" "fmt" "github.com/emvi/logbuch" "github.com/muety/wakapi/config" "github.com/muety/wakapi/models" "github.com/muety/wakapi/repositories" "sync" ) type AliasService struct { config *config.Config repository repositories.IAliasRepository } func NewAliasService(aliasRepo repositories.IAliasRepository) *AliasService { return &AliasService{ config: config.Get(), repository: aliasRepo, } } var userAliases sync.Map func (srv *AliasService) IsInitialized(userId string) bool { if _, ok := userAliases.Load(userId); ok { return true } return false } func (srv *AliasService) InitializeUser(userId string) error { aliases, err := srv.repository.GetByUser(userId) if err == nil { userAliases.Store(userId, aliases) } return err } func (srv *AliasService) MayInitializeUser(userId string) { if err := srv.InitializeUser(userId); err != nil { logbuch.Error("failed to initialize user alias map for user %s", userId) } } func (srv *AliasService) GetByUser(userId string) ([]*models.Alias, error) { if !srv.IsInitialized(userId) { srv.MayInitializeUser(userId) } if aliases, ok := userAliases.Load(userId); ok { return aliases.([]*models.Alias), nil } else { return nil, errors.New(fmt.Sprintf("no user aliases loaded for user %s", userId)) } } func (srv *AliasService) GetByUserAndKeyAndType(userId, key string, summaryType uint8) ([]*models.Alias, error) { if !srv.IsInitialized(userId) { srv.MayInitializeUser(userId) } if aliases, ok := userAliases.Load(userId); ok { filteredAliases := make([]*models.Alias, 0, len(aliases.([]*models.Alias))) for _, a := range aliases.([]*models.Alias) { if a.Key == key && a.Type == summaryType { filteredAliases = append(filteredAliases, a) } } return filteredAliases, nil } else { return nil, errors.New(fmt.Sprintf("no user aliases loaded for user %s", userId)) } } func (srv *AliasService) GetAliasOrDefault(userId string, summaryType uint8, value string) (string, error) { if !srv.IsInitialized(userId) { srv.MayInitializeUser(userId) } if aliases, ok := userAliases.Load(userId); ok { for _, a := range aliases.([]*models.Alias) { if a.Type == summaryType && a.Value == value { return a.Key, nil } } } return value, nil } func (srv *AliasService) Create(alias *models.Alias) (*models.Alias, error) { result, err := srv.repository.Insert(alias) if err != nil { return nil, err } go srv.MayInitializeUser(alias.UserID) return result, nil } func (srv *AliasService) Delete(alias *models.Alias) error { if alias.UserID == "" { return errors.New("no user id specified") } err := srv.repository.Delete(alias.ID) go srv.MayInitializeUser(alias.UserID) return err } func (srv *AliasService) DeleteMulti(aliases []*models.Alias) error { ids := make([]uint, len(aliases)) affectedUsers := make(map[string]bool) for i, a := range aliases { if a.UserID == "" { return errors.New("no user id specified") } affectedUsers[a.UserID] = true ids[i] = a.ID } err := srv.repository.DeleteBatch(ids) for k := range affectedUsers { go srv.MayInitializeUser(k) } return err }