57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
|
package datastore
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/drone/drone/model"
|
||
|
|
"github.com/russross/meddler"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (db *datastore) GetAgent(id int64) (*model.Agent, error) {
|
||
|
|
var agent = new(model.Agent)
|
||
|
|
var err = meddler.Load(db, agentTable, agent, id)
|
||
|
|
return agent, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *datastore) GetAgentAddr(addr string) (*model.Agent, error) {
|
||
|
|
var agent = new(model.Agent)
|
||
|
|
var err = meddler.QueryRow(db, agent, rebind(agentAddrQuery), addr)
|
||
|
|
return agent, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *datastore) GetAgentList() ([]*model.Agent, error) {
|
||
|
|
var agents = []*model.Agent{}
|
||
|
|
var err = meddler.QueryAll(db, &agents, rebind(agentListQuery))
|
||
|
|
return agents, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *datastore) CreateAgent(agent *model.Agent) error {
|
||
|
|
return meddler.Insert(db, agentTable, agent)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *datastore) UpdateAgent(agent *model.Agent) error {
|
||
|
|
return meddler.Update(db, agentTable, agent)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *datastore) DeleteAgent(agent *model.Agent) error {
|
||
|
|
var _, err = db.Exec(rebind(agentDeleteStmt), agent.ID)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
const agentTable = "agents"
|
||
|
|
|
||
|
|
const agentAddrQuery = `
|
||
|
|
SELECT *
|
||
|
|
FROM agents
|
||
|
|
WHERE agent_addr=?
|
||
|
|
LIMIT 1
|
||
|
|
`
|
||
|
|
|
||
|
|
const agentListQuery = `
|
||
|
|
SELECT *
|
||
|
|
FROM agents
|
||
|
|
ORDER BY agent_addr ASC
|
||
|
|
`
|
||
|
|
|
||
|
|
const agentDeleteStmt = `
|
||
|
|
DELETE FROM agents WHERE agent_id = ?
|
||
|
|
`
|