chan

2022-07-27 19:04:26 阅读：151 来源： 互联网

标签：time chan atomic func semaphore channel size

package main

import (
   "fmt"
   "math"
   "sync"
   "sync/atomic"
   "time"
)

type semaphore chan byte
func wait(s semaphore) bool {
   _, ok := <- s
   return ok
}

func wake(s semaphore)  {
   s <- 'k'
}

type channel struct {
   sync.Mutex
   rc, wc uint32
   r, w semaphore
   i, j uint32
   size uint32
   q  []interface{}
}

func NewChannel(size uint32) *channel {
   c := new(channel)
   c.r = make(semaphore, size)
   c.w = make(semaphore, size)
   size++
   c.size = size
   c.q = make([]interface{}, size)
   return c
}

func (c *channel) Write(s interface{})  {
   for {
      c.Lock()
      j := atomic.LoadUint32(&c.j)
      i := atomic.LoadUint32(&c.i)
      if (j + 1) % c.size == i {
         atomic.AddUint32(&c.wc, 1)
         c.Unlock()
         wait(c.w)
      } else {
         j++
         j %= c.size
         c.q[j] = s
         atomic.StoreUint32(&c.j, j)
         c.Unlock()
         if atomic.LoadUint32(&c.rc) > 0 {
            atomic.AddUint32(&c.rc, math.MaxUint32)
            wake(c.r)
         }
         return
      }
   }
}

func (c *channel) Read() interface{} {
   for {
      c.Lock()
      j := atomic.LoadUint32(&c.j)
      i := atomic.LoadUint32(&c.i)
      if j == i {
         atomic.AddUint32(&c.rc, 1)
         c.Unlock()
         wait(c.r)
      } else {
         i++
         i %= c.size
         r := c.q[i]
         atomic.StoreUint32(&c.i, i)
         c.Unlock()
         if atomic.LoadUint32(&c.wc) > 0 {
            atomic.AddUint32(&c.wc, math.MaxUint32)
            wake(c.w)
         }
         return r
      }
   }
}

func (c *channel) Close() error {
   return nil
}


func main() {
   c := NewChannel(1)
   go func() {
      for i := 0; i < 700; i++ {
         //if i % 100 == 0 {
         // time.Sleep(time.Second)
         //}
         c.Write(i)
      }
      time.Sleep(time.Second*4)
      fmt.Println("end")
   }()
   for i := 0; i < 80; i++ {
      go func() {
         for
         {
            //time.Sleep(time.Minute)
            i := c.Read().(int)
            fmt.Println(i)
         }
      }()
   }
   time.Sleep(time.Minute*10)
}

标签：time,chan,atomic,func,semaphore,channel,size
来源： https://www.cnblogs.com/Janly/p/16525931.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

chan