func (app Application) StartWorkers() { zonedUAAClient := uaa.NewZonedUAAClient(app.env.UAAClientID, app.env.UAAClientSecret, app.env.VerifySSL, UAAPublicKey) WorkerGenerator{ InstanceIndex: app.env.VCAPApplication.InstanceIndex, Count: WorkerCount, }.Work(func(i int) Worker { cloak, err := conceal.NewCloak(app.env.EncryptionKey) if err != nil { panic(err) } v1Workflow := postal.NewV1Process(postal.V1ProcessConfig{ DBTrace: app.env.DBLoggingEnabled, UAAHost: app.env.UAAHost, Sender: app.env.Sender, Domain: app.env.Domain, Packager: postal.NewPackager(app.mother.TemplatesLoader(), cloak), MailClient: app.mother.MailClient(), Database: app.mother.Database(), TokenLoader: uaa.NewTokenLoader(zonedUAAClient), UserLoader: postal.NewUserLoader(zonedUAAClient), KindsRepo: app.mother.KindsRepo(), ReceiptsRepo: app.mother.ReceiptsRepo(), UnsubscribesRepo: app.mother.UnsubscribesRepo(), GlobalUnsubscribesRepo: app.mother.GlobalUnsubscribesRepo(), MessageStatusUpdater: postal.NewMessageStatusUpdater(app.mother.MessagesRepo()), DeliveryFailureHandler: postal.NewDeliveryFailureHandler(), }) database := v2models.NewDatabase(app.mother.SQLDatabase(), v2models.Config{}) messageStatusUpdater := postal.NewV2MessageStatusUpdater(v2models.NewMessagesRepository(util.NewClock())) v2Workflow := postal.NewV2Workflow(app.mother.MailClient(), postal.NewPackager(app.mother.TemplatesLoader(), cloak), postal.NewUserLoader(zonedUAAClient), uaa.NewTokenLoader(zonedUAAClient), messageStatusUpdater, database, app.env.Sender, app.env.Domain, app.env.UAAHost) worker := postal.NewDeliveryWorker(v1Workflow, v2Workflow, postal.DeliveryWorkerConfig{ ID: i, UAAHost: app.env.UAAHost, Logger: app.mother.Logger(), Queue: app.mother.Queue(), DBTrace: app.env.DBLoggingEnabled, Database: database, StrategyDeterminer: strategy.NewStrategyDeterminer(app.mother.UserStrategy(), app.mother.SpaceStrategy(), app.mother.OrganizationStrategy(), app.mother.EmailStrategy()), DeliveryFailureHandler: postal.NewDeliveryFailureHandler(), MessageStatusUpdater: messageStatusUpdater, }) return &worker }) }
func (ware DatabaseAllocator) ServeHTTP(w http.ResponseWriter, req *http.Request, context stack.Context) bool { db := models.NewDatabase(ware.DB, models.Config{}) logger := gorpCompatibleLogger{ logger: context.Get("logger").(lager.Logger).WithData(lager.Data{ VCAPRequestIDKey: context.Get(VCAPRequestIDKey), }), } if ware.Trace { db.TraceOn("", logger) } context.Set("database", db) return true }
"github.com/cloudfoundry-incubator/notifications/testing/mocks" "github.com/cloudfoundry-incubator/notifications/v2/models" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) var _ = Describe("Messages Repository", func() { var ( repo models.MessagesRepository conn *db.Connection clock *mocks.Clock ) BeforeEach(func() { database := models.NewDatabase(sqlDB, models.Config{}) helpers.TruncateTables(db.NewDatabase(sqlDB, db.Config{})) conn = database.Connection().(*db.Connection) conn.AddTableWithName(models.Message{}, "messages") clock = mocks.NewClock() repo = models.NewMessagesRepository(clock) }) Describe("CountByStatus", func() { BeforeEach(func() { err := conn.Insert(&models.Message{ ID: "message-id-123", CampaignID: "some-campaign-id", Status: postal.StatusDelivered, })