Add runtime stream function to list stream members. (a9a3db7f) · Commits · public-projects / Nakama

server/runtime_nakama_module.go

+74 −0

Original line number	Diff line number	Diff line
		@@ -128,6 +128,7 @@ func (n NakamaModule) Loader(l lua.LState) int {
		"logger_info": n.loggerInfo,
		"logger_warn": n.loggerWarn,
		"logger_error": n.loggerError,
		"stream_user_list": n.streamUserList,
		"stream_user_get": n.streamUserGet,
		"stream_user_join": n.streamUserJoin,
		"stream_user_leave": n.streamUserLeave,
		@@ -1084,6 +1085,79 @@ func (n NakamaModule) loggerError(l lua.LState) int {
		return 1
		}

		func (n NakamaModule) streamUserList(l lua.LState) int {
		// Parse input stream identifier.
		streamTable := l.CheckTable(1)
		if streamTable == nil {
		l.ArgError(1, "expects a valid stream")
		return 0
		}
		stream := PresenceStream{}
		conversionError := ""
		streamTable.ForEach(func(k lua.LValue, v lua.LValue) {
		switch k.String() {
		case "mode":
		if v.Type() != lua.LTNumber {
		conversionError = "stream mode must be a number"
		return
		}
		stream.Mode = uint8(lua.LVAsNumber(v))
		case "subject":
		if v.Type() != lua.LTString {
		conversionError = "stream subject must be a string"
		return
		}
		sid, err := uuid.FromString(v.String())
		if err != nil {
		conversionError = "stream subject must be a valid identifier"
		return
		}
		stream.Subject = sid
		case "descriptor":
		if v.Type() != lua.LTString {
		conversionError = "stream descriptor must be a string"
		return
		}
		did, err := uuid.FromString(v.String())
		if err != nil {
		conversionError = "stream descriptor must be a valid identifier"
		return
		}
		stream.Subject = did
		case "label":
		if v.Type() != lua.LTString {
		conversionError = "stream label must be a string"
		return
		}
		stream.Label = v.String()
		}
		})
		if conversionError != "" {
		l.ArgError(1, conversionError)
		return 0
		}

		presences := n.tracker.ListByStream(stream)

		size := len(presences)
		presencesTable := l.CreateTable(size, size)
		for i, p := range presences {
		presenceTable := l.CreateTable(7, 7)
		presenceTable.RawSetString("user_id", lua.LString(p.UserID.String()))
		presenceTable.RawSetString("session_id", lua.LString(p.ID.SessionID.String()))
		presenceTable.RawSetString("node_id", lua.LString(p.ID.Node))
		presenceTable.RawSetString("hidden", lua.LBool(p.Meta.Hidden))
		presenceTable.RawSetString("persistence", lua.LBool(p.Meta.Persistence))
		presenceTable.RawSetString("username", lua.LString(p.Meta.Username))
		presenceTable.RawSetString("status", lua.LString(p.Meta.Status))

		presencesTable.RawSetInt(i+1, presenceTable)
		}

		l.Push(presencesTable)
		return 1
		}

		func (n NakamaModule) streamUserGet(l lua.LState) int {
		// Parse input User ID.
		userIDString := l.CheckString(1)

server/tracker.go

+4 −4

Original line number	Diff line number	Diff line
		@@ -84,7 +84,7 @@ type Tracker interface {
		UntrackLocalByStream(stream PresenceStream)

		// List the nodes that have at least one presence for the given stream.
		ListNodesForStream(stream PresenceStream) []string
		ListNodesForStream(stream PresenceStream) map[string]struct{}

		// Check if a stream exists (has any presences) or not.
		StreamExists(stream PresenceStream) bool
		@@ -431,15 +431,15 @@ func (t *LocalTracker) UntrackByStream(stream PresenceStream) {
		t.Unlock()
		}

		func (t *LocalTracker) ListNodesForStream(stream PresenceStream) []string {
		func (t *LocalTracker) ListNodesForStream(stream PresenceStream) map[string]struct{} {
		t.RLock()
		_, anyTracked := t.presencesByStream[stream.Mode][stream]
		t.RUnlock()
		if anyTracked {
		// For the local tracker having any presences for this stream is enough.
		return []string{t.name}
		return map[string]struct{}{t.name: struct{}{}}
		}
		return []string{}
		return map[string]struct{}{}
		}

		func (t *LocalTracker) StreamExists(stream PresenceStream) bool {