mycontext.go

package main

type MyContext struct {
	AuthService    AuthService
	UserStore      UserStore
	PasswordStore  PasswordStore
	RoomStore      RoomStore
	RoomProvider   RoomProvider
	RoomProviderGC RoomProviderGC
	RTMT           RealtimeMessageTransport
	Hub            *Hub
}

func newMyContextForTest() *MyContext {
	userStore := newUserStoreImpl()
	passwordStore := newPasswordStoreImpl()
	authService := newAuthServiceImpl()
	authService.SetUserStore(userStore)
	authService.SetPasswordStore(passwordStore)
	roomStore := NewRoomStoreImpl()
	roomProvider := NewRoomProviderImpl(roomStore)
	roomProviderGC := NewRoomProviderGCImpl(roomProvider, userStore)
	rtmt := NewRealtimeMessageTransportWebSocket()
	rtmt.ListenTestChannel()
	hub := NewHub()
	go hub.Run()
	go hub.HandleMessage()
	return &MyContext{
		UserStore:      userStore,
		PasswordStore:  passwordStore,
		AuthService:    authService,
		RoomStore:      roomStore,
		RoomProvider:   roomProvider,
		RoomProviderGC: roomProviderGC,
		RTMT:           rtmt,
		Hub:            hub,
	}
}

func newMyContext() *MyContext {
	options := getDatabaseConnectionVariables()
	db := CreateDatabaseConnection(options)
	userStore := newUserStoreDBImpl(db)
	passwordStore := newPasswordStoreDBImpl(db)
	authService := newAuthServiceImpl()
	authService.SetUserStore(userStore)
	authService.SetPasswordStore(passwordStore)
	roomStore := NewRoomStoreDatabaseImpl(db)
	roomProvider := NewRoomProviderImpl(roomStore)
	roomProviderGC := NewRoomProviderGCImpl(roomProvider, userStore)
	rtmt := NewRealtimeMessageTransportWebSocket()
	rtmt.ListenTestChannel()
	hub := NewHub()
	go hub.Run()
	go hub.HandleMessage()
	return &MyContext{
		UserStore:      userStore,
		PasswordStore:  passwordStore,
		AuthService:    authService,
		RoomStore:      roomStore,
		RoomProvider:   roomProvider,
		RoomProviderGC: roomProviderGC,
		RTMT:           rtmt,
		Hub:            hub,
	}
}