mleku
fa71e9e334
- Introduced a new skill for Golang, providing comprehensive guidance on writing, debugging, and best practices for Go programming. - Added reference materials including effective Go guidelines, common patterns, and a quick reference cheat sheet to support users in Go development. - Created a skill creator guide to assist in developing new skills with structured templates and resource management. - Implemented scripts for skill initialization and packaging to streamline the skill creation process.
69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package publish
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"next.orly.dev/pkg/encoders/event"
|
|
"next.orly.dev/pkg/interfaces/publisher"
|
|
"next.orly.dev/pkg/interfaces/typer"
|
|
)
|
|
|
|
// WriteRequest represents a write operation to be performed by the write worker
|
|
type WriteRequest struct {
|
|
Data []byte
|
|
MsgType int
|
|
IsControl bool
|
|
Deadline time.Time
|
|
IsPing bool // Special marker for ping messages
|
|
}
|
|
|
|
// WriteChanSetter defines the interface for setting write channels
|
|
type WriteChanSetter interface {
|
|
SetWriteChan(*websocket.Conn, chan WriteRequest)
|
|
GetWriteChan(*websocket.Conn) (chan WriteRequest, bool)
|
|
}
|
|
|
|
// S is the control structure for the subscription management scheme.
|
|
type S struct {
|
|
publisher.Publishers
|
|
}
|
|
|
|
// New creates a new publish.S.
|
|
func New(p ...publisher.I) (s *S) {
|
|
s = &S{Publishers: p}
|
|
return
|
|
}
|
|
|
|
var _ publisher.I = &S{}
|
|
|
|
func (s *S) Type() string { return "publish" }
|
|
|
|
func (s *S) Deliver(ev *event.E) {
|
|
for _, p := range s.Publishers {
|
|
p.Deliver(ev)
|
|
}
|
|
}
|
|
|
|
func (s *S) Receive(msg typer.T) {
|
|
t := msg.Type()
|
|
for _, p := range s.Publishers {
|
|
if p.Type() == t {
|
|
p.Receive(msg)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// GetSocketPublisher returns the socketapi publisher instance
|
|
func (s *S) GetSocketPublisher() WriteChanSetter {
|
|
for _, p := range s.Publishers {
|
|
if p.Type() == "socketapi" {
|
|
if socketPub, ok := p.(WriteChanSetter); ok {
|
|
return socketPub
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|