Exemplo n.º 1
0
// Process function
func (s *Server) Process(str *string) (string, error) {
	buffer := new(bytes.Buffer)
	_, err := buffer.WriteString(*str)
	if err != nil {
		s.Err(err)
		return "", err
	}

	t1 := &thrift.TMemoryBuffer{Buffer: buffer}
	in := thrift.NewTJSONProtocol(t1)
	t2 := thrift.NewTMemoryBufferLen(1024)
	//defer t2.Close()
	out := thrift.NewTJSONProtocol(t2)

	s.processor.Process(in, out)
	return t2.String(), nil
}
Exemplo n.º 2
0
// NewThriftHandler retuns http.Handler
func NewThriftHandler(processor thrift.TProcessor) func(http.ResponseWriter, *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.Method != `POST` {
			http.Error(w, "Thrift Handler requires POST access", http.StatusInternalServerError)
			return
		}

		buffer := new(bytes.Buffer)
		_, err := buffer.ReadFrom(r.Body)
		if err != nil {
			http.Error(w, "Could not read body", http.StatusInternalServerError)
			return
		}

		inTransport := &thrift.TMemoryBuffer{Buffer: buffer}
		inProtocol := thrift.NewTJSONProtocol(inTransport)
		outTransport := thrift.NewTMemoryBufferLen(1024)
		outProtocol := thrift.NewTJSONProtocol(outTransport)

		processor.Process(inProtocol, outProtocol)

		fmt.Fprint(w, outTransport)
	}
}
Exemplo n.º 3
0
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"github.com/huodon/wpc/thrift/impl"
	"github.com/huodon/wpc/thrift/service"
)

var upgrade = websocket.Upgrader{
	HandshakeTimeout: 15 * time.Second,
	Error: func(w http.ResponseWriter, r *http.Request, status int, reason error) {
		log.Printf("some error happend: %v, state: %d, reason: %s", r.Host, status, reason)
	},
}

var (
	trans   = thrift.NewTMemoryBufferLen(bytes.MinRead << 4)
	proto   = thrift.NewTJSONProtocol(trans)
	process = thrift.NewTMultiplexedProcessor()
	boxProc = service.NewBoxServiceProcessor(impl.GetServiceImpl())
	inChan  = make(chan []byte)
)

func init() {
	process.RegisterDefault(boxProc)
}

func wsHandle(w http.ResponseWriter, r *http.Request) {
	log.Println(`incomeing websocket request`)
	c, err := upgrade.Upgrade(w, r, nil)
	defer c.Close()

	go ThriftProcess(c)