utils.go 1009 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package util
  2. import (
  3. "encoding/json"
  4. "encoding/xml"
  5. )
  6. const (
  7. _ = iota
  8. // KB kilo byte
  9. KB int = 1 << (10 * iota)
  10. // MB mega byte
  11. MB
  12. // GB giga byte
  13. GB
  14. // TB tera byte
  15. //TB
  16. // PB peta byte
  17. //PB
  18. )
  19. // Map is a shortcut for map[string]interface{}
  20. type Map map[string]interface{}
  21. // String map to json string
  22. func (m Map) String() string {
  23. bs, _ := json.Marshal(m)
  24. return string(bs)
  25. }
  26. // Bytes map to json bytes
  27. func (m Map) Bytes() []byte {
  28. bs, _ := json.Marshal(m)
  29. return bs
  30. }
  31. // MarshalXML allows type Map to be used with xml.Marshal.
  32. func (m Map) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
  33. start.Name = xml.Name{
  34. Space: "",
  35. Local: "map",
  36. }
  37. if err := e.EncodeToken(start); err != nil {
  38. return err
  39. }
  40. for key, value := range m {
  41. elem := xml.StartElement{
  42. Name: xml.Name{Space: "", Local: key},
  43. Attr: []xml.Attr{},
  44. }
  45. if err := e.EncodeElement(value, elem); err != nil {
  46. return err
  47. }
  48. }
  49. return e.EncodeToken(xml.EndElement{Name: start.Name})
  50. }