update
This commit is contained in:
parent
f224c77bac
commit
7823fb86ab
63
module/twitch-irc/message-queue.go
Normal file
63
module/twitch-irc/message-queue.go
Normal file
@ -0,0 +1,63 @@
|
||||
package twitchirc
|
||||
|
||||
import (
|
||||
"sync"
|
||||
)
|
||||
|
||||
// MsgObj -
|
||||
type MsgObj struct {
|
||||
Command string
|
||||
Params []string
|
||||
}
|
||||
|
||||
// QueueList -
|
||||
type QueueList struct {
|
||||
Messages []*MsgObj
|
||||
Lock *sync.Mutex
|
||||
}
|
||||
|
||||
// NewQueue -
|
||||
func NewQueue() *QueueList {
|
||||
return &QueueList{}
|
||||
}
|
||||
|
||||
// Add - add element
|
||||
func (q *QueueList) Add(m *MsgObj) {
|
||||
q.Messages = append(q.Messages, m)
|
||||
}
|
||||
|
||||
// Get - get element
|
||||
func (q *QueueList) Get() (m *MsgObj) {
|
||||
if q.IsEmpty() {
|
||||
return nil
|
||||
}
|
||||
m = q.Messages[0]
|
||||
q.Messages = q.Messages[1:]
|
||||
return
|
||||
}
|
||||
|
||||
// IsEmpty -
|
||||
func (q *QueueList) IsEmpty() bool {
|
||||
if len(q.Messages) == 0 {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// Size -
|
||||
func (q *QueueList) Size() int {
|
||||
return len(q.Messages)
|
||||
}
|
||||
|
||||
// Clear -
|
||||
func (q *QueueList) Clear() {
|
||||
if q.IsEmpty() {
|
||||
return
|
||||
}
|
||||
|
||||
for i := 0; i < len(q.Messages); i++ {
|
||||
q.Messages[i] = nil
|
||||
}
|
||||
q.Messages = nil
|
||||
return
|
||||
}
|
@ -3,6 +3,7 @@ package twitchirc
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"time"
|
||||
|
||||
"gopkg.in/irc.v2"
|
||||
|
||||
@ -10,6 +11,8 @@ import (
|
||||
)
|
||||
|
||||
var client *irc.Client
|
||||
var queue *QueueList
|
||||
var channels []string
|
||||
|
||||
// InitIRC -
|
||||
func InitIRC() (err error) {
|
||||
@ -25,9 +28,91 @@ func InitIRC() (err error) {
|
||||
client = irc.NewClient(conn, config)
|
||||
|
||||
err = client.Run()
|
||||
|
||||
queue = NewQueue()
|
||||
go runQueue()
|
||||
|
||||
channels = make([]string, 0)
|
||||
return
|
||||
}
|
||||
|
||||
func ircHandle(c *irc.Client, m *irc.Message) {
|
||||
fmt.Println(m.String())
|
||||
// JoinChannel -
|
||||
func JoinChannel(ch string) {
|
||||
if len(ch) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
if indexOf(channels, ch) != -1 {
|
||||
return
|
||||
}
|
||||
|
||||
m := &MsgObj{
|
||||
Command: "JOIN",
|
||||
Params: []string{
|
||||
fmt.Sprintf("#%s", ch),
|
||||
},
|
||||
}
|
||||
queue.Add(m)
|
||||
|
||||
// msg := &irc.Message{}
|
||||
// msg.Command = "JOIN"
|
||||
// msg.Params = []string{
|
||||
// fmt.Sprintf("#%s", ch),
|
||||
// }
|
||||
// client.WriteMessage(msg)
|
||||
}
|
||||
|
||||
// LeaveChannel -
|
||||
func LeaveChannel(ch string) {
|
||||
if len(ch) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
if indexOf(channels, ch) == -1 {
|
||||
return
|
||||
}
|
||||
|
||||
m := &MsgObj{
|
||||
Command: "PART",
|
||||
Params: []string{
|
||||
fmt.Sprintf("#%s", ch),
|
||||
},
|
||||
}
|
||||
queue.Add(m)
|
||||
}
|
||||
|
||||
func runQueue() {
|
||||
for {
|
||||
if !queue.IsEmpty() {
|
||||
m := queue.Get()
|
||||
msg := &irc.Message{}
|
||||
msg.Command = m.Command
|
||||
msg.Params = m.Params
|
||||
err := client.WriteMessage(msg)
|
||||
if err == nil {
|
||||
if m.Command == "JOIN" {
|
||||
|
||||
} else if m.Command == "PART" {
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
time.Sleep(time.Microsecond * 1500)
|
||||
}
|
||||
}
|
||||
|
||||
func ircHandle(c *irc.Client, m *irc.Message) {
|
||||
|
||||
}
|
||||
|
||||
func indexOf(c []string, data string) int {
|
||||
if len(c) == 0 {
|
||||
return -1
|
||||
}
|
||||
for k, v := range c {
|
||||
if v == data {
|
||||
return k
|
||||
}
|
||||
}
|
||||
return -1
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user