Browse Source

修改channel 数据类型

feature_crossplant
zhangxin 2 years ago
parent
commit
6ac7c1f370
1 changed files with 6 additions and 7 deletions
  1. +6
    -7
      models/channel/ReleaseTask.go

+ 6
- 7
models/channel/ReleaseTask.go View File

@ -1,20 +1,19 @@
package channel
import (
plnModel "LAPP_ACURA_MOM_BACKEND/models/pln"
"errors"
"sync"
)
type ReleaseTaskChannel struct {
c chan plnModel.CustOrder
c chan interface{}
closed bool
mutex sync.Mutex
}
func NewReleaseChannel() *ReleaseTaskChannel {
return &ReleaseTaskChannel{
c: make(chan plnModel.CustOrder, 100),
c: make(chan interface{}, 100),
}
}
@ -26,13 +25,13 @@ func (c *ReleaseTaskChannel) IsClosed() bool {
}
// SendData 发送数据
func (c *ReleaseTaskChannel) SendData(custOrder plnModel.CustOrder) error {
func (c *ReleaseTaskChannel) SendData(data interface{}) error {
c.mutex.Lock()
defer c.mutex.Unlock()
if c.closed {
return errors.New("管道已关闭")
}
c.c <- custOrder
c.c <- data
return nil
}
@ -46,7 +45,7 @@ func (c *ReleaseTaskChannel) Close() {
c.closed = true
}
func (c *ReleaseTaskChannel) Read() (data plnModel.CustOrder, ok bool, err error) {
func (c *ReleaseTaskChannel) Read() (data interface{}, ok bool, err error) {
c.mutex.Lock()
if c.closed {
err = errors.New("管道已关闭")
@ -65,6 +64,6 @@ func (c *ReleaseTaskChannel) ResetChannel() (err error) {
return
}
c.closed = false
c.c = make(chan plnModel.CustOrder, 50)
c.c = make(chan interface{}, 50)
return
}

Loading…
Cancel
Save