зеркало из https://github.com/github/ruby.git
load.c: resolve_feature_path
* load.c (rb_resolve_feature_path): search the path for already loaded feature. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@66661 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
Родитель
c5e8627f65
Коммит
58508e8b55
14
load.c
14
load.c
|
@ -840,8 +840,10 @@ rb_f_require_relative(VALUE obj, VALUE fname)
|
|||
return rb_require_safe(rb_file_absolute_path(fname, base), rb_safe_level());
|
||||
}
|
||||
|
||||
typedef int (*feature_func)(const char *feature, const char *ext, int rb, int expanded, const char **fn);
|
||||
|
||||
static int
|
||||
search_required(VALUE fname, volatile VALUE *path, int safe_level)
|
||||
search_required(VALUE fname, volatile VALUE *path, int safe_level, feature_func rb_feature_p)
|
||||
{
|
||||
VALUE tmp;
|
||||
char *ext, *ftptr;
|
||||
|
@ -945,6 +947,12 @@ load_ext(VALUE path)
|
|||
|
||||
/* Method is documented in vm.c */
|
||||
|
||||
static int
|
||||
no_feature_p(const char *feature, const char *ext, int rb, int expanded, const char **fn)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
VALUE
|
||||
rb_resolve_feature_path(VALUE klass, VALUE fname)
|
||||
{
|
||||
|
@ -954,7 +962,7 @@ rb_resolve_feature_path(VALUE klass, VALUE fname)
|
|||
|
||||
fname = rb_get_path_check(fname, 0);
|
||||
path = rb_str_encode_ospath(fname);
|
||||
found = search_required(path, &path, 0);
|
||||
found = search_required(path, &path, 0, no_feature_p);
|
||||
|
||||
switch (found) {
|
||||
case 'r':
|
||||
|
@ -1003,7 +1011,7 @@ rb_require_internal(VALUE fname, int safe)
|
|||
rb_set_safe_level_force(0);
|
||||
|
||||
RUBY_DTRACE_HOOK(FIND_REQUIRE_ENTRY, RSTRING_PTR(fname));
|
||||
found = search_required(path, &path, safe);
|
||||
found = search_required(path, &path, safe, rb_feature_p);
|
||||
RUBY_DTRACE_HOOK(FIND_REQUIRE_RETURN, RSTRING_PTR(fname));
|
||||
|
||||
if (found) {
|
||||
|
|
|
@ -897,4 +897,24 @@ class TestRequire < Test::Unit::TestCase
|
|||
assert_operator(result, :end_with?, "/real/a.rb")
|
||||
}
|
||||
end
|
||||
|
||||
if defined?(RubyVM.resolve_feature_path)
|
||||
def test_resolve_feature_path
|
||||
paths, loaded = $:.dup, $".dup
|
||||
Dir.mktmpdir do |tmp|
|
||||
Tempfile.create(%w[feature .rb], tmp) do |file|
|
||||
file.close
|
||||
path = File.realpath(file.path)
|
||||
dir, base = File.split(path)
|
||||
$:.unshift(dir)
|
||||
assert_equal([:rb, path], RubyVM.resolve_feature_path(base))
|
||||
$".push(path)
|
||||
assert_equal([:rb, path], RubyVM.resolve_feature_path(base))
|
||||
end
|
||||
end
|
||||
ensure
|
||||
$:.replace(paths)
|
||||
$".replace(loaded)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
Загрузка…
Ссылка в новой задаче