]> Cypherpunks.ru repositories - nncp.git/blob - src/cypherpunks.ru/nncp/call.go
Simple rate limiter
[nncp.git] / src / cypherpunks.ru / nncp / call.go
1 /*
2 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
3 Copyright (C) 2016-2018 Sergey Matveev <stargrave@stargrave.org>
4
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
9
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 */
18
19 package nncp
20
21 import (
22         "net"
23         "strconv"
24
25         "github.com/gorhill/cronexpr"
26 )
27
28 type Call struct {
29         Cron           *cronexpr.Expression
30         Nice           uint8
31         Xx             TRxTx
32         RxRate         int
33         TxRate         int
34         Addr           *string
35         OnlineDeadline uint
36         MaxOnlineTime  uint
37 }
38
39 func (ctx *Ctx) CallNode(node *Node, addrs []string, nice uint8, xxOnly TRxTx, rxRate, txRate int, onlineDeadline, maxOnlineTime uint) (isGood bool) {
40         for _, addr := range addrs {
41                 sds := SDS{"node": node.Id, "addr": addr}
42                 ctx.LogD("call", sds, "dialing")
43                 conn, err := net.Dial("tcp", addr)
44                 if err != nil {
45                         ctx.LogD("call", SdsAdd(sds, SDS{"err": err}), "dialing")
46                         continue
47                 }
48                 ctx.LogD("call", sds, "connected")
49                 state, err := ctx.StartI(
50                         conn,
51                         node.Id,
52                         nice,
53                         xxOnly,
54                         rxRate,
55                         txRate,
56                         onlineDeadline,
57                         maxOnlineTime,
58                 )
59                 if err == nil {
60                         ctx.LogI("call-start", sds, "connected")
61                         state.Wait()
62                         ctx.LogI("call-finish", SDS{
63                                 "node":     state.Node.Id,
64                                 "duration": strconv.FormatInt(int64(state.Duration.Seconds()), 10),
65                                 "rxbytes":  strconv.FormatInt(state.RxBytes, 10),
66                                 "txbytes":  strconv.FormatInt(state.TxBytes, 10),
67                                 "rxspeed":  strconv.FormatInt(state.RxSpeed, 10),
68                                 "txspeed":  strconv.FormatInt(state.TxSpeed, 10),
69                         }, "")
70                         isGood = true
71                         conn.Close()
72                         break
73                 } else {
74                         ctx.LogE("call-start", SdsAdd(sds, SDS{"err": err}), "")
75                         conn.Close()
76                 }
77         }
78         return
79 }