aboutsummaryrefslogtreecommitdiffstats
path: root/accounts/keystore/watch.go
blob: 0b44012554560e4af35abddf8fd67742f1f4b023 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright 2016 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

// +build darwin,!ios freebsd linux,!arm64 netbsd solaris

package keystore

import (
    "time"

    "github.com/ethereum/go-ethereum/logger"
    "github.com/ethereum/go-ethereum/logger/glog"
    "github.com/rjeczalik/notify"
)

type watcher struct {
    ac       *accountCache
    starting bool
    running  bool
    ev       chan notify.EventInfo
    quit     chan struct{}
}

func newWatcher(ac *accountCache) *watcher {
    return &watcher{
        ac:   ac,
        ev:   make(chan notify.EventInfo, 10),
        quit: make(chan struct{}),
    }
}

// starts the watcher loop in the background.
// Start a watcher in the background if that's not already in progress.
// The caller must hold w.ac.mu.
func (w *watcher) start() {
    if w.starting || w.running {
        return
    }
    w.starting = true
    go w.loop()
}

func (w *watcher) close() {
    close(w.quit)
}

func (w *watcher) loop() {
    defer func() {
        w.ac.mu.Lock()
        w.running = false
        w.starting = false
        w.ac.mu.Unlock()
    }()

    err := notify.Watch(w.ac.keydir, w.ev, notify.All)
    if err != nil {
        glog.V(logger.Detail).Infof("can't watch %s: %v", w.ac.keydir, err)
        return
    }
    defer notify.Stop(w.ev)
    glog.V(logger.Detail).Infof("now watching %s", w.ac.keydir)
    defer glog.V(logger.Detail).Infof("no longer watching %s", w.ac.keydir)

    w.ac.mu.Lock()
    w.running = true
    w.ac.mu.Unlock()

    // Wait for file system events and reload.
    // When an event occurs, the reload call is delayed a bit so that
    // multiple events arriving quickly only cause a single reload.
    var (
        debounce          = time.NewTimer(0)
        debounceDuration  = 500 * time.Millisecond
        inCycle, hadEvent bool
    )
    defer debounce.Stop()
    for {
        select {
        case <-w.quit:
            return
        case <-w.ev:
            if !inCycle {
                debounce.Reset(debounceDuration)
                inCycle = true
            } else {
                hadEvent = true
            }
        case <-debounce.C:
            w.ac.mu.Lock()
            w.ac.reload()
            w.ac.mu.Unlock()
            if hadEvent {
                debounce.Reset(debounceDuration)
                inCycle, hadEvent = true, false
            } else {
                inCycle, hadEvent = false, false
            }
        }
    }
}