Subversion Repositories SmartDukaan

Rev

Go to most recent revision | Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
30 ashish 1 
#
2 
# Licensed to the Apache Software Foundation (ASF) under one
3 
# or more contributor license agreements. See the NOTICE file
4 
# distributed with this work for additional information
5 
# regarding copyright ownership. The ASF licenses this file
6 
# to you under the Apache License, Version 2.0 (the
7 
# "License"); you may not use this file except in compliance
8 
# with the License. You may obtain a copy of the License at
9 
#
10 
#   http://www.apache.org/licenses/LICENSE-2.0
11 
#
12 
# Unless required by applicable law or agreed to in writing,
13 
# software distributed under the License is distributed on an
14 
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 
# KIND, either express or implied. See the License for the
16 
# specific language governing permissions and limitations
17 
# under the License.
18 
#
19 
 
20 
require 'set'
21 
 
22 
module Thrift
23 
  module Types
24 
    STOP = 0
25 
    VOID = 1
26 
    BOOL = 2
27 
    BYTE = 3
28 
    DOUBLE = 4
29 
    I16 = 6
30 
    I32 = 8
31 
    I64 = 10
32 
    STRING = 11
33 
    STRUCT = 12
34 
    MAP = 13
35 
    SET = 14
36 
    LIST = 15
37 
  end
38 
 
39 
  class << self
40 
    attr_accessor :type_checking
41 
  end
42 
 
43 
  class TypeError < Exception
44 
  end
45 
 
46 
  def self.check_type(value, field, name, skip_nil=true)
47 
    return if value.nil? and skip_nil
48 
    klasses = case field[:type]
49 
              when Types::VOID
50 
                NilClass
51 
              when Types::BOOL
52 
                [TrueClass, FalseClass]
53 
              when Types::BYTE, Types::I16, Types::I32, Types::I64
54 
                Integer
55 
              when Types::DOUBLE
56 
                Float
57 
              when Types::STRING
58 
                String
59 
              when Types::STRUCT
60 
                Struct
61 
              when Types::MAP
62 
                Hash
63 
              when Types::SET
64 
                Set
65 
              when Types::LIST
66 
                Array
67 
              end
68 
    valid = klasses && [*klasses].any? { |klass| klass === value }
69 
    raise TypeError, "Expected #{type_name(field[:type])}, received #{value.class} for field #{name}" unless valid
70 
    # check elements now
71 
    case field[:type]
72 
    when Types::MAP
73 
      value.each_pair do |k,v|
74 
        check_type(k, field[:key], "#{name}.key", false)
75 
        check_type(v, field[:value], "#{name}.value", false)
76 
      end
77 
    when Types::SET, Types::LIST
78 
      value.each do |el|
79 
        check_type(el, field[:element], "#{name}.element", false)
80 
      end
81 
    when Types::STRUCT
82 
      raise TypeError, "Expected #{field[:class]}, received #{value.class} for field #{name}" unless field[:class] == value.class
83 
    end
84 
  end
85 
 
86 
  def self.type_name(type)
87 
    Types.constants.each do |const|
88 
      return "Types::#{const}" if Types.const_get(const) == type
89 
    end
90 
    nil
91 
  end
92 
 
93 
  module MessageTypes
94 
    CALL = 1
95 
    REPLY = 2
96 
    EXCEPTION = 3
97 
    ONEWAY = 4
98 
  end
99 
end
100 
 
101 
Thrift.type_checking = false if Thrift.type_checking.nil?