bosh-director 1.5.0.pre.1113

data/lib/bosh/director/models/compiled_package.rb

@@ -0,0 +1,33 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class CompiledPackage < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :package
+    many_to_one :stemcell
+
+    def validate
+      validates_presence [:package_id, :stemcell_id, :sha1,
+                          :blobstore_id, :dependency_key]
+      validates_unique [:package_id, :stemcell_id, :dependency_key]
+      validates_unique [:package_id, :stemcell_id, :build]
+    end
+
+    def name
+      package.name
+    end
+
+    def version
+      package.version
+    end
+
+    def self.generate_build_number(package, stemcell)
+      attrs = {
+        :package_id => package.id,
+        :stemcell_id => stemcell.id
+      }
+
+      filter(attrs).max(:build).to_i + 1
+    end
+
+  end
+end

data/lib/bosh/director/models/deployment.rb

@@ -0,0 +1,22 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Deployment < Sequel::Model(Bosh::Director::Config.db)
+    many_to_many :stemcells
+    many_to_many :releases
+    many_to_many :release_versions
+    one_to_many  :job_instances, :class => "Bosh::Director::Models::Instance"
+    one_to_many  :vms
+    one_to_many  :properties, :class => "Bosh::Director::Models::DeploymentProperty"
+    one_to_many  :problems, :class => "Bosh::Director::Models::DeploymentProblem"
+
+    def validate
+      validates_presence :name
+      validates_unique :name
+      validates_format VALID_ID, :name
+    end
+  end
+
+  Deployment.plugin :association_dependencies
+  Deployment.add_association_dependencies :stemcells => :nullify, :problems => :destroy
+end

data/lib/bosh/director/models/deployment_problem.rb

@@ -0,0 +1,49 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class DeploymentProblem < Sequel::Model(Bosh::Director::Config.db)
+
+    many_to_one :deployment
+
+    STATES = %w(open closed resolved)
+
+    def validate
+      validates_presence :deployment_id
+      validates_presence :resource_id
+      validates_presence :type
+      validates_presence :data_json
+      validates_presence :state
+      validates_includes STATES, :state
+    end
+
+    def before_create
+      self.created_at ||= Time.now
+      self.last_seen_at ||= Time.now
+    end
+
+    def data
+      Yajl::Parser.parse(data_json)
+    end
+
+    def data=(raw_data)
+      self.data_json = Yajl::Encoder.encode(raw_data)
+    end
+
+    def handler
+      @handler ||= Bosh::Director::ProblemHandlers::Base.create_from_model(self)
+    end
+
+    def resolutions
+      handler.resolutions
+    end
+
+    def description
+      handler.description
+    end
+
+    def open?
+      state == "open"
+    end
+
+  end
+end

data/lib/bosh/director/models/deployment_property.rb

@@ -0,0 +1,21 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class DeploymentProperty < Sequel::Model(Bosh::Director::Config.db)
+
+    VALID_PROPERTY_NAME = /^[-a-z0-9_.]+$/i
+
+    many_to_one :deployment
+
+    def validate
+      validates_presence :deployment_id
+      validates_presence :name
+      validates_presence :value
+
+      validates_unique [:name, :deployment_id]
+
+      validates_format VALID_PROPERTY_NAME, :name, :allow_blank => true
+    end
+
+  end
+end

data/lib/bosh/director/models/director_attribute.rb

@@ -0,0 +1,9 @@
+module Bosh::Director::Models
+  class DirectorAttribute < Sequel::Model(Bosh::Director::Config.db)
+
+    def validate
+      validates_unique [:uuid]
+    end
+
+  end
+end

data/lib/bosh/director/models/dns.rb

@@ -0,0 +1,9 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+require "bosh/director/models/dns/domain"
+require "bosh/director/models/dns/record"
+
+module Bosh::Director::Models
+  module Dns
+  end
+end

data/lib/bosh/director/models/dns/domain.rb

@@ -0,0 +1,9 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models::Dns
+  class Domain < Sequel::Model(Bosh::Director::Config.dns_db)
+    extend Bosh::Director::ModelHelper
+
+    one_to_many :records
+  end
+end

data/lib/bosh/director/models/dns/record.rb

@@ -0,0 +1,7 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models::Dns
+  class Record < Sequel::Model(Bosh::Director::Config.dns_db)
+    many_to_one :domain
+  end
+end

data/lib/bosh/director/models/helpers/model_helper.rb

@@ -0,0 +1,7 @@
+module Bosh::Director::ModelHelper
+  def safe_find_or_create(options)
+    Bosh::Common.retryable(sleep: 0.1, tries: 2, on: [Exception]) do
+      find_or_create(options)
+    end
+  end
+end

data/lib/bosh/director/models/instance.rb

@@ -0,0 +1,28 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Instance < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :deployment
+    many_to_one :vm
+    one_to_many :persistent_disks
+
+    def validate
+      validates_presence [:deployment_id, :job, :index, :state]
+      validates_unique [:deployment_id, :job, :index]
+      validates_unique [:vm_id] if vm_id
+      validates_integer :index
+      validates_includes ["started", "stopped", "detached"], :state
+    end
+
+    def persistent_disk
+      # Currently we support only 1 persistent disk.
+      self.persistent_disks.find { |disk| disk.active }
+    end
+
+    def persistent_disk_cid
+      disk = persistent_disk
+      return disk.disk_cid if disk
+      nil
+    end
+  end
+end

data/lib/bosh/director/models/log_bundle.rb

@@ -0,0 +1,10 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class LogBundle < Sequel::Model(Bosh::Director::Config.db)
+    def validate
+      validates_presence [:blobstore_id, :timestamp ]
+      validates_unique [ :blobstore_id ]
+    end
+  end
+end

data/lib/bosh/director/models/package.rb

@@ -0,0 +1,30 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Package < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :release
+    many_to_many :release_versions
+    one_to_many :compiled_packages
+
+    # @return [Set<String>] A set of package names this package depends on
+    def dependency_set
+      json = self.dependency_set_json
+
+      ::Set.new(json ? Yajl::Parser.parse(json) : nil)
+    end
+
+    def dependency_set=(deps)
+      self.dependency_set_json = Yajl::Encoder.encode(deps.to_a)
+    end
+
+    def validate
+      validates_presence [:release_id, :name, :version, :blobstore_id, :sha1]
+      validates_unique [:release_id, :name, :version]
+      validates_format VALID_ID, [:name, :version]
+    end
+
+    def desc
+      "#{name}/#{version}"
+    end
+  end
+end

data/lib/bosh/director/models/persistent_disk.rb

@@ -0,0 +1,13 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class PersistentDisk < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :instance
+    one_to_many :snapshots
+
+    def validate
+      validates_presence [:instance_id, :disk_cid]
+      validates_unique [:disk_cid]
+    end
+  end
+end

data/lib/bosh/director/models/release.rb

@@ -0,0 +1,17 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Release < Sequel::Model(Bosh::Director::Config.db)
+    one_to_many :versions, :class => "Bosh::Director::Models::ReleaseVersion"
+    one_to_many :packages
+    one_to_many :templates
+
+    many_to_many :deployments
+
+    def validate
+      validates_presence :name
+      validates_unique :name
+      validates_format VALID_ID, :name
+    end
+  end
+end

data/lib/bosh/director/models/release_version.rb

@@ -0,0 +1,16 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class ReleaseVersion < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one  :release
+    many_to_many :packages
+    many_to_many :templates
+    many_to_many :deployments
+
+    def validate
+      validates_presence [:release_id, :version]
+      validates_unique [:release_id, :version]
+      validates_format VALID_ID, :version
+    end
+  end
+end

data/lib/bosh/director/models/snapshot.rb

@@ -0,0 +1,13 @@
+module Bosh::Director::Models
+  class Snapshot < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :persistent_disk
+
+    def validate
+      validates_unique [:snapshot_cid]
+    end
+
+    def before_create
+      self.created_at ||= Time.now
+    end
+  end
+end

data/lib/bosh/director/models/stemcell.rb

@@ -0,0 +1,18 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Stemcell < Sequel::Model(Bosh::Director::Config.db)
+    many_to_many :deployments
+    one_to_many :compiled_packages
+
+    def validate
+      validates_presence [:name, :version, :cid]
+      validates_unique [:name, :version]
+      validates_format VALID_ID, [:name, :version]
+    end
+
+    def desc
+      "#{name}/#{version}"
+    end
+  end
+end

data/lib/bosh/director/models/task.rb

@@ -0,0 +1,10 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Task < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one  :user
+    def validate
+      validates_presence [:state, :timestamp, :description]
+    end
+  end
+end

data/lib/bosh/director/models/template.rb

@@ -0,0 +1,44 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Template < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :release
+    many_to_many :release_versions
+
+    def package_names
+      result = self.package_names_json
+      result ? Yajl::Parser.parse(result) : nil
+    end
+
+    def package_names=(packages)
+      self.package_names_json = Yajl::Encoder.encode(packages)
+    end
+
+    def logs=(logs_spec)
+      self.logs_json = Yajl::Encoder.encode(logs_spec)
+    end
+
+    def logs
+      result = self.logs_json
+      result ? Yajl::Parser.parse(result) : nil
+    end
+
+    # @param [Object] property_spec Property spec from job spec
+    def properties=(property_spec)
+      self.properties_json = Yajl::Encoder.encode(property_spec)
+    end
+
+    # @return [Hash] Template properties (as provided in job spec)
+    # @return [nil] if no properties have been defined in job spec
+    def properties
+      result = self.properties_json
+      result ? Yajl::Parser.parse(result) : nil
+    end
+
+    def validate
+      validates_presence [:release_id, :name, :version, :blobstore_id, :sha1]
+      validates_unique [:release_id, :name, :version]
+      validates_format VALID_ID, [:name, :version]
+    end
+  end
+end

data/lib/bosh/director/models/user.rb

@@ -0,0 +1,11 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class User < Sequel::Model(Bosh::Director::Config.db)
+    def validate
+      validates_presence :username
+      validates_presence :password
+      validates_unique :username
+    end
+  end
+end

data/lib/bosh/director/models/vm.rb

@@ -0,0 +1,42 @@
+# Copyright (c) 2009-2012 VMware, Inc.
+
+module Bosh::Director::Models
+  class Vm < Sequel::Model(Bosh::Director::Config.db)
+    many_to_one :deployment
+    one_to_one  :instance
+
+    def validate
+      validates_presence [:deployment_id, :agent_id]
+      validates_unique :agent_id
+    end
+
+    def apply_spec
+      return nil if apply_spec_json.nil?
+      Yajl::Parser.parse(apply_spec_json)
+    end
+
+    def apply_spec=(spec)
+      self.apply_spec_json = Yajl::Encoder.encode(spec)
+    end
+
+    # @param [Hash] env_hash Environment hash
+    def env=(env_hash)
+      self.env_json = Yajl::Encoder.encode(env_hash)
+    end
+
+    # @return [Hash] VM environment hash
+    def env
+      return nil if env_json.nil?
+      Yajl::Parser.parse(env_json)
+    end
+
+    def credentials
+      return nil if credentials_json.nil?
+      Yajl::Parser.parse(credentials_json)
+    end
+
+    def credentials=(spec)
+      self.credentials_json = Yajl::Encoder.encode(spec)
+    end
+  end
+end