Exemplo n.º 1
0
func (manager *Manager) registerManagerResource(grpcAddr string) error {
	node, err := scale.GetOwnNodeName()
	if err != nil {
		return err
	}
	ttl := 30 * time.Second
	resource := "swarm://" + node
	res, success, err := scale.RegisterResourceCustom(
		ManagerService, resource, grpcAddr, node, ttl)
	if err != nil {
		return err
	}
	if !success {
		// Leftover from a previous life. Clean up and try again.
		err := scale.ExistingResource(ManagerService, resource, "").Deregister()
		if err != nil {
			return err
		}
		return manager.registerManagerResource(grpcAddr)
	}
	manager.resource = res
	manager.sessionSKA = scale.NewSessionSelfKeepAlive(
		res.GetSessionID(), ttl/2)
	return nil
}
Exemplo n.º 2
0
// RegisterThisHost registers the current swarm node against the current
// service.
func RegisterThisHost(
	hostAddr string) (
	serviceSka *scale.SelfKeepAlive, sessionSka *scale.SelfKeepAlive,
	err error) {
	service := leverutil.ServiceFlag.Get()
	instanceID := leverutil.InstanceIDFlag.Get()
	err = scale.RegisterServiceLocal(
		service, instanceID, hostAddr, 30*time.Second)
	if err != nil {
		return nil, nil, err
	}
	node, err := scale.GetOwnNodeName()
	if err != nil {
		return nil, nil, err
	}
	res, success, err := scale.RegisterResourceLocal(
		"swarm://"+node, 30*time.Second)
	if err != nil {
		return nil, nil, err
	}
	if !success {
		return nil, nil, fmt.Errorf("Duplicate Lever host on this node")
	}
	return scale.NewServiceSelfKeepAlive(instanceID, 15*time.Second),
		scale.NewSessionSelfKeepAlive(res.GetSessionID(), 15*time.Second),
		nil
}