Method setLocalUserID

core/gstate.go:954–978

(c context.Context, conn *sql.Conn)

}

func (s *gstate) setLocalUserID(c context.Context, conn *sql.Conn) (err error) {
	if v := c.Value(UserIDKey); v == nil {
		return nil
	} else {
		var val string
		switch v1 := v.(type) {
		case string:
			val = v1
		case int:
			val = strconv.Itoa(v1)
		}

		q := s.getTargetPsqlCompiler().RenderSetSessionVar("user.id", val)
		if q == "" {
			return nil
		}

		if tx := s.tx(); tx != nil {
			_, err = tx.ExecContext(c, q)
		} else {
			_, err = conn.ExecContext(c, q)
		}
	}
	return
}

var errValidationFailed = errors.New("validation failed")