You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
148 lines
3.4 KiB
148 lines
3.4 KiB
![]()
9 years ago
|
package rpc
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/json"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"net/http"
|
||
![]()
8 years ago
|
"net/url"
|
||
|
"sync"
|
||
|
"sync/atomic"
|
||
![]()
9 years ago
|
"time"
|
||
|
|
||
![]()
7 years ago
|
"github.com/sammy007/monero-stratum/pool"
|
||
![]()
9 years ago
|
)
|
||
|
|
||
|
type RPCClient struct {
|
||
![]()
8 years ago
|
sync.RWMutex
|
||
|
Url *url.URL
|
||
|
login string
|
||
|
password string
|
||
|
Name string
|
||
|
sick bool
|
||
|
sickRate int
|
||
|
successRate int
|
||
|
Accepts uint64
|
||
|
Rejects uint64
|
||
|
LastSubmissionAt int64
|
||
|
client *http.Client
|
||
|
FailsCount uint64
|
||
![]()
9 years ago
|
}
|
||
|
|
||
|
type GetBlockTemplateReply struct {
|
||
|
Blob string `json:"blocktemplate_blob"`
|
||
|
Difficulty int64 `json:"difficulty"`
|
||
|
ReservedOffset int `json:"reserved_offset"`
|
||
|
Height int64 `json:"height"`
|
||
|
PrevHash string `json:"prev_hash"`
|
||
|
}
|
||
|
|
||
|
type JSONRpcResp struct {
|
||
|
Id *json.RawMessage `json:"id"`
|
||
|
Result *json.RawMessage `json:"result"`
|
||
|
Error map[string]interface{} `json:"error"`
|
||
|
}
|
||
|
|
||
![]()
8 years ago
|
func NewRPCClient(cfg *pool.Upstream) (*RPCClient, error) {
|
||
|
rawUrl := fmt.Sprintf("http://%s:%v/json_rpc", cfg.Host, cfg.Port)
|
||
|
url, err := url.Parse(rawUrl)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
rpcClient := &RPCClient{Name: cfg.Name, Url: url}
|
||
|
timeout, _ := time.ParseDuration(cfg.Timeout)
|
||
![]()
9 years ago
|
rpcClient.client = &http.Client{
|
||
|
Timeout: timeout,
|
||
|
}
|
||
![]()
8 years ago
|
return rpcClient, nil
|
||
![]()
9 years ago
|
}
|
||
|
|
||
![]()
8 years ago
|
func (r *RPCClient) GetBlockTemplate(reserveSize int, address string) (*GetBlockTemplateReply, error) {
|
||
![]()
9 years ago
|
params := map[string]interface{}{"reserve_size": reserveSize, "wallet_address": address}
|
||
![]()
8 years ago
|
rpcResp, err := r.doPost(r.Url.String(), "getblocktemplate", params)
|
||
|
var reply *GetBlockTemplateReply
|
||
![]()
9 years ago
|
if err != nil {
|
||
![]()
8 years ago
|
return nil, err
|
||
![]()
9 years ago
|
}
|
||
![]()
8 years ago
|
if rpcResp.Result != nil {
|
||
|
err = json.Unmarshal(*rpcResp.Result, &reply)
|
||
![]()
9 years ago
|
}
|
||
|
return reply, err
|
||
|
}
|
||
|
|
||
![]()
8 years ago
|
func (r *RPCClient) SubmitBlock(hash string) (*JSONRpcResp, error) {
|
||
|
return r.doPost(r.Url.String(), "submitblock", []string{hash})
|
||
![]()
9 years ago
|
}
|
||
|
|
||
![]()
8 years ago
|
func (r *RPCClient) doPost(url, method string, params interface{}) (*JSONRpcResp, error) {
|
||
|
jsonReq := map[string]interface{}{"jsonrpc": "2.0", "id": 0, "method": method, "params": params}
|
||
![]()
9 years ago
|
data, _ := json.Marshal(jsonReq)
|
||
|
req, err := http.NewRequest("POST", url, bytes.NewBuffer(data))
|
||
|
req.Header.Set("Content-Length", (string)(len(data)))
|
||
|
req.Header.Set("Content-Type", "application/json")
|
||
|
req.Header.Set("Accept", "application/json")
|
||
![]()
8 years ago
|
req.SetBasicAuth(r.login, r.password)
|
||
![]()
9 years ago
|
resp, err := r.client.Do(req)
|
||
|
if err != nil {
|
||
![]()
8 years ago
|
r.markSick()
|
||
|
return nil, err
|
||
![]()
9 years ago
|
}
|
||
|
defer resp.Body.Close()
|
||
|
|
||
![]()
8 years ago
|
if resp.StatusCode < 200 || resp.StatusCode >= 400 {
|
||
|
return nil, errors.New(resp.Status)
|
||
|
}
|
||
|
|
||
|
var rpcResp *JSONRpcResp
|
||
|
err = json.NewDecoder(resp.Body).Decode(&rpcResp)
|
||
|
if err != nil {
|
||
|
r.markSick()
|
||
|
return nil, err
|
||
|
}
|
||
|
if rpcResp.Error != nil {
|
||
|
r.markSick()
|
||
|
return nil, errors.New(rpcResp.Error["message"].(string))
|
||
|
}
|
||
![]()
9 years ago
|
return rpcResp, err
|
||
|
}
|
||
![]()
8 years ago
|
|
||
|
func (r *RPCClient) Check(reserveSize int, address string) (bool, error) {
|
||
|
_, err := r.GetBlockTemplate(reserveSize, address)
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
r.markAlive()
|
||
|
return !r.Sick(), nil
|
||
|
}
|
||
|
|
||
|
func (r *RPCClient) Sick() bool {
|
||
|
r.RLock()
|
||
|
defer r.RUnlock()
|
||
|
return r.sick
|
||
|
}
|
||
|
|
||
|
func (r *RPCClient) markSick() {
|
||
|
r.Lock()
|
||
|
if !r.sick {
|
||
|
atomic.AddUint64(&r.FailsCount, 1)
|
||
|
}
|
||
|
r.sickRate++
|
||
|
r.successRate = 0
|
||
|
if r.sickRate >= 5 {
|
||
|
r.sick = true
|
||
|
}
|
||
|
r.Unlock()
|
||
|
}
|
||
|
|
||
|
func (r *RPCClient) markAlive() {
|
||
|
r.Lock()
|
||
|
r.successRate++
|
||
|
if r.successRate >= 5 {
|
||
|
r.sick = false
|
||
|
r.sickRate = 0
|
||
|
r.successRate = 0
|
||
|
}
|
||
|
r.Unlock()
|
||
|
}
|