50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package store
|
|
|
|
import (
|
|
"github.com/drone/drone/model"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type NodeStore interface {
|
|
// Get gets a user by unique ID.
|
|
Get(int64) (*model.Node, error)
|
|
|
|
// GetList gets a list of all nodes in the system.
|
|
GetList() ([]*model.Node, error)
|
|
|
|
// Count gets a count of all nodes in the system.
|
|
Count() (int, error)
|
|
|
|
// Create creates a node.
|
|
Create(*model.Node) error
|
|
|
|
// Update updates a node.
|
|
Update(*model.Node) error
|
|
|
|
// Delete deletes a node.
|
|
Delete(*model.Node) error
|
|
}
|
|
|
|
func GetNode(c context.Context, id int64) (*model.Node, error) {
|
|
return FromContext(c).Nodes().Get(id)
|
|
}
|
|
|
|
func GetNodeList(c context.Context) ([]*model.Node, error) {
|
|
return FromContext(c).Nodes().GetList()
|
|
}
|
|
|
|
func CountNodes(c context.Context) (int, error) {
|
|
return FromContext(c).Nodes().Count()
|
|
}
|
|
|
|
func CreateNode(c context.Context, node *model.Node) error {
|
|
return FromContext(c).Nodes().Create(node)
|
|
}
|
|
|
|
func UpdateNode(c context.Context, node *model.Node) error {
|
|
return FromContext(c).Nodes().Update(node)
|
|
}
|
|
|
|
func DeleteNode(c context.Context, node *model.Node) error {
|
|
return FromContext(c).Nodes().Delete(node)
|
|
}
|