-
Notifications
You must be signed in to change notification settings - Fork 90
/
interface.go
66 lines (57 loc) · 1.9 KB
/
interface.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
/**
* Copyright (c) 2016-present, Facebook, Inc.
* All rights reserved.
*
* This source code is licensed under the BSD-style license found in the
* LICENSE file in the root directory of this source tree. An additional grant
* of patent rights can be found in the PATENTS file in the same directory.
*/
package dhcplb
import (
"fmt"
"net"
)
// DHCPMessage represents coordinates of a dhcp message.
type DHCPMessage struct {
XID uint32
Peer *net.UDPAddr
ClientID []byte
Mac []byte
}
type id string
func (m *DHCPMessage) id() id {
return id(fmt.Sprintf("%s%d%x", m.Peer.IP, m.XID, m.ClientID))
}
// DHCPBalancingAlgorithm defines an interface for load balancing algorithms.
// Users can implement their own and add them to config.go (in the
// configSpec.algorithm method)
type DHCPBalancingAlgorithm interface {
SelectServerFromList(list []*DHCPServer, message *DHCPMessage) (*DHCPServer, error)
SelectRatioBasedDhcpServer(message *DHCPMessage) (*DHCPServer, error)
UpdateStableServerList(list []*DHCPServer) error
UpdateRCServerList(list []*DHCPServer) error
SetRCRatio(ratio uint32)
// An unique name for the algorithm, this string can be used in the
// configuration file, in the section where the algorithm is selecetd.
Name() string
}
// Server is the main interface implementing the DHCP server.
type Server interface {
SetConfig(config *Config)
ListenAndServe() error
HasServers() bool
}
// DHCPServerSourcer is an interface used to fetch stable, rc and servers from
// a "tier" (group of servers).
type DHCPServerSourcer interface {
GetStableServers() ([]*DHCPServer, error)
GetRCServers() ([]*DHCPServer, error)
GetServersFromTier(tier string) ([]*DHCPServer, error)
}
// Throttle is interface that implements rate limiting per key
type Throttle interface {
// Returns whether the rate is below max for a key
OK(interface{}) (bool, error)
// Returns the number of items
len() int
}