ferry/pkg/ldap/connection.go

package ldap

import (
	"crypto/tls"
	"errors"
	"ferry/pkg/logger"
	"fmt"
	"time"

	"github.com/spf13/viper"

	"github.com/go-ldap/ldap/v3"
)

/*
  @Author : lanyulei
*/

var conn *ldap.Conn

// ldap连接
func ldapConnection() (err error) {
	var ldapConn = fmt.Sprintf("%v:%v", viper.GetString("settings.ldap.host"), viper.GetString("settings.ldap.port"))

	if viper.GetInt("settings.ldap.port") == 636 {
		conn, err = ldap.DialTLS(
			"tcp",
			ldapConn,
			&tls.Config{InsecureSkipVerify: true},
		)
	} else {
		conn, err = ldap.Dial(
			"tcp",
			ldapConn,
		)
	}

	if err != nil {
		err = errors.New(fmt.Sprintf("无法连接到ldap服务器，%v", err))
		logger.Error(err)
		return
	}

	//设置超时时间
	conn.SetTimeout(5 * time.Second)

	return
}