diff --git a/metrics/generic/generic_test.go b/metrics/generic/generic_test.go index cb725e02f..b5747f97b 100644 --- a/metrics/generic/generic_test.go +++ b/metrics/generic/generic_test.go @@ -5,6 +5,12 @@ package generic_test // generic to use its Histogram in the Quantiles helper function. import ( + "go/ast" + "go/importer" + "go/parser" + "go/token" + "go/types" + "io/ioutil" "math" "math/rand" "sync" @@ -107,3 +113,72 @@ func TestSimpleHistogram(t *testing.T) { t.Errorf("want %f, have %f", want, have) } } + +// Naive atomic alignment test. +// The problem is related to the use of `atomic.*` and not directly to a structure. +// But currently works for Counter and Gauge. +// To have a more solid test, this test should be removed and the other tests should be run on a 32-bit arch. +func TestAtomicAlignment(t *testing.T) { + content, err := ioutil.ReadFile("./generic.go") + if err != nil { + t.Fatal(err) + } + + fset := token.NewFileSet() + + file, err := parser.ParseFile(fset, "generic.go", content, parser.ParseComments) + if err != nil { + t.Fatal(err) + } + + conf := types.Config{Importer: importer.ForCompiler(fset, "source", nil)} + + pkg, err := conf.Check(".", fset, []*ast.File{file}, nil) + if err != nil { + t.Fatal(err) + } + + // uses ARM as reference for 32-bit arch + sizes := types.SizesFor("gc", "arm") + + names := []string{"Counter", "Gauge"} + + for _, name := range names { + t.Run(name, func(t *testing.T) { + checkAtomicAlignment(t, sizes, pkg.Scope().Lookup(name), pkg) + }) + } +} + +func checkAtomicAlignment(t *testing.T, sizes types.Sizes, obj types.Object, pkg *types.Package) { + t.Helper() + + st := obj.Type().Underlying().(*types.Struct) + + posToCheck := make(map[int]types.BasicKind) + + var vars []*types.Var + for i := 0; i < st.NumFields(); i++ { + field := st.Field(i) + + if v, ok := field.Type().(*types.Basic); ok { + switch v.Kind() { + case types.Uint64, types.Float64, types.Int64: + posToCheck[i] = v.Kind() + } + } + + vars = append(vars, types.NewVar(field.Pos(), pkg, field.Name(), field.Type())) + } + + offsets := sizes.Offsetsof(vars) + for i, offset := range offsets { + if _, ok := posToCheck[i]; !ok { + continue + } + + if offset%8 != 0 { + t.Errorf("misalignment detected in %s, offset %d", obj.Name(), offset) + } + } +}