Commit a0ae7fac authored by Lantao Liu's avatar Lantao Liu

Implement stop function in streaming server.

parent a235ba4e
...@@ -141,6 +141,11 @@ func NewServer(config Config, runtime Runtime) (Server, error) { ...@@ -141,6 +141,11 @@ func NewServer(config Config, runtime Runtime) (Server, error) {
handler := restful.NewContainer() handler := restful.NewContainer()
handler.Add(ws) handler.Add(ws)
s.handler = handler s.handler = handler
s.server = &http.Server{
Addr: s.config.Addr,
Handler: s.handler,
TLSConfig: s.config.TLSConfig,
}
return s, nil return s, nil
} }
...@@ -150,6 +155,7 @@ type server struct { ...@@ -150,6 +155,7 @@ type server struct {
runtime *criAdapter runtime *criAdapter
handler http.Handler handler http.Handler
cache *requestCache cache *requestCache
server *http.Server
} }
func (s *server) GetExec(req *runtimeapi.ExecRequest) (*runtimeapi.ExecResponse, error) { func (s *server) GetExec(req *runtimeapi.ExecRequest) (*runtimeapi.ExecResponse, error) {
...@@ -197,21 +203,15 @@ func (s *server) Start(stayUp bool) error { ...@@ -197,21 +203,15 @@ func (s *server) Start(stayUp bool) error {
return errors.New("stayUp=false is not yet implemented") return errors.New("stayUp=false is not yet implemented")
} }
server := &http.Server{
Addr: s.config.Addr,
Handler: s.handler,
TLSConfig: s.config.TLSConfig,
}
if s.config.TLSConfig != nil { if s.config.TLSConfig != nil {
return server.ListenAndServeTLS("", "") // Use certs from TLSConfig. return s.server.ListenAndServeTLS("", "") // Use certs from TLSConfig.
} else { } else {
return server.ListenAndServe() return s.server.ListenAndServe()
} }
} }
func (s *server) Stop() error { func (s *server) Stop() error {
// TODO(tallclair): Implement this. return s.server.Close()
return errors.New("not yet implemented")
} }
func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment