utils.go 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. package common
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "io/ioutil"
  6. "os"
  7. "path"
  8. "reflect"
  9. "regexp"
  10. "strings"
  11. "unicode"
  12. )
  13. var (
  14. environment = map[string]string{}
  15. wordRegexp = regexp.MustCompile(`\w+`)
  16. )
  17. // Setenv - cria ou atualiza uma variavel de ambiente dentro da aplicação
  18. func Setenv(id, value string) {
  19. environment[id] = value
  20. }
  21. // Getenv - recupera uma variavel de ambiente dentro da aplicação
  22. func Getenv(id string, fallback ...string) (value string) {
  23. var exist bool
  24. if value, exist = environment[id]; !exist {
  25. for _, value = range fallback {
  26. break
  27. }
  28. }
  29. return value
  30. }
  31. func LcFirst(str string) string {
  32. for i, v := range str {
  33. return string(unicode.ToLower(v)) + str[i+1:]
  34. }
  35. return ""
  36. }
  37. func UpFirst(str string) string {
  38. for i, v := range str {
  39. return string(unicode.ToUpper(v)) + str[i+1:]
  40. }
  41. return ""
  42. }
  43. func JSONStringfy(v interface{}) (string, error) {
  44. o, err := json.Marshal(v)
  45. if err != nil {
  46. return "", err
  47. }
  48. return string(o), nil
  49. }
  50. // FileExists reports whether the named file or directory exists.
  51. func FileExists(name string) bool {
  52. if _, err := os.Stat(name); err != nil {
  53. if os.IsNotExist(err) {
  54. return false
  55. }
  56. }
  57. return true
  58. }
  59. func ResolveParams(format string, data map[string]interface{}) string {
  60. for k, v := range data {
  61. format = strings.Replace(
  62. format,
  63. fmt.Sprintf("${%s}", k),
  64. fmt.Sprintf("%v", v),
  65. -1,
  66. )
  67. }
  68. return format
  69. }
  70. func UnderToCamel(input string) string {
  71. input = strings.ToLower(input)
  72. input = strings.ReplaceAll(input, "_", " ")
  73. input = strings.Title(input)
  74. return strings.ReplaceAll(input, " ", "")
  75. }
  76. func Mkdir(perm os.FileMode, paths ...string) (err error) {
  77. for _, path := range paths {
  78. if _, err := os.Stat(path); os.IsNotExist(err) {
  79. if err = os.MkdirAll(path, perm); err != nil {
  80. return err
  81. }
  82. }
  83. }
  84. return
  85. }
  86. func JsonParseMode(path, mode string, v interface{}) (err error) {
  87. defer func() {
  88. if rec := recover(); rec != nil {
  89. err = fmt.Errorf("%s", rec)
  90. }
  91. }()
  92. if err = ParseJson(path+".json", v); err != nil {
  93. return
  94. }
  95. // fmt.Println(fmt.Sprintf("%s.%s.json", path, mode))
  96. err = ParseJson(fmt.Sprintf("%s.%s.json", path, mode), v, true)
  97. return
  98. }
  99. func ParseJson(filename string, v interface{}, ignoreFileExists ...bool) error {
  100. // fmt.Println("load file ", filename)
  101. data, err := FileGetContents(filename)
  102. if err != nil {
  103. if len(ignoreFileExists) > 0 && ignoreFileExists[0] {
  104. return nil
  105. }
  106. return err
  107. }
  108. if err = json.Unmarshal([]byte(data), v); err != nil {
  109. return err
  110. }
  111. return nil
  112. }
  113. func CamelToUnder(input string) string {
  114. out := camelToUnderRegex.ReplaceAllStringFunc(input, func(m string) string {
  115. return "_" + strings.ToLower(m)
  116. })
  117. return strings.TrimLeft(out, "_")
  118. }
  119. func FilePutContents(filename string, content string, perm os.FileMode) error {
  120. return FilePutContentsBytes(filename, []byte(content), perm)
  121. }
  122. func FilePutContentsBytes(filename string, content []byte, perm os.FileMode) error {
  123. var (
  124. dir = path.Dir(filename)
  125. )
  126. if _, err := os.Stat(dir); os.IsNotExist(err) {
  127. if err = os.MkdirAll(dir, perm); err != nil {
  128. return err
  129. }
  130. }
  131. return ioutil.WriteFile(filename, []byte(content), 0644)
  132. }
  133. func RemoveFile(file string) error {
  134. if err := os.Remove(file); err != nil {
  135. return err
  136. }
  137. return nil
  138. }
  139. func GetFiles(directory string) ([]os.FileInfo, error) {
  140. files, err := ioutil.ReadDir(directory)
  141. if err != nil {
  142. return []os.FileInfo{}, err
  143. }
  144. return files, nil
  145. }
  146. func FileGetContents(filename string) (string, error) {
  147. b, err := ioutil.ReadFile(filename)
  148. if err != nil {
  149. return "", err
  150. }
  151. return string(b), nil
  152. }
  153. func WriteToJson(path string, object interface{}) (err error) {
  154. var data []byte
  155. if data, err = json.MarshalIndent(object, "", " "); err == nil {
  156. err = FilePutContentsBytes(path, data, 0777)
  157. }
  158. return
  159. }
  160. func IsEmpty(reference interface{}) bool {
  161. return isEmptyReflection(reflect.ValueOf(reference))
  162. }
  163. func isEmptyReflection(v reflect.Value) bool {
  164. switch v.Kind() {
  165. case reflect.String, reflect.Array:
  166. return v.Len() == 0
  167. case reflect.Map, reflect.Slice:
  168. return v.Len() == 0 || v.IsNil()
  169. // case reflect.Bool:
  170. // return !v.Bool()
  171. // case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
  172. // return v.Int() == 0
  173. // case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
  174. // return v.Uint() == 0
  175. // case reflect.Float32, reflect.Float64:
  176. // return v.Float() == 0
  177. case reflect.Interface, reflect.Ptr:
  178. return v.IsNil()
  179. }
  180. return reflect.DeepEqual(v.Interface(), reflect.Zero(v.Type()).Interface())
  181. }