PageRenderTime 63ms CodeModel.GetById 20ms RepoModel.GetById 0ms app.codeStats 0ms

/cmd/cluster-capacity/go/src/github.com/kubernetes-incubator/cluster-capacity/vendor/github.com/gophercloud/gophercloud/openstack/compute/v2/extensions/schedulerhints/requests.go

https://bitbucket.org/enterstudiosbiz/origin
Go | 164 lines | 108 code | 27 blank | 29 comment | 26 complexity | 108ecbfe03c45e2dddd6d903bb60005e MD5 | raw file
Possible License(s): Apache-2.0, 0BSD, GPL-2.0, BSD-2-Clause, BSD-3-Clause, MPL-2.0-no-copyleft-exception, MIT, LGPL-3.0
  1. package schedulerhints
  2. import (
  3. "net"
  4. "regexp"
  5. "strings"
  6. "github.com/gophercloud/gophercloud"
  7. "github.com/gophercloud/gophercloud/openstack/compute/v2/servers"
  8. )
  9. // SchedulerHints represents a set of scheduling hints that are passed to the
  10. // OpenStack scheduler.
  11. type SchedulerHints struct {
  12. // Group specifies a Server Group to place the instance in.
  13. Group string
  14. // DifferentHost will place the instance on a compute node that does not
  15. // host the given instances.
  16. DifferentHost []string
  17. // SameHost will place the instance on a compute node that hosts the given
  18. // instances.
  19. SameHost []string
  20. // Query is a conditional statement that results in compute nodes able to
  21. // host the instance.
  22. Query []interface{}
  23. // TargetCell specifies a cell name where the instance will be placed.
  24. TargetCell string `json:"target_cell,omitempty"`
  25. // BuildNearHostIP specifies a subnet of compute nodes to host the instance.
  26. BuildNearHostIP string
  27. // AdditionalProperies are arbitrary key/values that are not validated by nova.
  28. AdditionalProperties map[string]interface{}
  29. }
  30. // CreateOptsBuilder builds the scheduler hints into a serializable format.
  31. type CreateOptsBuilder interface {
  32. ToServerSchedulerHintsCreateMap() (map[string]interface{}, error)
  33. }
  34. // ToServerSchedulerHintsMap builds the scheduler hints into a serializable format.
  35. func (opts SchedulerHints) ToServerSchedulerHintsCreateMap() (map[string]interface{}, error) {
  36. sh := make(map[string]interface{})
  37. uuidRegex, _ := regexp.Compile("^[a-z0-9]{8}-[a-z0-9]{4}-[1-5][a-z0-9]{3}-[a-z0-9]{4}-[a-z0-9]{12}$")
  38. if opts.Group != "" {
  39. if !uuidRegex.MatchString(opts.Group) {
  40. err := gophercloud.ErrInvalidInput{}
  41. err.Argument = "schedulerhints.SchedulerHints.Group"
  42. err.Value = opts.Group
  43. err.Info = "Group must be a UUID"
  44. return nil, err
  45. }
  46. sh["group"] = opts.Group
  47. }
  48. if len(opts.DifferentHost) > 0 {
  49. for _, diffHost := range opts.DifferentHost {
  50. if !uuidRegex.MatchString(diffHost) {
  51. err := gophercloud.ErrInvalidInput{}
  52. err.Argument = "schedulerhints.SchedulerHints.DifferentHost"
  53. err.Value = opts.DifferentHost
  54. err.Info = "The hosts must be in UUID format."
  55. return nil, err
  56. }
  57. }
  58. sh["different_host"] = opts.DifferentHost
  59. }
  60. if len(opts.SameHost) > 0 {
  61. for _, sameHost := range opts.SameHost {
  62. if !uuidRegex.MatchString(sameHost) {
  63. err := gophercloud.ErrInvalidInput{}
  64. err.Argument = "schedulerhints.SchedulerHints.SameHost"
  65. err.Value = opts.SameHost
  66. err.Info = "The hosts must be in UUID format."
  67. return nil, err
  68. }
  69. }
  70. sh["same_host"] = opts.SameHost
  71. }
  72. /*
  73. Query can be something simple like:
  74. [">=", "$free_ram_mb", 1024]
  75. Or more complex like:
  76. ['and',
  77. ['>=', '$free_ram_mb', 1024],
  78. ['>=', '$free_disk_mb', 200 * 1024]
  79. ]
  80. Because of the possible complexity, just make sure the length is a minimum of 3.
  81. */
  82. if len(opts.Query) > 0 {
  83. if len(opts.Query) < 3 {
  84. err := gophercloud.ErrInvalidInput{}
  85. err.Argument = "schedulerhints.SchedulerHints.Query"
  86. err.Value = opts.Query
  87. err.Info = "Must be a conditional statement in the format of [op,variable,value]"
  88. return nil, err
  89. }
  90. sh["query"] = opts.Query
  91. }
  92. if opts.TargetCell != "" {
  93. sh["target_cell"] = opts.TargetCell
  94. }
  95. if opts.BuildNearHostIP != "" {
  96. if _, _, err := net.ParseCIDR(opts.BuildNearHostIP); err != nil {
  97. err := gophercloud.ErrInvalidInput{}
  98. err.Argument = "schedulerhints.SchedulerHints.BuildNearHostIP"
  99. err.Value = opts.BuildNearHostIP
  100. err.Info = "Must be a valid subnet in the form 192.168.1.1/24"
  101. return nil, err
  102. }
  103. ipParts := strings.Split(opts.BuildNearHostIP, "/")
  104. sh["build_near_host_ip"] = ipParts[0]
  105. sh["cidr"] = "/" + ipParts[1]
  106. }
  107. if opts.AdditionalProperties != nil {
  108. for k, v := range opts.AdditionalProperties {
  109. sh[k] = v
  110. }
  111. }
  112. return sh, nil
  113. }
  114. // CreateOptsExt adds a SchedulerHints option to the base CreateOpts.
  115. type CreateOptsExt struct {
  116. servers.CreateOptsBuilder
  117. // SchedulerHints provides a set of hints to the scheduler.
  118. SchedulerHints CreateOptsBuilder
  119. }
  120. // ToServerCreateMap adds the SchedulerHints option to the base server creation options.
  121. func (opts CreateOptsExt) ToServerCreateMap() (map[string]interface{}, error) {
  122. base, err := opts.CreateOptsBuilder.ToServerCreateMap()
  123. if err != nil {
  124. return nil, err
  125. }
  126. schedulerHints, err := opts.SchedulerHints.ToServerSchedulerHintsCreateMap()
  127. if err != nil {
  128. return nil, err
  129. }
  130. if len(schedulerHints) == 0 {
  131. return base, nil
  132. }
  133. base["os:scheduler_hints"] = schedulerHints
  134. return base, nil
  135. }