mirror of
https://github.com/nbd-wtf/go-nostr.git
synced 2025-03-17 21:32:56 +01:00
06a15fdaab
- make BatchedSubManyEose() use a single duplicate id index and use it for replaceable loaders; - fixes parsing follow entry from kind:3 events (and others); - adds a "cause" to most cancelation errors in relay/pool; - remove the inherent cache from dataloader (we have our own hopefully); - increase max frame size we can read from any websocket to 2**18 (262k), which gives over 2000 item lists.
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package nostr
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
ws "github.com/coder/websocket"
|
|
)
|
|
|
|
type Connection struct {
|
|
conn *ws.Conn
|
|
}
|
|
|
|
func NewConnection(ctx context.Context, url string, requestHeader http.Header, tlsConfig *tls.Config) (*Connection, error) {
|
|
c, _, err := ws.Dial(ctx, url, getConnectionOptions(requestHeader, tlsConfig))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
c.SetReadLimit(262144) // this should be enough for contact lists of over 2000 people
|
|
|
|
return &Connection{
|
|
conn: c,
|
|
}, nil
|
|
}
|
|
|
|
func (c *Connection) WriteMessage(ctx context.Context, data []byte) error {
|
|
if err := c.conn.Write(ctx, ws.MessageText, data); err != nil {
|
|
return fmt.Errorf("failed to write message: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Connection) ReadMessage(ctx context.Context, buf io.Writer) error {
|
|
_, reader, err := c.conn.Reader(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get reader: %w", err)
|
|
}
|
|
if _, err := io.Copy(buf, reader); err != nil {
|
|
return fmt.Errorf("failed to read message: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Connection) Close() error {
|
|
return c.conn.Close(ws.StatusNormalClosure, "")
|
|
}
|
|
|
|
func (c *Connection) Ping(ctx context.Context) error {
|
|
return c.conn.Ping(ctx)
|
|
}
|