aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/type/date_time.rb
blob: e8614b16e044536c136b682d35ba90c44e96c481 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
module ActiveRecord
  module Type
    class DateTime < Value # :nodoc:
      include Helpers::TimeValue
      include Helpers::AcceptsMultiparameterTime.new(
        defaults: { 4 => 0, 5 => 0 }
      )

      def type
        :datetime
      end

      def type_cast_for_database(value)
        return super unless value.acts_like?(:time)

        zone_conversion_method = ActiveRecord::Base.default_timezone == :utc ? :getutc : :getlocal

        if value.respond_to?(zone_conversion_method)
          value = value.send(zone_conversion_method)
        end

        return value unless has_precision?

        result = value.to_s(:db)
        if value.respond_to?(:usec) && (1..6).cover?(precision)
          "#{result}.#{sprintf("%0#{precision}d", value.usec / 10 ** (6 - precision))}"
        else
          result
        end
      end

      private

      alias has_precision? precision

      def cast_value(string)
        return string unless string.is_a?(::String)
        return if string.empty?

        fast_string_to_time(string) || fallback_string_to_time(string)
      end

      # '0.123456' -> 123456
      # '1.123456' -> 123456
      def microseconds(time)
        time[:sec_fraction] ? (time[:sec_fraction] * 1_000_000).to_i : 0
      end

      def fallback_string_to_time(string)
        time_hash = ::Date._parse(string)
        time_hash[:sec_fraction] = microseconds(time_hash)

        new_time(*time_hash.values_at(:year, :mon, :mday, :hour, :min, :sec, :sec_fraction, :offset))
      end

      def value_from_multiparameter_assignment(values_hash)
        missing_parameter = (1..3).detect { |key| !values_hash.key?(key) }
        if missing_parameter
          raise ArgumentError, missing_parameter
        end
        super
      end
    end
  end
end