github.com/onsi/gomega@v1.32.0/gmeasure/enum_support.go (about)

     1  package gmeasure
     2  
     3  import "encoding/json"
     4  
     5  type enumSupport struct {
     6  	toString map[uint]string
     7  	toEnum   map[string]uint
     8  	maxEnum  uint
     9  }
    10  
    11  func newEnumSupport(toString map[uint]string) enumSupport {
    12  	toEnum, maxEnum := map[string]uint{}, uint(0)
    13  	for k, v := range toString {
    14  		toEnum[v] = k
    15  		if maxEnum < k {
    16  			maxEnum = k
    17  		}
    18  	}
    19  	return enumSupport{toString: toString, toEnum: toEnum, maxEnum: maxEnum}
    20  }
    21  
    22  func (es enumSupport) String(e uint) string {
    23  	if e > es.maxEnum {
    24  		return es.toString[0]
    25  	}
    26  	return es.toString[e]
    27  }
    28  
    29  func (es enumSupport) UnmarshJSON(b []byte) (uint, error) {
    30  	var dec string
    31  	if err := json.Unmarshal(b, &dec); err != nil {
    32  		return 0, err
    33  	}
    34  	out := es.toEnum[dec] // if we miss we get 0 which is what we want anyway
    35  	return out, nil
    36  }
    37  
    38  func (es enumSupport) MarshJSON(e uint) ([]byte, error) {
    39  	if e == 0 || e > es.maxEnum {
    40  		return json.Marshal(nil)
    41  	}
    42  	return json.Marshal(es.toString[e])
    43  }