-
Notifications
You must be signed in to change notification settings - Fork 28
/
notification_connection.go
48 lines (41 loc) · 1.69 KB
/
notification_connection.go
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
/*
* Copyright 2018 The Service Manager Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package postgres
import (
"time"
notificationConnection "github.com/Peripli/service-manager/storage/postgres/notification_connection"
"github.com/lib/pq"
)
//go:generate counterfeiter . notificationConnectionCreator
type notificationConnectionCreator interface {
// NewConnection returns new connection with callback for events
NewConnection(eventCallback func(isRunning bool, err error)) notificationConnection.NotificationConnection
}
type notificationConnectionCreatorImpl struct {
storageURI string
minReconnectInterval time.Duration
maxReconnectInterval time.Duration
}
func (ncci *notificationConnectionCreatorImpl) NewConnection(eventCallback func(isRunning bool, err error)) notificationConnection.NotificationConnection {
return pq.NewListener(ncci.storageURI, ncci.minReconnectInterval, ncci.maxReconnectInterval, func(event pq.ListenerEventType, err error) {
switch event {
case pq.ListenerEventConnected, pq.ListenerEventReconnected:
eventCallback(true, err)
case pq.ListenerEventDisconnected, pq.ListenerEventConnectionAttemptFailed:
eventCallback(false, err)
}
})
}