EnsureMongoServer function for MachineAgent

agent/agent.go

 // Copyright 2013 Canonical Ltd.
 // Licensed under the AGPLv3, see LICENCE file for details.
 
 package agent
 
 import (
         "fmt"
         "path"
         "regexp"
         "sync"
@@ skipping 76 matching lines @@
         // APIServerDetails returns the details needed to run an API server.
         APIServerDetails() (port int, cert, key []byte)
 
         // Value returns the value associated with the key, or an empty string if
         // the key is not found.
         Value(key string) string
 
         // SetValue updates the value for the specified key.
         SetValue(key, value string)
 
+        // StatePort specifies the TCP port that will be used
+        // by the MongoDB server if one is running on this machine.
+        StatePort() int
+
         StateInitializer
 }
 
 // Ensure that the configInternal struct implements the Config interface.
 var _ Config = (*configInternal)(nil)
 
 // The configMutex should be locked before any writing to disk during the
 // write commands, and unlocked when the writing is complete.  This process
 // wide lock should stop any unintended concurrent writes.  This may happen
 // when multiple go-routines may be adding things to the agent config, and
@@ skipping 18 matching lines @@
         tag             string
         nonce           string
         caCert          []byte
         stateDetails    *connectionDetails
         apiDetails      *connectionDetails
         oldPassword     string
         stateServerCert []byte
         stateServerKey  []byte
         apiPort         int
         values          map[string]string
+        statePort       int
 }
 
 type AgentConfigParams struct {
         DataDir        string
         Tag            string
         Password       string
         Nonce          string
         StateAddresses []string
         APIAddresses   []string
         CACert         []byte
@@ skipping 62 matching lines @@
                 return nil, requiredError("state server key")
         }
         config0, err := NewAgentConfig(params.AgentConfigParams)
         if err != nil {
                 return nil, err
         }
         config := config0.(*configInternal)
         config.stateServerCert = params.StateServerCert
         config.stateServerKey = params.StateServerKey
         config.apiPort = params.APIPort
+        config.statePort = params.StatePort
         return config, nil
 }
 
 // Dir returns the agent-specific data directory.
 func Dir(dataDir, agentName string) string {
         return path.Join(dataDir, "agents", agentName)
 }
 
 // ReadConf reads configuration data for the given
 // entity from the given data directory.
@@ skipping 86 matching lines @@
 }
 
 func (c *configInternal) Tag() string {
         return c.tag
 }
 
 func (c *configInternal) Dir() string {
         return Dir(c.dataDir, c.tag)
 }
 
+func (c *configInternal) StatePort() int {
+        return c.statePort
+}
+
 func (c *configInternal) check() error {
         if c.stateDetails == nil && c.apiDetails == nil {
                 return requiredError("state or API addresses")
         }
         if c.stateDetails != nil {
                 if err := checkAddrs(c.stateDetails.addresses, "state server address"); err != nil {
                         return err
                 }
         }
         if c.apiDetails != nil {
@@ skipping 109 matching lines @@
                 }
                 // TODO(rog) remove this fallback behaviour when
                 // all initial connections are via the API.
                 if !errors.IsUnauthorizedError(err) {
                         return nil, err
                 }
         }
         info.Password = c.oldPassword
         return state.Open(&info, state.DefaultDialOpts())
 }