本站源代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

68 lines
1.7KB

  1. // Copyright 2015 go-swagger maintainers
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. // +build go1.9
  15. package swag
  16. import (
  17. "sort"
  18. "sync"
  19. )
  20. // indexOfInitialisms is a thread-safe implementation of the sorted index of initialisms.
  21. // Since go1.9, this may be implemented with sync.Map.
  22. type indexOfInitialisms struct {
  23. sortMutex *sync.Mutex
  24. index *sync.Map
  25. }
  26. func newIndexOfInitialisms() *indexOfInitialisms {
  27. return &indexOfInitialisms{
  28. sortMutex: new(sync.Mutex),
  29. index: new(sync.Map),
  30. }
  31. }
  32. func (m *indexOfInitialisms) load(initial map[string]bool) *indexOfInitialisms {
  33. m.sortMutex.Lock()
  34. defer m.sortMutex.Unlock()
  35. for k, v := range initial {
  36. m.index.Store(k, v)
  37. }
  38. return m
  39. }
  40. func (m *indexOfInitialisms) isInitialism(key string) bool {
  41. _, ok := m.index.Load(key)
  42. return ok
  43. }
  44. func (m *indexOfInitialisms) add(key string) *indexOfInitialisms {
  45. m.index.Store(key, true)
  46. return m
  47. }
  48. func (m *indexOfInitialisms) sorted() (result []string) {
  49. m.sortMutex.Lock()
  50. defer m.sortMutex.Unlock()
  51. m.index.Range(func(key, value interface{}) bool {
  52. k := key.(string)
  53. result = append(result, k)
  54. return true
  55. })
  56. sort.Sort(sort.Reverse(byInitialism(result)))
  57. return
  58. }
上海开阖软件有限公司 沪ICP备12045867号-1