File size: 1,598 Bytes
7107f0b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
package message
import (
"time"
"github.com/alist-org/alist/v3/server/common"
"github.com/gin-gonic/gin"
"github.com/pkg/errors"
)
type Http struct {
Received chan string // received messages from web
ToSend chan Message // messages to send to web
}
type Req struct {
Message string `json:"message" form:"message"`
}
func (p *Http) GetHandle(c *gin.Context) {
select {
case message := <-p.ToSend:
common.SuccessResp(c, message)
default:
common.ErrorStrResp(c, "no message", 404)
}
}
func (p *Http) SendHandle(c *gin.Context) {
var req Req
if err := c.ShouldBind(&req); err != nil {
common.ErrorResp(c, err, 400)
return
}
select {
case p.Received <- req.Message:
common.SuccessResp(c)
default:
common.ErrorStrResp(c, "nowhere needed", 500)
}
}
func (p *Http) Send(message Message) error {
select {
case p.ToSend <- message:
return nil
default:
return errors.New("send failed")
}
}
func (p *Http) Receive() (string, error) {
select {
case message := <-p.Received:
return message, nil
default:
return "", errors.New("receive failed")
}
}
func (p *Http) WaitSend(message Message, d int) error {
select {
case p.ToSend <- message:
return nil
case <-time.After(time.Duration(d) * time.Second):
return errors.New("send timeout")
}
}
func (p *Http) WaitReceive(d int) (string, error) {
select {
case message := <-p.Received:
return message, nil
case <-time.After(time.Duration(d) * time.Second):
return "", errors.New("receive timeout")
}
}
var HttpInstance = &Http{
Received: make(chan string),
ToSend: make(chan Message),
}
|