Ruby On Rails--rake 任务中定义方法的陷阱

桃扇骨 2022-07-26 04:09 182阅读 0赞

### 问题描述 ###

在rails项目中，有时候需要导出数据、处理历史数据等，这时用rake是很方便的。

在rake中，有时候由于逻辑比较复杂，所以我们就会分离逻辑或者需要重用代码，这时候可能会定义方法。

但是，在rake中定义方法有一个问题：在不同rake文件和不同命名空间下定义的方法，如果存在了同名方法，那么后来定义的方法会覆盖掉之前定义的方法，这时候可能会导致bug。

下面我们用一个例子来具体说明下：  
比如有个项目 sample\_app，我们有2个rake文件：  
sample\_app/lib/tasks/export\_data/export\_users.rake  
sample\_app/lib/tasks/handle\_old\_data/hanlde\_old\_users.rake  
在项目中排列顺序和这里列出的顺序一致。

对应代码如下：

# sample_app/lib/tasks/export_data/export_users.rake
    namespace :export_data
      task export_users: :environment do
        user_data = get_user_data
        # other operations
      end
    
      def get_user_data
        puts "In export_data, get_user_data"
        # return user_data
      end
    end
    
    #sample_app/lib/tasks/handle_old_data/hanlde_old_users.rake
    namespace :handle_old_data
      task handle_old_users: :environment do
          user_data = get_user_data
        # other operations
      end
    
      def get_user_data
        puts "In handle_old_data, get_user_data"
        # return user_data
      end
    end

在命令行执行 rake handle\_old\_data:handle\_old\_users时会打印出：  
In handle\_old\_data, get\_user\_data

这时如果在命令行执行 rake export\_data:export\_users ,那么会打印出来什么呢？  
答案是：  
In handle\_old\_data, get\_user\_data

也就是说，在第一个rake中调用 get\_user\_data 方法时，实际上是调用的 第二个rake中定义的该方法。

那么这是为什么呢？

### 问题解答和解决方案 ###

这个问题出现的原因，其实仔细思考一下是不难得到答案的：  
lib/tasks/ 目录下的文件是依次加载的，而且rake中的命名空间（namespace）是只对rake任务起作用的，不会对文件中的方法起作用，所以后加载的第二个rake文件中的方法定义覆盖掉了前一个同名方法。

那么，怎么解决这个问题呢？

也许有人会说，不要用同名方法不就行了吗？

这样当然是一种解决问题的方式。但是这样对于开发者来说很不友好–谁会喜欢在定义方法前，去全局搜索下别人有没有用过这个方法名？

所以，我们需要更加优雅的解决方案。

我们可以有一种方式来很好地解决这个问题：  
我们在rake中定义方法时，给方法加上命名空间–可以新定义一个class或者一个module，然后将方法定义到其中;这样相当于将我们的方法定义限制在了我们期望的范围内。

以同时使用module和class为例，可以这样解决问题：

# sample_app/lib/tasks/export_data/export_users.rake
    namespace :export_data
      task export_users: :environment do
        user_data = ExportData::ExportUsers.get_user_data
        # other operations
      end
    
      module ExportData
        class ExportUsers
          def self.get_user_data
            puts "In export_data, get_user_data"
            # return user_data
          end
        end
      end
    
    end
    
    #sample_app/lib/tasks/handle_old_data/hanlde_old_users.rake
    namespace :handle_old_data
      task handle_old_users: :environment do
          user_data = HandleOldData::HandleOldUsers.get_user_data
        # other operations
      end
    
      module HandleOldData
        class HandleOldUsers
          def self.get_user_data
            puts "In handle_old_data, get_user_data"
            # return user_data
          end
        end
      end
    end

这样，这两个方法就相互不影响。

在命令行执行 rake handle\_old\_data:handle\_old\_users 时会打印出：  
In handle\_old\_data, get\_user\_data

这时如果在命令行执行 rake export\_data:export\_users ,那么会打印出来：  
In export\_data, get\_user\_data