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"))

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

	if viper.GetBool("settings.ldap.tls") {
		err = conn.StartTLS(&tls.Config{
			InsecureSkipVerify: true,
		})
		if err != nil {
			err = errors.New(fmt.Sprintf("升级到加密方式失败，%v", err))
			logger.Error(err)
			return
		}
	}

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

	return
}