-
Notifications
You must be signed in to change notification settings - Fork 0
/
sizeof.go
100 lines (86 loc) · 2.26 KB
/
sizeof.go
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package sizeof
import (
"reflect"
)
func SizeOf(i interface{}) int64 {
return sizeOf(reflect.ValueOf(i))
}
func a() {
}
func sizeOf(valOf reflect.Value) int64 {
var typeOf = valOf.Type()
var typeKind = typeOf.Kind()
switch typeKind {
case reflect.Bool, reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32,
reflect.Int64, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32,
reflect.Uint64, reflect.Uintptr, reflect.Float32, reflect.Float64, reflect.Complex64,
reflect.Complex128, reflect.Chan, reflect.UnsafePointer, reflect.Func:
return int64(typeOf.Size())
case reflect.Ptr:
return sizeOfPtr(valOf)
case reflect.Array:
return int64(typeOf.Size())
case reflect.Map:
return sizeOfMap(valOf)
case reflect.Slice:
return sizeOfSlice(valOf)
case reflect.String:
return sizeOfString(valOf)
case reflect.Struct:
return sizeOfStruct(valOf)
case reflect.Interface:
return sizeOf(valOf.Elem()) + int64(valOf.Type().Size())
}
return 0
}
func sizeOfPtr(valOf reflect.Value) int64 {
if valOf.IsNil() {
return 0
}
return int64(valOf.Type().Size()) + sizeOf(valOf.Elem())
}
func sizeOfSlice(valOf reflect.Value) int64 {
if valOf.IsNil() {
return 0
}
var sizeOfArr int64
for i := 0; i < valOf.Len(); i++ {
sizeOfArr += sizeOf(valOf.Index(i))
}
return sizeOfArr + int64(valOf.Type().Size())
}
func sizeOfMap(valOf reflect.Value) int64 {
if valOf.IsNil() {
return 0
}
var sizeOfMap int64
var iter = valOf.MapRange()
for iter.Next() {
sizeOfMap += sizeOf(iter.Key())
sizeOfMap += sizeOf(iter.Value())
}
return sizeOfMap + int64(valOf.Type().Size())
}
func sizeOfString(valOf reflect.Value) int64 {
return int64(valOf.Type().Size()) + int64(len(valOf.String())*8)
}
func sizeOfStruct(valOf reflect.Value) int64 {
var sizeOfStruct int64
var sizeOfFields int64
for i := 0; i < valOf.NumField(); i++ {
var field = valOf.Field(i)
var size = sizeOf(field)
sizeOfStruct += size
if size == 0 {
switch field.Kind() {
case reflect.Chan, reflect.Func, reflect.Interface, reflect.Map, reflect.Ptr, reflect.Slice:
if field.IsNil() {
sizeOfStruct += int64(field.Type().Size())
}
}
}
sizeOfFields += int64(field.Type().Size())
}
var align = int64(valOf.Type().Size()) - sizeOfFields
return sizeOfStruct + align
}