plugins/core/instrument/struct.go

// Licensed to Apache Software Foundation (ASF) under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Apache Software Foundation (ASF) licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package instrument

import "github.com/dave/dst"

// WithFieldExists filter the struct has the field with specific name.
func WithFieldExists(fieldName string) StructFilterOption {
	return func(structType *dst.TypeSpec, files []*dst.File) bool {
		st := structType.Type.(*dst.StructType)
		for _, field := range st.Fields.List {
			if field.Names[0].Name == fieldName {
				return true
			}
		}
		return false
	}
}

// WithFiledType filter the struct has the field with specific name and type.
func WithFiledType(filedName, filedType string) StructFilterOption {
	return func(structType *dst.TypeSpec, files []*dst.File) bool {
		st := structType.Type.(*dst.StructType)
		for _, field := range st.Fields.List {
			if filedName == "" && (len(field.Names) == 0 || field.Names[0].Name == filedName) {
				return verifyTypeName(field.Type, filedType)
			} else if field.Names[0].Name == filedName {
				return verifyTypeName(field.Type, filedType)
			}
		}
		return false
	}
}