v0.9.9 : Handle nil connection when LDAP server is down
This commit is contained in:
+18
-2
@@ -33,7 +33,7 @@ import (
|
||||
)
|
||||
|
||||
const (
|
||||
version = "0.9.8"
|
||||
version = "0.9.9"
|
||||
// Match le nom de domaine, soit les 2 dernieres chaines separees par un point.
|
||||
// Peut aussi terminer par un point (Ex: srv.domaine.com. => domaine.com)
|
||||
DomainRegexpFormat = `([0-9A-Za-z\-]*)\.([0-9A-Za-z\-]*)(?:\.)?$`
|
||||
@@ -55,12 +55,14 @@ var (
|
||||
timeout *int
|
||||
)
|
||||
|
||||
// attempt is useless now. It was introduced to break after N attempts, but this is not implemented. Should it be?
|
||||
func ldapSearch(searchReq *ldap.SearchRequest, attempt int) (*ldap.SearchResult, error) {
|
||||
mutex.Lock()
|
||||
result, err := conLdap.Search(searchReq)
|
||||
mutex.Unlock()
|
||||
// Let's just manage connection errors here
|
||||
if err != nil && strings.HasSuffix(err.Error(), "ldap: connection closed") {
|
||||
if err != nil {
|
||||
if strings.HasSuffix(err.Error(), "ldap: connection closed") {
|
||||
logstream.Err("LDAP connection closed, retrying")
|
||||
mutex.Lock()
|
||||
conLdap.Close()
|
||||
@@ -72,6 +74,20 @@ func ldapSearch(searchReq *ldap.SearchRequest, attempt int) (*ldap.SearchResult,
|
||||
attempt = attempt + 1
|
||||
return ldapSearch(searchReq, attempt)
|
||||
}
|
||||
} else if err == ldap.ErrNilConnection {
|
||||
logstream.Err("LDAP connection is nil, reconnecting")
|
||||
mutex.Lock()
|
||||
// conLdap is nil so this would be a sigsegv/panic
|
||||
//conLdap.Close()
|
||||
conLdap, err = connectLdap()
|
||||
mutex.Unlock()
|
||||
if err != nil {
|
||||
return result, err
|
||||
} else {
|
||||
attempt = attempt + 1
|
||||
return ldapSearch(searchReq, attempt)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return result, err
|
||||
|
||||
Reference in New Issue
Block a user