> SrSSKJr SSKJr SSKJr SSKJr SSKrSSKJrJ r J r SSK J r SS K Jr SS KJr SS KJr SSKrS rS r"SS\R,5rS\ \R0S\\\44SjrS\\\4S\\\4S\\\44SjrS\\\4S\\\44SjrS\\\44SjrS\\\4S\\\44SjrS\R04Sjr Sr!Sr"S \\\44S!jr#S\ \R0S\\\ \\$44S\ \R04S"jr%S\ \R0S#\S\ \R04S$jr&S%r'S\ \R0S\4S&jr("S'S(\)5r*S)\*4S*jr+S\ \ \R0\RX4S#\S\\\ \R0\RX444S+jr-S,\R\S\ \*4S-jr/g).z0Operations on WorkerPool V2 API instance splits.)absolute_import)division)print_function)unicode_literalsN)DictListUnion)resource_name_conversion) exceptions)instance_split) worker_pool-LATESTc\rSrSrSrSrg)&InvalidInstanceSplitSpecificationError)z:Error to indicate an invalid instance split specification.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r7lib/googlecloudsdk/command_lib/run/v2/instance_split.pyrr)sBrrsplitsreturnc0nUHnUR[RR:XaURU[ 'M@UR[RR :XdMjURXR'M U$)z:Returns the current instance split percentages into a map.)type_r InstanceSplitAllocationType%INSTANCE_SPLIT_ALLOCATION_TYPE_LATESTpercentLATEST_REVISION_KEY'INSTANCE_SPLIT_ALLOCATION_TYPE_REVISIONrevision)rcurrent_splitssplits r_GetCurrentSplitsMapr(/sv.e  5 5 [ [ \-2MMn()  5 5 ] ] ^(-}}n^^$ rnew_percentagesr&cR0nUR5Hup4X0;dM XBU'M U$)z`Returns the instance splits that are in the current splits but not specified in new_percentages.)items)r)r&resulttargetr"s r_GetUnspecifiedSplitsr.Bs2 &'--/of $Vn0 -r new_splitsunspecified_targetsc2[UR55S:a [S5eUR5H+up#US:dUS:dM[SR X#55e U(d)[UR55S:a [S5egg)z-Validates the new instance split percentages.dz5The sum of instance split specifications exceeds 100.rzAInstance split specification for {} is {}%, not between 0 and 100z[Every target with instance split is updated but 100% of total split has not been specified.N)sumvaluesrr+format)r/r0r-r"s r_ValidateNewSplitsr6Ns    # 0? $))+of{gm 2 M 6& " , Z%6%6%8!9C!? 0 # "@ rcSnUR5H/up#US:dUS:a[SRX#55eX- nM1 US:wa[SRU55eg)z1Validates the current instance split percentages.rr2zFCurrent instance split allocation for {} is {}%, not between 0 and 100z:Current instance split allocation of {} is not 100 percentN)r+ ValueErrorr5)r& total_percentr-r"s r_ValidateCurrentSplitsr:es|-'--/of{gm  ( M 0c DKK   runspecified_splitscS[UR55- nUS:Xa0$[UR55n[U5U- n0nUR5H upgXt-XV'M U$)zoModifies the unspecified splits by assigning the remaining split percent proportionally to the original splits.r2r)r3r4floatr+)r/r;percent_to_assignoriginal_splits_percentreduction_ratiomodified_splitsr-r"s r_ModifyUnspecifiedSplitsrBws}C 1 1 344! I 2 9 9 ;<+,/FF//+113of%7O4 rr'cUR[RR:Xa[nO UR n[ U5$)zSorted key function to order InstanceSplit objects by key. Args: split: A InstanceSplit. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )typer r r!r#r%_SortKeyFromKey)r'keys r_SortKeyFromInstanceSplitrGs; jj  3 3 Y YZ C ..C  rc.U[:XaSU4nU$SU4nU$)agSorted key function to order InstanceSplit keys. InstanceSplits keys are one of: o revisionName o LATEST_REVISION_KEY Note LATEST_REVISION_KEY is not a str so its ordering with respect to revisionName keys is hard to predict. Args: key: Key for a InstanceSplits dictionary. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )r#)rFr,s rrErEs-" XF -XF -rcDUupSU[U5- - U[U5/$)aReturns object that sorts in the order we correct split rounding errors. The caller specifies explicit split percentages for some revisions and this module scales instance split for remaining revisions that are already serving instance split up or down to assure that 100% of instance split is assigned. This scaling can result in non integer percentages that Cloud Run does not supprt. We correct by: - Trimming the decimal part of float_percent, int(float_percent) - Adding an extra 1 percent instance split to enough revisions that have had their instance split reduced to get us to 100% The returned value sorts in the order we correct revisions: 1) Revisions with a bigger loss due are corrected before revisions with a smaller loss. Since 0 <= loss < 1 we sort by the value: 1 - loss. 2) In the case of ties revisions with less instance split are corrected before revisions with more instance split. 3) In case of a tie revisions with a smaller key are corrected before revisions with a larger key. Args: key_and_percent: tuple with (key, float_percent) Returns: A value that sorts with respect to values returned for other revisions in the order we correct for rounding errors. rJ)intrE)key_and_percentrF float_percents r _NewRoundingCorrectionPrecedencerOs3<'#=3}- -.c rfloat_percentagesc@UVs0sHo[X5_M nn[[[UR5555[UR55- n[ UR 5[ S9nUSUHupVX%==S- ss'M U$s snf)z$Returns rounded integer percentages.rFNrJ)rLroundr3r4sortedr+rO)rPkrounded_percentageslosscorrection_precedencerF_s r_IntPercentagesrZs->,=q  ! "",= U3(//12 3 4s  "8 $!%E&et,fc!- sBc [U5n[U5 [X5n[X5 [ X5nUR U5 [ U5n[UR5VVs/sHkupgUS:dM [R"U[:Xa[RRO[RRU[:waUOSUS9PMm snn[S9$s snnf)z$Returns the updated instance splits.rNrr%r"rR)r(r:r.r6rBupdaterZrTr+r InstanceSplitr#r r!r$rG)r&r/current_splits_mapr;unspecified_splits_modifiedint_percent_splitsrFr"s rGetUpdatedSplitsrbs ,N;+,,ZLZ4!9! /0&z2 1668 9lc q[ . & &++#>>dd!==ee!%88sd  9  $   s , C) >dd!==ee!%88sd  9  $   s  C *A"C c,U[:Xa[$US3$)N%)_MISSING_PERCENT)r"s r_FormatPercentagerl2s   Ya=rcV[UVs/sHoRPM sn5$s snf)z2Returns the sum of the instance split percentages.)r3r")rr's r _SumPercentrn9s!  0mm 0 11 0s&c\rSrSrSrS\\RS\\RS\ S\ 4Sjr \ S5r \ S 5r\ S 5r\ S 5r\ S 5r\ S 5r\ S5rSrg)InstanceSplitPairi>zHolder for InstanceSplit status information. The representation of the status of instance split for a worker pool includes: o User requested assignments (instance_splits) o Actual assignments (instance_split_statuses) target_splitsr& revision_namergc4XlX lX0lX@lg)aCreates a new InstanceSplitPair. Args: target_splits: A list of target instance splits that all reference the same revision, either by name or the latest ready. current_splits: A list of current instance splits that all reference the same revision, either by name or the latest ready. revision_name: The name of the revision referenced by the instance splits. latest: A boolean indicating if these instance splits reference the latest ready revision. Returns: A new InstanceSplitPair instance. N)_target_splits_current_splits_revision_name_latest)selfrqr&rrrgs r__init__InstanceSplitPair.__init__Gs*()'LrcHUR(a[$UR$)zThe key for the instance split.latest_revisionr#rrrxs rrFInstanceSplitPair.keyas#'"6"6 NDs.667>==>   4 O O              rrppairc`UR(a[nO URn[U5$)zSorted key function to order InstanceSplitPair objects by key. Args: pair: A InstanceSplitPair. Returns: A value that sorts by revisionName with LATEST_REVISION_KEY last. )r}r#rrrE)rrFs r_SortKeyFromInstanceSplitPairrs(  C   C  rc*[R"[5nUHrnUR[R R :XdURU:XaU[RU5 MUX#RRU5 Mt U$)a[Returns the instance split list into a map. The map uses LATEST_REVISION_KEY as the key for the latest ready revision. Args: splits: A list of InstanceSplit or InstanceSplitStatus objects. latest_ready_revision_name: The name of the latest ready revision. Returns: A map of revision names to InstanceSplit or InstanceSplitStatus objects. ) collections defaultdictlistrr r r!r%r#append)rrc splits_mapr's r _GetSplitsMaprsx&&&t,*e  5 5 [ [ \ >>7 7$%,,U3 ''. rr c /n[R"UR5n[ UR U5n[ UR U5n[U5RU5HSnU[:XaUOUnUR[URU5URU5UU[:H55 MU [U[S9$![a SnNf=f)z5Returns the instance split pairs for the worker pool.rR)r GetNameFromFullChildNamelatest_ready_revisionAttributeErrorrinstance_splitsinstance_split_statusessetunionr#rrprfrTr)r instance_split_pairsrcrqr&rFrrs rGetInstanceSplitPairsrs$ 99  - -  !!#=-!))+E.  % %n 5c&)-@&@"c   c "   s #  & &   6 $*G HH) $!#$s C CC)0r __future__rrrrrtypingrrr googlecloudsdk.command_lib.runr googlecloudsdk.corer ;googlecloudsdk.generated_clients.gapic_clients.run_v2.typesr r worker_pool_objectsrrkr#Errorrr^rrLr(r.r6r:rBrGrErOrZr=rbrhrlrnobjectrprrr WorkerPoolrrrrrs7'%'$$C*Vj Z-=-=  -- . #s(^& #s(^ cN  #s(^ S#X59#s(^.4S>$S#X48cN6^%A%A(0#L tCH~  556 S%U ++,  . & &' F556 # . & &'62^99:2s2 S S l (9    n**N,N,NNO !$   ~++^-O-OO PP @I$//I Ir