python-langserver/langserver-python.go at v0 · sourcegraph/python-langserver

This repository was archived by the owner on Mar 27, 2019. It is now read-only.

Repository navigation

History

package main

import (

"context"

"flag"

"fmt"

"log"

"net"

"os"

"sourcegraph.com/sourcegraph/sourcegraph/pkg/debugserver"

"sourcegraph.com/sourcegraph/sourcegraph/pkg/jsonrpc2"

"github.com/sourcegraph/langserver-python/python"

)

var (

mode = flag.String("mode", "stdio", "communication mode (stdio|tcp)")

addr = flag.String("addr", ":2088", "server listen address (tcp)")

profbind = flag.String("prof-http", ":6060", "net/http/pprof http bind address")

logfile = flag.String("log", "/tmp/langserver-python.log", "write log output to this file (and stderr)")

)

func main() {

flag.Parse()

log.SetFlags(0)

if err := run(); err != nil {

fmt.Fprintln(os.Stderr, err)

os.Exit(1)

}

func run() error {

l := log.New(os.Stderr, "", 0)

if *logfile != "" {

f, err := os.Create(*logfile)

if err != nil {

return err

}

defer f.Close()

l.SetOutput(f)

log.SetOutput(f)

}

if *profbind != "" {

go debugserver.Start(*profbind)

}

h := jsonrpc2.HandlerWithError((&python.Handler{}).Handle)

switch *mode {

case "tcp":

lis, err := net.Listen("tcp", *addr)

if err != nil {

return err

}

defer lis.Close()

log.Println("listening on", *addr)

return jsonrpc2.Serve(context.Background(), lis, h)

case "stdio":

log.Println("reading on stdin, writing on stdout")

<-jsonrpc2.NewConn(context.Background(), stdrwc{}, h, jsonrpc2.LogMessages(l)).DisconnectNotify()

log.Println("connection closed")

return nil

default:

return fmt.Errorf("invalid mode %q", *mode)

}

type stdrwc struct{}

func (stdrwc) Read(p []byte) (int, error) {

return os.Stdin.Read(p)

}

func (stdrwc) Write(p []byte) (int, error) {

return os.Stdout.Write(p)

}

func (stdrwc) Close() error {

if err := os.Stdin.Close(); err != nil {

return err

}

return os.Stdout.Close()

}

FilesExpand file tree

langserver-python.go

Latest commit

History

langserver-python.go

File metadata and controls