cleaning up now that i actually know this language
This commit is contained in:
101
cmd/hehand/main.go
Normal file
101
cmd/hehand/main.go
Normal file
@@ -0,0 +1,101 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"github.com/sneak/poker"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/rpc"
|
||||
"net/rpc/jsonrpc"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
type JSONRPCServer struct {
|
||||
*rpc.Server
|
||||
}
|
||||
|
||||
func NewJSONRPCServer() *JSONRPCServer {
|
||||
return &JSONRPCServer{rpc.NewServer()}
|
||||
}
|
||||
|
||||
func (s *JSONRPCServer) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
||||
log.Println("rpc server got a request")
|
||||
conn, _, err := w.(http.Hijacker).Hijack()
|
||||
if err != nil {
|
||||
log.Print("rpc hijacking ", req.RemoteAddr, ": ", err.Error())
|
||||
return
|
||||
}
|
||||
io.WriteString(conn, "HTTP/1.0 200 Connected to Go JSON-RPC\n\n")
|
||||
codec := jsonrpc.NewServerCodec(conn)
|
||||
log.Println("ServeCodec")
|
||||
s.Server.ServeCodec(codec)
|
||||
log.Println("finished serving request")
|
||||
}
|
||||
|
||||
type Args struct {
|
||||
A, B int
|
||||
}
|
||||
|
||||
type Quotient struct {
|
||||
Quo, Rem int
|
||||
}
|
||||
|
||||
type Arith int
|
||||
|
||||
func (t *Arith) Multiply(args *Args, reply *int) error {
|
||||
*reply = args.A * args.B
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Arith) Divide(args *Args, quo *Quotient) error {
|
||||
if args.B == 0 {
|
||||
return errors.New("divide by zero")
|
||||
}
|
||||
quo.Quo = args.A / args.B
|
||||
quo.Rem = args.A % args.B
|
||||
return nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
//log.SetFormatter(&log.JSONFormatter{})
|
||||
|
||||
// Output to stdout instead of the default stderr
|
||||
// Can be any io.Writer, see below for File example
|
||||
log.SetOutput(os.Stdout)
|
||||
|
||||
// Only log the warning severity or above.
|
||||
//log.SetLevel(log.WarnLevel)
|
||||
|
||||
log.Infof("starting up")
|
||||
|
||||
go runHttpServer()
|
||||
|
||||
running := true
|
||||
|
||||
for running {
|
||||
time.Sleep(1 * time.Second)
|
||||
}
|
||||
}
|
||||
|
||||
func runHttpServer() {
|
||||
js := NewJSONRPCServer()
|
||||
arith := new(Arith)
|
||||
js.Register(arith)
|
||||
|
||||
port := 8080
|
||||
|
||||
listenaddr := fmt.Sprintf("0.0.0.0:%d", port)
|
||||
|
||||
s := &http.Server{
|
||||
Addr: listenaddr,
|
||||
Handler: js,
|
||||
ReadTimeout: 10 * time.Second,
|
||||
WriteTimeout: 10 * time.Second,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
}
|
||||
log.Infof("starting up http server %s", listenaddr)
|
||||
log.Fatal(s.ListenAndServe())
|
||||
}
|
||||
Reference in New Issue
Block a user