mirror of
https://github.com/1f349/dendrite.git
synced 2024-11-10 06:53:00 +00:00
d5876abbe9
This adds the main component of the fulltext search. This PR doesn't do anything yet, besides creating an empty fulltextindex folder if enabled. Indexing events is done in a separate PR.
114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
package jetstream
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"time"
|
|
|
|
"github.com/nats-io/nats.go"
|
|
)
|
|
|
|
const (
|
|
UserID = "user_id"
|
|
RoomID = "room_id"
|
|
EventID = "event_id"
|
|
)
|
|
|
|
var (
|
|
InputRoomEvent = "InputRoomEvent"
|
|
InputDeviceListUpdate = "InputDeviceListUpdate"
|
|
InputSigningKeyUpdate = "InputSigningKeyUpdate"
|
|
OutputRoomEvent = "OutputRoomEvent"
|
|
OutputSendToDeviceEvent = "OutputSendToDeviceEvent"
|
|
OutputKeyChangeEvent = "OutputKeyChangeEvent"
|
|
OutputTypingEvent = "OutputTypingEvent"
|
|
OutputClientData = "OutputClientData"
|
|
OutputNotificationData = "OutputNotificationData"
|
|
OutputReceiptEvent = "OutputReceiptEvent"
|
|
OutputStreamEvent = "OutputStreamEvent"
|
|
OutputReadUpdate = "OutputReadUpdate"
|
|
RequestPresence = "GetPresence"
|
|
OutputPresenceEvent = "OutputPresenceEvent"
|
|
InputFulltextReindex = "InputFulltextReindex"
|
|
)
|
|
|
|
var safeCharacters = regexp.MustCompile("[^A-Za-z0-9$]+")
|
|
|
|
func Tokenise(str string) string {
|
|
return safeCharacters.ReplaceAllString(str, "_")
|
|
}
|
|
|
|
func InputRoomEventSubj(roomID string) string {
|
|
return fmt.Sprintf("%s.%s", InputRoomEvent, Tokenise(roomID))
|
|
}
|
|
|
|
var streams = []*nats.StreamConfig{
|
|
{
|
|
Name: InputRoomEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: InputDeviceListUpdate,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: InputSigningKeyUpdate,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputRoomEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputSendToDeviceEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputKeyChangeEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputTypingEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.MemoryStorage,
|
|
MaxAge: time.Second * 60,
|
|
},
|
|
{
|
|
Name: OutputClientData,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputReceiptEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputNotificationData,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputStreamEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputReadUpdate,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.FileStorage,
|
|
},
|
|
{
|
|
Name: OutputPresenceEvent,
|
|
Retention: nats.InterestPolicy,
|
|
Storage: nats.MemoryStorage,
|
|
MaxAge: time.Minute * 5,
|
|
},
|
|
}
|