GeezerVisonThe World Wide Web, restored to 800 pixels of glory. |
| Home|About|Gaming cabinet|Archives|How it works|Starter sites |
GeezerVison is showing github.com — kite/server.go at master · koding/kite Expand file tree/ Copy pathserver.goMore file actions 220 lines (179 loc) · 4.53 KB / Copy pathserver.goFile metadata and controls220 lines (179 loc) · 4.53 KB 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 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 // Package server implements a HTTP(S) server for kites. package kite import ( "crypto/tls" "fmt" "io/ioutil" "net" "net/http" "os" "strconv" "strings" "sync" ) // Run is a blocking method. It runs the kite server and then accepts requests // asynchronously. It supports graceful restart via SIGUSR2. func (k *Kite) Run() { if os.Getenv("KITE_VERSION") != "" { fmt.Println(k.Kite().Version) os.Exit(0) } // An error string equivalent to net.errClosing for using with http.Serve() // during a graceful exit. Needed to declare here again because it is not // exported by "net" package. const errClosing = "use of closed network connection" err := k.listenAndServe() if err != nil { if strings.Contains(err.Error(), errClosing) { // The server is closed by Close() method k.Log.Info("Kite server is closed.") return } k.Log.Fatal(err.Error()) } } // Close stops the server and the kontrol client instance. func (k *Kite) Close() { k.Log.Info("Closing kite...") k.kontrol.Lock() if k.kontrol != nil && k.kontrol.Client != nil { k.kontrol.Close() } k.kontrol.Unlock() if k.listener != nil { k.listener.Close() k.listener = nil } k.mu.Lock() cache := k.verifyCache k.mu.Unlock() if cache != nil { cache.StopGC() } } func (k *Kite) Addr() string { return net.JoinHostPort(k.Config.IP, strconv.Itoa(k.Config.Port)) } // listenAndServe listens on the TCP network address k.URL.Host and then // calls Serve to handle requests on incoming connectionk. func (k *Kite) listenAndServe() error { // create a new one if there doesn't exist l, err := net.Listen("tcp4", k.Addr()) if err != nil { return err } k.Log.Info("New listening: %s", l.Addr()) if k.TLSConfig != nil { if k.TLSConfig.NextProtos == nil { k.TLSConfig.NextProtos = []string{"http/1.1"} } l = tls.NewListener(l, k.TLSConfig) } k.listener = newGracefulListener(l) // listener is ready, notify waiters. close(k.readyC) defer close(k.closeC) // serving is finished, notify waiters. k.Log.Info("Serving...") return k.serve(k.listener, k) } func (k *Kite) serve(l net.Listener, h http.Handler) error { if k.Config.Serve != nil { return k.Config.Serve(l, h) } return http.Serve(l, h) } // Port returns the TCP port number that the kite listens. // Port must be called after the listener is initialized. // You can use ServerReadyNotify function to get notified when listener is ready. // // Kite starts to listen the port when Run() is called. // Since Run() is blocking you need to run it as a goroutine the call this function when listener is ready. // // Example: // // k := kite.New("x", "1.0.0") // go k.Run() // <-k.ServerReadyNotify() // port := k.Port() // func (k *Kite) Port() int { if k.listener == nil { return 0 } return k.listener.Addr().(*net.TCPAddr).Port } func (k *Kite) UseTLS(certPEM, keyPEM string) { if k.TLSConfig == nil { k.TLSConfig = &tls.Config{} } cert, err := tls.X509KeyPair([]byte(certPEM), []byte(keyPEM)) if err != nil { panic(err) } k.TLSConfig.Certificates = append(k.TLSConfig.Certificates, cert) } func (k *Kite) UseTLSFile(certFile, keyFile string) { certData, err := ioutil.ReadFile(certFile) if err != nil { k.Log.Fatal("Cannot read certificate file: %s", err.Error()) } keyData, err := ioutil.ReadFile(keyFile) if err != nil { k.Log.Fatal("Cannot read certificate file: %s", err.Error()) } k.UseTLS(string(certData), string(keyData)) } func (k *Kite) ServerCloseNotify() chan bool { return k.closeC } func (k *Kite) ServerReadyNotify() chan bool { return k.readyC } // gracefulListener closes all accepted connections upon Close to ensure // no dangling websocket/xhr sessions outlive the kite. type gracefulListener struct { net.Listener conns map[net.Conn]struct{} connsMu sync.Mutex } func newGracefulListener(l net.Listener) *gracefulListener { return &gracefulListener{ Listener: l, conns: make(map[net.Conn]struct{}), } } func (l *gracefulListener) Accept() (net.Conn, error) { conn, err := l.Listener.Accept() if err != nil { return nil, err } l.connsMu.Lock() l.conns[conn] = struct{}{} l.connsMu.Unlock() return &gracefulConn{ Conn: conn, close: func() { l.connsMu.Lock() delete(l.conns, conn) l.connsMu.Unlock() }, }, nil } func (l *gracefulListener) Close() error { err := l.Listener.Close() l.connsMu.Lock() for conn := range l.conns { conn.Close() } l.conns = nil l.connsMu.Unlock() return err } type gracefulConn struct { net.Conn close func() } func (c *gracefulConn) Close() error { c.close() return c.Conn.Close() } |
| You are visitor 01402520 — GeezerVison does not store the pages you look up. Best viewed with Netscape Navigator 4.7 or Internet Explorer 6 at 800×600. No cookies required. |