diff options
author | Matthew Heon <matthew.heon@gmail.com> | 2018-03-26 11:40:57 -0400 |
---|---|---|
committer | Atomic Bot <atomic-devel@projectatomic.io> | 2018-03-26 20:18:24 +0000 |
commit | b18f089545b27aaa09b565651851822d18ef0dc3 (patch) | |
tree | 3fc09828c79f034b281f65192f9dfb4154906221 /libpod/sql_state.go | |
parent | f2894f243bb39f6daae0daee305ec5899ca3e4d1 (diff) | |
download | podman-b18f089545b27aaa09b565651851822d18ef0dc3.tar.gz podman-b18f089545b27aaa09b565651851822d18ef0dc3.tar.bz2 podman-b18f089545b27aaa09b565651851822d18ef0dc3.zip |
Remove current SQLite DB driver
The SQLite DB backend has become an unmanageable nightmare. I
like having the option for DB work, but it's become an active
hindrance to further development, and it's definitely not in any
shape to be actively used.
Signed-off-by: Matthew Heon <matthew.heon@gmail.com>
Closes: #548
Approved by: baude
Diffstat (limited to 'libpod/sql_state.go')
-rw-r--r-- | libpod/sql_state.go | 1091 |
1 files changed, 0 insertions, 1091 deletions
diff --git a/libpod/sql_state.go b/libpod/sql_state.go deleted file mode 100644 index f583febda..000000000 --- a/libpod/sql_state.go +++ /dev/null @@ -1,1091 +0,0 @@ -package libpod - -import ( - "database/sql" - "encoding/json" - "os" - - "github.com/containernetworking/cni/pkg/types" - cnitypes "github.com/containernetworking/cni/pkg/types/current" - "github.com/pkg/errors" - "github.com/sirupsen/logrus" - - // Use SQLite backend for sql package - _ "github.com/mattn/go-sqlite3" -) - -// DBSchema is the current DB schema version -// Increments every time a change is made to the database's tables -const DBSchema = 14 - -// SQLState is a state implementation backed by a persistent SQLite3 database -type SQLState struct { - db *sql.DB - specsDir string - lockDir string - runtime *Runtime - valid bool -} - -// NewSQLState initializes a SQL-backed state, created the database if necessary -func NewSQLState(dbPath, specsDir, lockDir string, runtime *Runtime) (State, error) { - state := new(SQLState) - - state.runtime = runtime - - // Make the directory that will hold JSON copies of container runtime specs - if err := os.MkdirAll(specsDir, 0750); err != nil { - // The directory is allowed to exist - if !os.IsExist(err) { - return nil, errors.Wrapf(err, "error creating OCI specs dir %s", specsDir) - } - } - state.specsDir = specsDir - - // Make the directory that will hold container lockfiles - if err := os.MkdirAll(lockDir, 0750); err != nil { - // The directory is allowed to exist - if !os.IsExist(err) { - return nil, errors.Wrapf(err, "error creating lockfiles dir %s", lockDir) - } - } - state.lockDir = lockDir - - // Open the database - // Use loc=auto to get accurate locales for timestamps - db, err := sql.Open("sqlite3", dbPath+"?_loc=auto") - if err != nil { - return nil, errors.Wrapf(err, "error opening database") - } - - // Ensure connectivity - if err := db.Ping(); err != nil { - return nil, errors.Wrapf(err, "cannot establish connection to database") - } - - // Prepare database - if err := prepareDB(db); err != nil { - return nil, err - } - - // Ensure that the database matches our config - if err := checkDB(db, runtime); err != nil { - return nil, err - } - - state.db = db - - state.valid = true - - return state, nil -} - -// Close the state's database connection -func (s *SQLState) Close() error { - if !s.valid { - return ErrDBClosed - } - - s.valid = false - - if err := s.db.Close(); err != nil { - return errors.Wrapf(err, "error closing database") - } - - return nil -} - -// Refresh clears the state after a reboot -// Resets mountpoint, PID, state, netns path for all containers -func (s *SQLState) Refresh() (err error) { - const refresh = `UPDATE containerState SET - State=?, - Mountpoint=?, - Pid=?, - NetNSPath=?, - ExecSessions=?, - IPs=?, - Routes=?, - BindMounts=?;` - - if !s.valid { - return ErrDBClosed - } - - committed := false - - tx, err := s.db.Begin() - if err != nil { - return errors.Wrapf(err, "error beginning database transaction") - } - defer func() { - if err != nil && !committed { - if err2 := tx.Rollback(); err2 != nil { - logrus.Errorf("Error rolling back transaction to refresh state: %v", err2) - } - } - }() - - // Refresh container state - // The constants could be moved into the SQL, but keeping them here - // will keep us in sync in case ContainerStateConfigured ever changes in - // the container state - _, err = tx.Exec(refresh, - ContainerStateConfigured, - "", - 0, - "", - "{}", - "[]", - "[]", - "{}") - if err != nil { - return errors.Wrapf(err, "error refreshing database state") - } - - committed = true - - if err := tx.Commit(); err != nil { - return errors.Wrapf(err, "error committing transaction to refresh database") - } - - return nil -} - -// Container retrieves a container from its full ID -func (s *SQLState) Container(id string) (*Container, error) { - const query = containerQuery + "WHERE containers.Id=?;" - - if id == "" { - return nil, ErrEmptyID - } - - if !s.valid { - return nil, ErrDBClosed - } - - row := s.db.QueryRow(query, id) - - ctr, err := s.ctrFromScannable(row) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving container %s from database", id) - } - - return ctr, nil -} - -// LookupContainer retrieves a container by full or unique partial ID or name -func (s *SQLState) LookupContainer(idOrName string) (*Container, error) { - const query = containerQuery + "WHERE (containers.Id LIKE ?) OR containers.Name=?;" - - if idOrName == "" { - return nil, ErrEmptyID - } - - if !s.valid { - return nil, ErrDBClosed - } - - rows, err := s.db.Query(query, idOrName+"%", idOrName) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving container %s row from database", idOrName) - } - defer rows.Close() - - foundResult := false - var ctr *Container - for rows.Next() { - if foundResult { - return nil, errors.Wrapf(ErrCtrExists, "more than one result for ID or name %s", idOrName) - } - - var err error - ctr, err = s.ctrFromScannable(rows) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving container %s from database", idOrName) - } - foundResult = true - } - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving rows for container ID or name %s", idOrName) - } - - if !foundResult { - return nil, errors.Wrapf(ErrNoSuchCtr, "no container with ID or name %s found", idOrName) - } - - return ctr, nil -} - -// HasContainer checks if the given container is present in the state -// It accepts a full ID -func (s *SQLState) HasContainer(id string) (bool, error) { - const query = "SELECT 1 FROM containers WHERE Id=?;" - - if id == "" { - return false, ErrEmptyID - } - - if !s.valid { - return false, ErrDBClosed - } - - row := s.db.QueryRow(query, id) - - var check int - err := row.Scan(&check) - if err != nil { - if err == sql.ErrNoRows { - return false, nil - } - - return false, errors.Wrapf(err, "error questing database for existence of container %s", id) - } else if check != 1 { - return false, errors.Wrapf(ErrInternal, "check digit for HasContainer query incorrect") - } - - return true, nil -} - -// AddContainer adds the given container to the state -// If the container belongs to a pod, that pod must already be present in the -// state, and the container will be added to the pod -func (s *SQLState) AddContainer(ctr *Container) (err error) { - if !ctr.valid { - return ErrCtrRemoved - } - - if ctr.config.Pod != "" { - return errors.Wrapf(ErrPodExists, "cannot add container that belongs to a pod, use AddContainerToPod instead") - } - - return s.addContainer(ctr, nil) -} - -// UpdateContainer updates a container's state from the database -func (s *SQLState) UpdateContainer(ctr *Container) error { - const query = `SELECT State, - ConfigPath, - RunDir, - Mountpoint, - StartedTime, - FinishedTime, - ExitCode, - OomKilled, - Pid, - NetNSPath, - ExecSessions, - IPs, - Routes, - BindMounts - FROM containerState WHERE ID=?;` - - var ( - state int - configPath string - runDir string - mountpoint string - startedTimeString string - finishedTimeString string - exitCode int32 - oomKilled int - pid int - netNSPath string - execSessions string - ipsJSON string - routesJSON string - bindMountsJSON string - ) - - if !s.valid { - return ErrDBClosed - } - - if !ctr.valid { - return ErrCtrRemoved - } - - row := s.db.QueryRow(query, ctr.ID()) - err := row.Scan( - &state, - &configPath, - &runDir, - &mountpoint, - &startedTimeString, - &finishedTimeString, - &exitCode, - &oomKilled, - &pid, - &netNSPath, - &execSessions, - &ipsJSON, - &routesJSON, - &bindMountsJSON) - if err != nil { - // The container may not exist in the database - if err == sql.ErrNoRows { - // Assume that the container was removed by another process - // As such make it invalid - ctr.valid = false - - return errors.Wrapf(ErrNoSuchCtr, "no container with ID %s found in database", ctr.ID()) - } - - return errors.Wrapf(err, "error parsing database state for container %s", ctr.ID()) - } - - newState := new(containerState) - newState.State = ContainerStatus(state) - newState.ConfigPath = configPath - newState.RunDir = runDir - newState.Mountpoint = mountpoint - newState.ExitCode = exitCode - newState.OOMKilled = boolFromSQL(oomKilled) - newState.PID = pid - - newState.ExecSessions = make(map[string]*ExecSession) - if err := json.Unmarshal([]byte(execSessions), &newState.ExecSessions); err != nil { - return errors.Wrapf(err, "error parsing container %s exec sessions", ctr.ID()) - } - - ips := []*cnitypes.IPConfig{} - if err := json.Unmarshal([]byte(ipsJSON), &ips); err != nil { - return errors.Wrapf(err, "error parsing container %s IPs JSON", ctr.ID()) - } - if len(ips) > 0 { - newState.IPs = ips - } - - routes := []*types.Route{} - if err := json.Unmarshal([]byte(routesJSON), &routes); err != nil { - return errors.Wrapf(err, "error parsing container %s routes JSON", ctr.ID()) - } - if len(routes) > 0 { - newState.Routes = routes - } - - bindMounts := make(map[string]string) - if err := json.Unmarshal([]byte(bindMountsJSON), &bindMounts); err != nil { - return errors.Wrapf(err, "error parsing container %s bind mounts JSON", ctr.ID()) - } - newState.BindMounts = bindMounts - - if newState.Mountpoint != "" { - newState.Mounted = true - } - - startedTime, err := timeFromSQL(startedTimeString) - if err != nil { - return errors.Wrapf(err, "error parsing container %s started time", ctr.ID()) - } - newState.StartedTime = startedTime - - finishedTime, err := timeFromSQL(finishedTimeString) - if err != nil { - return errors.Wrapf(err, "error parsing container %s finished time", ctr.ID()) - } - newState.FinishedTime = finishedTime - - // Do we need to replace the container's netns? - if netNSPath != "" { - // Check if the container's old state has a good netns - if ctr.state.NetNS != nil && netNSPath == ctr.state.NetNS.Path() { - newState.NetNS = ctr.state.NetNS - } else { - // Tear down the existing namespace - if err := s.runtime.teardownNetNS(ctr); err != nil { - return err - } - - // Open the new network namespace - ns, err := joinNetNS(netNSPath) - if err != nil { - return errors.Wrapf(err, "error joining network namespace for container %s", ctr.ID()) - } - newState.NetNS = ns - } - } else { - // The container no longer has a network namespace - // Tear down the old one - if err := s.runtime.teardownNetNS(ctr); err != nil { - return err - } - } - - // New state compiled successfully, swap it into the current state - ctr.state = newState - - return nil -} - -// SaveContainer updates a container's state in the database -func (s *SQLState) SaveContainer(ctr *Container) (err error) { - const update = `UPDATE containerState SET - State=?, - ConfigPath=?, - RunDir=?, - Mountpoint=?, - StartedTime=?, - FinishedTime=?, - ExitCode=?, - OomKilled=?, - Pid=?, - NetNSPath=?, - ExecSessions=?, - IPs=?, - Routes=?, - BindMounts=? - WHERE Id=?;` - - if !ctr.valid { - return ErrCtrRemoved - } - - execSessionsJSON, err := json.Marshal(ctr.state.ExecSessions) - if err != nil { - return errors.Wrapf(err, "error marshalling container %s exec sessions", ctr.ID()) - } - - netNSPath := "" - if ctr.state.NetNS != nil { - netNSPath = ctr.state.NetNS.Path() - } - - ipsJSON, err := json.Marshal(ctr.state.IPs) - if err != nil { - return errors.Wrapf(err, "error marshalling container %s IPs to JSON", ctr.ID()) - } - - routesJSON, err := json.Marshal(ctr.state.Routes) - if err != nil { - return errors.Wrapf(err, "error marshalling container %s routes to JSON", ctr.ID()) - } - - bindMountsJSON, err := json.Marshal(ctr.state.BindMounts) - if err != nil { - return errors.Wrapf(err, "error marshalling container %s bind mounts to JSON", ctr.ID()) - } - - if !s.valid { - return ErrDBClosed - } - - committed := false - - tx, err := s.db.Begin() - if err != nil { - return errors.Wrapf(err, "error beginning database transaction") - } - defer func() { - if err != nil && !committed { - if err2 := tx.Rollback(); err2 != nil { - logrus.Errorf("Error rolling back transaction to add container %s: %v", ctr.ID(), err2) - } - } - }() - - // Add container state to the database - result, err := tx.Exec(update, - ctr.state.State, - ctr.state.ConfigPath, - ctr.state.RunDir, - ctr.state.Mountpoint, - timeToSQL(ctr.state.StartedTime), - timeToSQL(ctr.state.FinishedTime), - ctr.state.ExitCode, - boolToSQL(ctr.state.OOMKilled), - ctr.state.PID, - netNSPath, - execSessionsJSON, - ipsJSON, - routesJSON, - bindMountsJSON, - ctr.ID()) - if err != nil { - return errors.Wrapf(err, "error updating container %s state in database", ctr.ID()) - } - rows, err := result.RowsAffected() - if err != nil { - return errors.Wrapf(err, "error retrieving number of rows modified by update of container %s", ctr.ID()) - } - if rows == 0 { - // Container was probably removed elsewhere - ctr.valid = false - return ErrNoSuchCtr - } - - committed = true - - if err := tx.Commit(); err != nil { - return errors.Wrapf(err, "error committing transaction to update container %s", ctr.ID()) - } - - return nil -} - -// ContainerInUse checks if other containers depend on the given container -// It returns the IDs of containers which depend on the given container -func (s *SQLState) ContainerInUse(ctr *Container) ([]string, error) { - const inUseQuery = `SELECT Id FROM containers WHERE - IPCNsCtr=? OR - MountNsCtr=? OR - NetNsCtr=? OR - PIDNsCtr=? OR - UserNsCtr=? OR - UTSNsCtr=? OR - CgroupNsCtr=?;` - - if !s.valid { - return nil, ErrDBClosed - } - - if !ctr.valid { - return nil, ErrCtrRemoved - } - - id := ctr.ID() - - rows, err := s.db.Query(inUseQuery, id, id, id, id, id, id, id) - if err != nil { - return nil, errors.Wrapf(err, "error querying database for containers that depend on container %s", id) - } - defer rows.Close() - - ids := []string{} - - for rows.Next() { - var ctrID string - if err := rows.Scan(&ctrID); err != nil { - return nil, errors.Wrapf(err, "error scanning container IDs from db rows for container %s", id) - } - - ids = append(ids, ctrID) - } - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving rows for container %s", id) - } - - return ids, nil -} - -// RemoveContainer removes the given container from the state -func (s *SQLState) RemoveContainer(ctr *Container) error { - if ctr.config.Pod != "" { - return errors.Wrapf(ErrPodExists, "container %s belongs to a pod, use RemoveContainerFromPod", ctr.ID()) - } - - return s.removeContainer(ctr, nil) -} - -// AllContainers retrieves all the containers presently in the state -func (s *SQLState) AllContainers() ([]*Container, error) { - const query = containerQuery + ";" - - if !s.valid { - return nil, ErrDBClosed - } - - rows, err := s.db.Query(query) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving containers from database") - } - defer rows.Close() - - containers := []*Container{} - - for rows.Next() { - ctr, err := s.ctrFromScannable(rows) - if err != nil { - return nil, err - } - - containers = append(containers, ctr) - } - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving container rows") - } - - return containers, nil -} - -// Pod retrieves a pod by its full ID -func (s *SQLState) Pod(id string) (*Pod, error) { - const query = "SELECT * FROM pods WHERE Id=?;" - - if !s.valid { - return nil, ErrDBClosed - } - - if id == "" { - return nil, ErrEmptyID - } - - row := s.db.QueryRow(query, id) - - pod, err := s.podFromScannable(row) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving pod %s from database", id) - } - - return pod, nil -} - -// LookupPod retrieves a pot by full or unique partial ID or name -func (s *SQLState) LookupPod(idOrName string) (*Pod, error) { - const query = "SELECT * FROM pods WHERE (Id LIKE ?) OR Name=?;" - - if idOrName == "" { - return nil, ErrEmptyID - } - - if !s.valid { - return nil, ErrDBClosed - } - - rows, err := s.db.Query(query, idOrName+"%", idOrName) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving pod %s row from database", idOrName) - } - defer rows.Close() - - foundResult := false - var pod *Pod - for rows.Next() { - if foundResult { - return nil, errors.Wrapf(ErrCtrExists, "more than one result for ID or name %s", idOrName) - } - - var err error - pod, err = s.podFromScannable(rows) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving pod %s from database", idOrName) - } - foundResult = true - } - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving rows for pod ID or name %s", idOrName) - } - - if !foundResult { - return nil, errors.Wrapf(ErrNoSuchCtr, "no pod with ID or name %s found", idOrName) - } - - return pod, nil -} - -// HasPod checks if a pod exists given its full ID -func (s *SQLState) HasPod(id string) (bool, error) { - if id == "" { - return false, ErrEmptyID - } - - if !s.valid { - return false, ErrDBClosed - } - - return s.podExists(id) -} - -// PodHasContainer checks if the given pod containers a container with the given -// ID -func (s *SQLState) PodHasContainer(pod *Pod, ctrID string) (bool, error) { - const query = "SELECT 1 FROM containers WHERE Id=? AND Pod=?;" - - if ctrID == "" { - return false, ErrEmptyID - } - - if !s.valid { - return false, ErrDBClosed - } - - if !pod.valid { - return false, ErrPodRemoved - } - - row := s.db.QueryRow(query, ctrID, pod.ID()) - - var check int - err := row.Scan(&check) - if err != nil { - if err == sql.ErrNoRows { - return false, nil - } - - return false, errors.Wrapf(err, "error questing database for existence of container %s", ctrID) - } else if check != 1 { - return false, errors.Wrapf(ErrInternal, "check digit for PodHasContainer query incorrect") - } - - return true, nil -} - -// PodContainersByID returns the container IDs of all containers in the given -// pod -func (s *SQLState) PodContainersByID(pod *Pod) ([]string, error) { - const query = "SELECT Id FROM containers WHERE Pod=?;" - - if !s.valid { - return nil, ErrDBClosed - } - - if !pod.valid { - return nil, ErrPodRemoved - } - - // Check to make sure pod still exists in DB - exists, err := s.podExists(pod.ID()) - if err != nil { - return nil, err - } - if !exists { - pod.valid = false - return nil, ErrPodRemoved - } - - // Get actual containers - rows, err := s.db.Query(query, pod.ID()) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving containers from database") - } - defer rows.Close() - - containers := []string{} - - for rows.Next() { - var id string - if err := rows.Scan(&id); err != nil { - if err == sql.ErrNoRows { - return nil, ErrNoSuchCtr - } - - return nil, errors.Wrapf(err, "error parsing database row into container ID") - } - - containers = append(containers, id) - } - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving container rows") - } - - return containers, nil -} - -// PodContainers returns all the containers in a pod given the pod's full ID -func (s *SQLState) PodContainers(pod *Pod) ([]*Container, error) { - const query = containerQuery + "WHERE containers.Pod=?;" - - if !s.valid { - return nil, ErrDBClosed - } - - if !pod.valid { - return nil, ErrPodRemoved - } - - // Check to make sure pod still exists in DB - exists, err := s.podExists(pod.ID()) - if err != nil { - return nil, err - } - if !exists { - pod.valid = false - return nil, ErrPodRemoved - } - - // Get actual containers - rows, err := s.db.Query(query, pod.ID()) - if err != nil { - return nil, errors.Wrapf(err, "error retrieving containers from database") - } - defer rows.Close() - - containers := []*Container{} - - for rows.Next() { - ctr, err := s.ctrFromScannable(rows) - if err != nil { - return nil, err - } - - containers = append(containers, ctr) - } - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving container rows") - } - - return containers, nil -} - -// AddPod adds a pod to the state -// Only empty pods can be added to the state -func (s *SQLState) AddPod(pod *Pod) (err error) { - const ( - podQuery = "INSERT INTO pods VALUES (?, ?, ?);" - registryQuery = "INSERT INTO registry VALUES (?, ?);" - ) - - if !s.valid { - return ErrDBClosed - } - - if !pod.valid { - return ErrPodRemoved - } - - labelsJSON, err := json.Marshal(pod.config.Labels) - if err != nil { - return errors.Wrapf(err, "error marshaling pod %s labels to JSON", pod.ID()) - } - - committed := false - - tx, err := s.db.Begin() - if err != nil { - return errors.Wrapf(err, "error beginning database transaction") - } - defer func() { - if err != nil && !committed { - if err2 := tx.Rollback(); err2 != nil { - logrus.Errorf("Error rolling back transaction to add pod %s: %v", pod.ID(), err2) - } - } - }() - - if _, err := tx.Exec(registryQuery, pod.ID(), pod.Name()); err != nil { - return errors.Wrapf(err, "error adding pod %s to name/ID registry", pod.ID()) - } - - if _, err = tx.Exec(podQuery, pod.ID(), pod.Name(), string(labelsJSON)); err != nil { - return errors.Wrapf(err, "error adding pod %s to database", pod.ID()) - } - - committed = true - - if err := tx.Commit(); err != nil { - return errors.Wrapf(err, "error committing transaction to add pod %s", pod.ID()) - } - - return nil -} - -// RemovePod removes a pod from the state -// Only empty pods can be removed -func (s *SQLState) RemovePod(pod *Pod) (err error) { - const ( - removePod = "DELETE FROM pods WHERE ID=?;" - removeRegistry = "DELETE FROM registry WHERE Id=?;" - ) - - if !s.valid { - return ErrDBClosed - } - - committed := false - - tx, err := s.db.Begin() - if err != nil { - return errors.Wrapf(err, "error beginning database transaction") - } - defer func() { - if err != nil && !committed { - if err2 := tx.Rollback(); err2 != nil { - logrus.Errorf("Error rolling back transaction to remove pod %s: %v", pod.ID(), err2) - } - } - }() - - // Check rows acted on for the first statement, verify we actually removed something - result, err := tx.Exec(removePod, pod.ID()) - if err != nil { - return errors.Wrapf(err, "error removing pod %s from pods table", pod.ID()) - } - rows, err := result.RowsAffected() - if err != nil { - return errors.Wrapf(err, "error retrieving number of rows in transaction removing pod %s", pod.ID()) - } else if rows == 0 { - pod.valid = false - return ErrNoSuchPod - } - - // We know it exists, remove it from registry - if _, err := tx.Exec(removeRegistry, pod.ID()); err != nil { - return errors.Wrapf(err, "error removing pod %s from name/ID registry", pod.ID()) - } - - committed = true - - if err := tx.Commit(); err != nil { - return errors.Wrapf(err, "error committing transaction to remove pod %s", pod.ID()) - } - - return nil -} - -// RemovePodContainers removes all containers in a pod simultaneously -// This can avoid issues with dependencies within the pod -// The operation will fail if any container in the pod has a dependency from -// outside the pod -func (s *SQLState) RemovePodContainers(pod *Pod) (err error) { - const ( - getPodCtrs = "SELECT Id FROM containers WHERE pod=?;" - removeCtr = "DELETE FROM containers WHERE pod=?;" - removeCtrState = "DELETE FROM containerState WHERE ID IN (SELECT Id FROM containers WHERE pod=?);" - ) - - if !s.valid { - return ErrDBClosed - } - - if !pod.valid { - return ErrPodRemoved - } - - committed := false - - tx, err := s.db.Begin() - if err != nil { - return errors.Wrapf(err, "error beginning database transaction") - } - defer func() { - if err != nil && !committed { - if err2 := tx.Rollback(); err2 != nil { - logrus.Errorf("Error rolling back transaction to remove pod %s containers: %v", pod.ID(), err2) - } - } - }() - - // Check if the pod exists - exists, err := podExistsTx(pod.ID(), tx) - if err != nil { - return err - } - if !exists { - pod.valid = false - return ErrNoSuchPod - } - - // First get all containers in the pod - rows, err := tx.Query(getPodCtrs, pod.ID()) - if err != nil { - return errors.Wrapf(err, "error retrieving containers from database") - } - defer rows.Close() - - containers := []string{} - - for rows.Next() { - var id string - if err := rows.Scan(&id); err != nil { - if err == sql.ErrNoRows { - return ErrNoSuchCtr - } - - return errors.Wrapf(err, "error parsing database row into container ID") - } - - containers = append(containers, id) - } - if err := rows.Err(); err != nil { - return errors.Wrapf(err, "error retrieving container rows") - } - - // Have container IDs, now exec SQL to remove containers in the pod - // Remove state first, as it needs the subquery on containers - // Don't bother checking if we actually removed anything, we just want to - // empty the pod - if _, err := tx.Exec(removeCtrState, pod.ID()); err != nil { - return errors.Wrapf(err, "error removing pod %s containers from state table", pod.ID()) - } - - if _, err := tx.Exec(removeCtr, pod.ID()); err != nil { - return errors.Wrapf(err, "error removing pod %s containers from containers table", pod.ID()) - } - - committed = true - - if err := tx.Commit(); err != nil { - return errors.Wrapf(err, "error committing transaction remove pod %s containers", pod.ID()) - } - - // Remove JSON files from the containers in question - hasError := false - for _, ctr := range containers { - jsonPath := getSpecPath(s.specsDir, ctr) - if err := os.Remove(jsonPath); err != nil { - logrus.Errorf("Error removing spec JSON for container %s: %v", ctr, err) - hasError = true - } - - portsPath := getPortsPath(s.specsDir, ctr) - if err := os.Remove(portsPath); err != nil { - if !os.IsNotExist(err) { - logrus.Errorf("Error removing ports JSON for container %s: %v", ctr, err) - hasError = true - } - } - } - if hasError { - return errors.Wrapf(ErrInternal, "error removing JSON state for some containers") - } - - return nil -} - -// AddContainerToPod adds a container to the given pod -func (s *SQLState) AddContainerToPod(pod *Pod, ctr *Container) error { - if !pod.valid { - return ErrPodRemoved - } - - if !ctr.valid { - return ErrCtrRemoved - } - - if ctr.config.Pod != pod.ID() { - return errors.Wrapf(ErrInvalidArg, "container's pod ID does not match given pod's ID") - } - - return s.addContainer(ctr, pod) -} - -// RemoveContainerFromPod removes a container from the given pod -func (s *SQLState) RemoveContainerFromPod(pod *Pod, ctr *Container) error { - if ctr.config.Pod != pod.ID() { - return errors.Wrapf(ErrInvalidArg, "container %s is not in pod %s", ctr.ID(), pod.ID()) - } - - return s.removeContainer(ctr, pod) -} - -// AllPods retrieves all pods presently in the state -func (s *SQLState) AllPods() ([]*Pod, error) { - const query = "SELECT * FROM pods;" - - if !s.valid { - return nil, ErrDBClosed - } - - rows, err := s.db.Query(query) - if err != nil { - return nil, errors.Wrapf(err, "error querying database for all pods") - } - defer rows.Close() - - pods := []*Pod{} - - for rows.Next() { - pod, err := s.podFromScannable(rows) - if err != nil { - return nil, err - } - - pods = append(pods, pod) - } - - if err := rows.Err(); err != nil { - return nil, errors.Wrapf(err, "error retrieving pod rows") - } - - return pods, nil -} |