use triOpt in node set (#230)

projecteru2 · Jul 27, 2020 · d9015db · d9015db
1 parent 4af90c7
commit d9015db
Show file tree

Hide file tree

Showing 6 changed files with 284 additions and 291 deletions.
diff --git a/cluster/calcium/node.go b/cluster/calcium/node.go
@@ -3,7 +3,6 @@ package calcium
 import (
 	"context"
 
-	"github.com/projecteru2/core/cluster"
 	"github.com/projecteru2/core/types"
 	"github.com/sanity-io/litter"
 	log "github.com/sirupsen/logrus"
@@ -39,7 +38,7 @@ func (c *Calcium) SetNode(ctx context.Context, opts *types.SetNodeOptions) (*typ
 		opts.Normalize(node)
 		n = node
 		litter.Dump(opts)
-		n.Available = (opts.Status == cluster.NodeUp) || (opts.Status == cluster.KeepNodeStatus && n.Available)
+		n.Available = (opts.Status == types.TriTrue) || (opts.Status == types.TriKeep && n.Available)
 		if opts.ContainersDown {
 			containers, err := c.store.ListNodeContainers(ctx, opts.Nodename, nil)
 			if err != nil {

diff --git a/cluster/cluster.go b/cluster/cluster.go
@@ -40,12 +40,6 @@ const (
 	ContainerLock = "clock_%s"
 	// NodeLock for lock node
 	NodeLock = "cnode_%s_%s"
-	// NodeUp for node up
-	NodeUp = 1
-	// NodeDown for node down
-	NodeDown = 0
-	// KeepNodeStatus for no change node status
-	KeepNodeStatus = 2
 )
 
 // Cluster define all interface