3177N/A#!/usr/bin/python2.7
1448N/A#
1448N/A# CDDL HEADER START
1448N/A#
1448N/A# The contents of this file are subject to the terms of the
1448N/A# Common Development and Distribution License (the "License").
1448N/A# You may not use this file except in compliance with the License.
1448N/A#
1448N/A# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
1448N/A# or http://www.opensolaris.org/os/licensing.
1448N/A# See the License for the specific language governing permissions
1448N/A# and limitations under the License.
1448N/A#
1448N/A# When distributing Covered Code, include this CDDL HEADER in each
1448N/A# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
1448N/A# If applicable, add the following below this CDDL HEADER, with the
1448N/A# fields enclosed by brackets "[]" replaced with your own identifying
1448N/A# information: Portions Copyright [yyyy] [name of copyright owner]
1448N/A#
1448N/A# CDDL HEADER END
1448N/A#
1448N/A
1448N/A#
3359N/A# Copyright (c) 2009, 2016, Oracle and/or its affiliates. All rights reserved.
1448N/A#
1448N/A
3143N/Afrom __future__ import print_function
1448N/Aimport gettext
1448N/Aimport locale
1448N/Aimport sys
1448N/A
1448N/Aimport depot
1448N/Aimport stats
1448N/Aimport pkg.misc as misc
1448N/A
1448N/Aclass Scenario(object):
1448N/A """A Scenario has a list of depot.RepositoryURIs."""
1448N/A
1448N/A def __init__(self, title):
1448N/A self.title = title
1448N/A self.repo_uris = []
1448N/A self.origin_uris = []
1448N/A
1448N/A def get_repo_uris(self):
1448N/A return self.repo_uris
1448N/A
1448N/A def add_repo_uri(self, label, speed, cspeed,
1448N/A error_rate=depot.ERROR_FREE, error_type=depot.ERROR_T_NET,
1448N/A speed_distribution=depot.MODAL_SINGLE):
1448N/A
1448N/A r = depot.RepositoryURI(label, speed, cspeed,
1448N/A error_rate=error_rate, error_type=error_type,
1448N/A modality=speed_distribution)
1448N/A
1448N/A self.repo_uris.append(r)
1448N/A
1448N/A def add_origin_uri(self, label, speed, cspeed,
1448N/A error_rate=depot.ERROR_FREE, error_type=depot.ERROR_T_NET,
1448N/A speed_distribution=depot.MODAL_SINGLE):
1448N/A
1448N/A r = depot.RepositoryURI(label, speed, cspeed,
1448N/A error_rate=error_rate, error_type=error_type,
1448N/A modality=speed_distribution)
1448N/A
1448N/A self.origin_uris.append(r)
1448N/A
1448N/A def get_megabytes(self):
1448N/A return self.__total_mb
1448N/A
1448N/A def set_megabytes(self, mb):
1448N/A self.__total_mb = mb
1448N/A
1448N/A def run(self):
3158N/A print("SCENARIO: {0}".format(self.title))
1448N/A
1448N/A total = self.__total_mb * 1024 * 1024
1448N/A
1448N/A rc = stats.RepoChooser()
1448N/A urilist = self.repo_uris[:]
1448N/A urilist.extend(self.origin_uris)
1448N/A
1448N/A while total > 0:
1448N/A s = rc.get_repostats(urilist, self.origin_uris)
1448N/A
1448N/A n = len(s)
1448N/A m = rc.get_num_visited(urilist)
1448N/A
1448N/A if m < n:
1448N/A c = 10
1448N/A else:
1448N/A c = 100
1448N/A
3158N/A print("bytes left {0:d}; retrieving {1:d} files".format(
3158N/A total, c))
1448N/A rc.dump()
1448N/A
1448N/A # get 100 files
1448N/A r = s[0]
1448N/A for n in range(c):
1448N/A req = r[1].request(rc)
1448N/A total -= req[1]
1448N/A
1448N/A rc.dump()
1448N/A
1448N/Amisc.setlocale(locale.LC_ALL)
1448N/Agettext.install("pkg", "/usr/share/locale")
1448N/A
3158N/Atotal_mb = 1000
1448N/A
1448N/A# Scenario 1. A single origin.
1448N/A
1448N/Asingle = Scenario("single origin")
1448N/A
1448N/Asingle.add_origin_uri("origin", depot.SPEED_FAST, depot.CSPEED_LAN)
1448N/A
1448N/Asingle.set_megabytes(total_mb)
1448N/A
1448N/Asingle.run()
1448N/A
1448N/A# Scenario 2a. An origin and mirror, mirror faster.
1448N/A
1448N/Aone_mirror = Scenario("origin and a faster mirror")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_FAST, depot.CSPEED_NEARBY)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_VERY_FAST, depot.CSPEED_LAN)
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
1448N/A
1448N/A# Scenario 2b. An origin and mirror, origin faster.
1448N/A
1448N/Aone_mirror = Scenario("origin and a slower mirror")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_VERY_FAST, depot.CSPEED_LAN)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_FAST, depot.CSPEED_NEARBY)
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
1448N/A
1448N/A# Scenario 2c. An origin and mirror, mirror faster, but decaying.
1448N/A
1448N/Aone_mirror = Scenario("origin and a faster, but decaying, mirror")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_FAST, depot.CSPEED_NEARBY)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_VERY_FAST, depot.CSPEED_LAN,
1448N/A speed_distribution=depot.MODAL_DECAY)
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
1448N/A
1448N/A# Scenario 2d. An origin and mirror, mirror slower, but increasing.
1448N/A
1448N/Aone_mirror = Scenario("origin and a slower, but increasing, mirror")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_MEDIUM, depot.CSPEED_MEDIUM)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_FAST, depot.CSPEED_LAN,
1448N/A speed_distribution=depot.MODAL_INCREASING)
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
1448N/A
1448N/A# Scenario 2e. An origin and mirror, mirror encountering decyable transport
1448N/A# errors.
1448N/A
1448N/Aone_mirror = Scenario("origin and a faster mirror. Mirror gets decayable errors")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_FAST, depot.CSPEED_LAN)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_SLIGHTLY_FASTER, depot.CSPEED_LAN,
1448N/A error_rate=depot.ERROR_LOW, error_type=depot.ERROR_T_DECAYABLE)
1448N/A
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
1448N/A
1448N/A# Scenario 3a. An origin and two mirrors, one decaying.
1448N/A
1448N/Aone_mirror = Scenario("origin and two mirrors")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_FAST, depot.CSPEED_NEARBY)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_SLIGHTLY_FASTER,
1448N/A depot.CSPEED_LAN)
1448N/Aone_mirror.add_repo_uri("mirror2", depot.SPEED_VERY_FAST, depot.CSPEED_NEARBY,
1448N/A speed_distribution=depot.MODAL_DECAY)
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
1448N/A
1448N/A# Scenario 3b. An origin and five mirrors.
1448N/A
1448N/Aone_mirror = Scenario("origin and five mirrors")
1448N/A
1448N/Aone_mirror.add_origin_uri("origin", depot.SPEED_MODERATE, depot.CSPEED_MEDIUM)
1448N/Aone_mirror.add_repo_uri("mirror", depot.SPEED_SLIGHTLY_FASTER, depot.CSPEED_LAN)
1448N/Aone_mirror.add_repo_uri("mirror2", depot.SPEED_MEDIUM, depot.CSPEED_SLOW)
1448N/Aone_mirror.add_repo_uri("mirror3", depot.SPEED_SLOW, depot.CSPEED_SLOW)
1448N/Aone_mirror.add_repo_uri("mirror4", depot.SPEED_VERY_SLOW,
1448N/A depot.CSPEED_VERY_SLOW)
1448N/Aone_mirror.add_repo_uri("mirror5", depot.SPEED_SLOW, depot.CSPEED_FARAWAY)
1448N/A
1448N/Aone_mirror.set_megabytes(total_mb)
1448N/A
1448N/Aone_mirror.run()
3359N/A
3359N/A# Scenario 4. Six origins.
3359N/A
3359N/Asix_origin = Scenario("six origins")
3359N/A
3359N/Asix_origin.add_origin_uri("origin1", depot.SPEED_VERY_SLOW,
3359N/A depot.CSPEED_VERY_SLOW)
3359N/Asix_origin.add_origin_uri("origin2", depot.SPEED_SLOW, depot.CSPEED_FARAWAY)
3359N/Asix_origin.add_origin_uri("origin3", depot.SPEED_MODERATE, depot.CSPEED_MEDIUM)
3359N/Asix_origin.add_origin_uri("origin4", depot.SPEED_SLIGHTLY_FASTER, depot.CSPEED_LAN)
3359N/Asix_origin.add_origin_uri("origin5", depot.SPEED_MEDIUM, depot.CSPEED_SLOW)
3359N/Asix_origin.add_origin_uri("origin6", depot.SPEED_FAST, depot.CSPEED_MEDIUM)
3359N/A
3359N/Asix_origin.set_megabytes(total_mb)
3359N/A
3359N/Asix_origin.run()