peter/websub-hub
1
0
Fork 0
Code Issues 1 Pull Requests Releases Activity

Rewrite subscriber if it exists

Browse Source
This commit is contained in:
Peter Stuifzand 2018-01-30 21:36:00 +01:00
parent 7167b61c5e
commit 62b4e4012d
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmd/hubserver/main.go
View File

@ -118,7 +118,6 @@ func (handler *subscriptionHandler) handleSubscription(w http.ResponseWriter, r
log.Println(validationURL) log.Println(validationURL)
if validateURL(validationURL.String(), ourChallenge) { if validateURL(validationURL.String(), ourChallenge) {
// challenge accepted
handler.addSubscriberCallback(topicURL.String(), Subscriber{callbackURL.String(), leaseSeconds, secret}) handler.addSubscriberCallback(topicURL.String(), Subscriber{callbackURL.String(), leaseSeconds, secret})
} }
}() }()
@ -151,8 +150,18 @@ func validateURL(url, challenge string) bool {
} }
func (handler *subscriptionHandler) addSubscriberCallback(topic string, subscriber Subscriber) { func (handler *subscriptionHandler) addSubscriberCallback(topic string, subscriber Subscriber) {
defer handler.save()
if subs, e := handler.Subscribers[topic]; e {
for i, sub := range subs {
if sub.Callback == subscriber.Callback {
handler.Subscribers[topic][i] = subscriber
return
}
}
}
// not found create a new subscription
handler.Subscribers[topic] = append(handler.Subscribers[topic], subscriber) handler.Subscribers[topic] = append(handler.Subscribers[topic], subscriber)
handler.save()
} }
func (handler *subscriptionHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (handler *subscriptionHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {