Skip to content

Commit

Permalink
Updated WS API. Fixes #219. Closes #220
Browse files Browse the repository at this point in the history
  • Loading branch information
obscuren committed Jan 6, 2015
1 parent 47e6b2c commit a26aecd
Show file tree
Hide file tree
Showing 4 changed files with 38 additions and 42 deletions.
70 changes: 35 additions & 35 deletions cmd/utils/websockets.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,73 +33,73 @@ func (self *WebSocketServer) Serv() {
data := ethutil.NewValue(msg.Args)
bcode, err := ethutil.Compile(data.Get(0).Str(), false)
if err != nil {
c.Write(args(nil, err.Error()), msg.Seed)
c.Write(args(nil, err.Error()), msg.Id)
}

code := ethutil.Bytes2Hex(bcode)
c.Write(args(code, nil), msg.Seed)
case "getBlockByNumber":
c.Write(args(code, nil), msg.Id)
case "eth_blockByNumber":
args := msg.Arguments()

block := pipe.BlockByNumber(int32(args.Get(0).Uint()))
c.Write(block, msg.Seed)
c.Write(block, msg.Id)

case "getKey":
c.Write(pipe.Key().PrivateKey, msg.Seed)
case "transact":
case "eth_blockByHash":
args := msg.Arguments()

c.Write(pipe.BlockByHash(args.Get(0).Str()), msg.Id)

case "eth_transact":
if mp, ok := msg.Args[0].(map[string]interface{}); ok {
object := mapToTxParams(mp)
c.Write(
args(pipe.Transact(object["from"], object["to"], object["value"], object["gas"], object["gasPrice"], object["data"])),
msg.Seed,
args(pipe.Transact(pipe.Key().PrivateKey, object["to"], object["value"], object["gas"], object["gasPrice"], object["data"])),
msg.Id,
)

}
case "getCoinBase":
c.Write(pipe.CoinBase(), msg.Seed)
case "eth_gasPrice":
c.Write("10000000000000", msg.Id)
case "eth_coinbase":
c.Write(pipe.CoinBase(), msg.Id)

case "getIsListening":
c.Write(pipe.IsListening(), msg.Seed)
case "eth_listening":
c.Write(pipe.IsListening(), msg.Id)

case "getIsMining":
c.Write(pipe.IsMining(), msg.Seed)

case "getPeerCoint":
c.Write(pipe.PeerCount(), msg.Seed)

case "getCountAt":
args := msg.Arguments()
case "eth_mining":
c.Write(pipe.IsMining(), msg.Id)

c.Write(pipe.TxCountAt(args.Get(0).Str()), msg.Seed)
case "eth_peerCount":
c.Write(pipe.PeerCount(), msg.Id)

case "getCodeAt":
case "eth_countAt":
args := msg.Arguments()

c.Write(len(pipe.CodeAt(args.Get(0).Str())), msg.Seed)
c.Write(pipe.TxCountAt(args.Get(0).Str()), msg.Id)

case "getBlockByHash":
case "eth_codeAt":
args := msg.Arguments()

c.Write(pipe.BlockByHash(args.Get(0).Str()), msg.Seed)
c.Write(len(pipe.CodeAt(args.Get(0).Str())), msg.Id)

case "getStorageAt":
case "eth_storageAt":
args := msg.Arguments()

c.Write(pipe.StorageAt(args.Get(0).Str(), args.Get(1).Str()), msg.Seed)
c.Write(pipe.StorageAt(args.Get(0).Str(), args.Get(1).Str()), msg.Id)

case "getBalanceAt":
case "eth_balanceAt":
args := msg.Arguments()

c.Write(pipe.BalanceAt(args.Get(0).Str()), msg.Seed)
c.Write(pipe.BalanceAt(args.Get(0).Str()), msg.Id)

case "getSecretToAddress":
case "eth_secretToAddress":
args := msg.Arguments()

c.Write(pipe.SecretToAddress(args.Get(0).Str()), msg.Seed)
c.Write(pipe.SecretToAddress(args.Get(0).Str()), msg.Id)

case "newFilter":
case "newFilterString":
case "messages":
case "eth_newFilter":
case "eth_newFilterString":
case "eth_messages":
// TODO
}

Expand Down
6 changes: 2 additions & 4 deletions websocket/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,8 @@ func (c *Client) Conn() *ws.Conn {
return c.ws
}

func (c *Client) Write(data interface{}, seed int) {
msg := &Message{Seed: seed, Data: data}
func (c *Client) Write(data interface{}, id int) {
msg := &Message{Id: id, Data: data}
select {
case c.ch <- msg:
default:
Expand All @@ -73,7 +73,6 @@ func (c *Client) Listen() {

// Listen write request via chanel
func (c *Client) listenWrite() {
wslogger.Debugln("Listening write to client")
for {
select {

Expand All @@ -93,7 +92,6 @@ func (c *Client) listenWrite() {

// Listen read request via chanel
func (c *Client) listenRead() {
wslogger.Debugln("Listening read from client")
for {
select {

Expand Down
2 changes: 1 addition & 1 deletion websocket/message.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ import "github.com/ethereum/go-ethereum/ethutil"
type Message struct {
Call string `json:"call"`
Args []interface{} `json:"args"`
Seed int `json:"seed"`
Id int `json:"_id"`
Data interface{} `json:"data"`
}

Expand Down
2 changes: 0 additions & 2 deletions websocket/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -81,8 +81,6 @@ func (s *Server) MessageFunc(f MsgFunc) {
// Listen and serve.
// It serves client connection and broadcast request.
func (s *Server) Listen() {
wslogger.Debugln("Listening server...")

// ws handler
onConnected := func(ws *ws.Conn) {
defer func() {
Expand Down

0 comments on commit a26aecd

Please sign in to comment.