123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- package internal
- import (
- "log"
- "net"
- "runtime"
- "sync"
- "time"
- )
- var limitSem = make(chan int, 100)
- func limitRelease() {
-
- select {
- case <-limitSem:
- default:
-
- log.Print("appengine: unbalanced limitSem release!")
- }
- }
- func limitDial(network, addr string) (net.Conn, error) {
- limitSem <- 1
-
-
- conn, err := net.DialTimeout(network, addr, 500*time.Millisecond)
- if err != nil {
- limitRelease()
- return nil, err
- }
- lc := &limitConn{Conn: conn}
- runtime.SetFinalizer(lc, (*limitConn).Close)
- return lc, nil
- }
- type limitConn struct {
- close sync.Once
- net.Conn
- }
- func (lc *limitConn) Close() error {
- defer lc.close.Do(func() {
- limitRelease()
- runtime.SetFinalizer(lc, nil)
- })
- return lc.Conn.Close()
- }
|