Fix concurrency issues creating/getting the space

main.go

@@ -180,8 +180,9 @@ type Bridge struct {
 	usersByUsername     map[string]*User
 	usersLock           sync.Mutex
 	spaceRooms          map[id.RoomID]*User
-	managementRoomsLock sync.Mutex
+	spaceRoomsLock      sync.Mutex
 	managementRooms     map[id.RoomID]*User
+	managementRoomsLock sync.Mutex
 	portalsByMXID       map[id.RoomID]*Portal
 	portalsByJID        map[database.PortalKey]*Portal
 	portalsLock         sync.Mutex
@@ -480,6 +481,7 @@ func main() {
 	(&Bridge{
 		usersByMXID:         make(map[id.UserID]*User),
 		usersByUsername:     make(map[string]*User),
+		spaceRooms:          make(map[id.RoomID]*User),
 		managementRooms:     make(map[id.RoomID]*User),
 		portalsByMXID:       make(map[id.RoomID]*Portal),
 		portalsByJID:        make(map[database.PortalKey]*Portal),

user.go

@@ -58,6 +58,7 @@ type User struct {
 	RelayWhitelisted bool
 
 	mgmtCreateLock  sync.Mutex
+	spaceCreateLock sync.Mutex
 	connLock        sync.Mutex
 
 	historySyncs     chan *events.HistorySync
@@ -182,10 +183,18 @@ func (bridge *Bridge) NewUser(dbUser *database.User) *User {
 func (user *User) getSpaceRoom() id.RoomID {
 	var roomID id.RoomID
 
-	user.log.Debugln("getSpaceRoom called")
-
 	if len(user.SpaceRoom) == 0 {
 		//TODO check if Spaces creation is enabled by config
+
+		//Create Space
+		user.log.Debugln("Locking to create space.")
+		user.spaceCreateLock.Lock()
+		defer user.spaceCreateLock.Unlock()
+
+		if len(user.SpaceRoom) != 0 {
+			roomID = user.SpaceRoom
+			user.log.Debugln("Returning space after lock" + user.SpaceRoom)
+		} else {
 			creationContent := make(map[string]interface{})
 			creationContent["type"] = "m.space"
 
@@ -206,21 +215,18 @@ func (user *User) getSpaceRoom() id.RoomID {
 				user.log.Errorln("Failed to auto-create space room:", err)
 			} else {
 				user.setSpaceRoom(resp.RoomID)
+				roomID = resp.RoomID
+			}
 		}
 	} else {
 		user.log.Debugln("Space found" + user.SpaceRoom)
 		roomID = user.SpaceRoom
 	}
+
 	return roomID
 }
 
 func (user *User) setSpaceRoom(spaceID id.RoomID) {
-	existingUser, ok := user.bridge.spaceRooms[spaceID]
-	if ok {
-		existingUser.SpaceRoom = ""
-		existingUser.Update()
-	}
-
 	user.SpaceRoom = spaceID
 	user.bridge.spaceRooms[user.SpaceRoom] = user
 	user.Update()