z*ddZddlmZddlmZddlmZddlmZddlZddlmZm Z m Z ddl m Z dd l mZddlZd Zd Zd Zej(d GddeZde eeej0fde eeej0fdefdZdZde j6de j6defdZy)zAWrapper for Cloud Run InstanceSplits messages in spec and status.)absolute_import)division)print_function)unicode_literalsN)ListMappingOptional)instance_split)run_v1_messages-cB|tk(rtSdj|S)Nz{}%)_MISSING_PERCENTformatpercents 5lib/googlecloudsdk/api_lib/run/instance_split_pair.py_FormatPercentager!s     <<  c&td|DS)z&Sums the percents of the given splits.c3NK|]}|js|jyw)Nr).0is r z_SumPercent..*s 41!))QYYs%%)sum)splitss r _SumPercentr(s 4 4 44rT)frozenceZdZUdZeej ed<eej ed<eed<e ed<dZ e e ed<e dZe d Ze d Ze d Ze d Ze d Ze dZy)InstanceSplitPairaHolder for InstanceSplit status information. The representation of the status of instance splits for a worker pool includes: o User requested assignments (spec.instanceSplits) o Actual assignments (status.instanceSplits) Each of spec.instanceSplits and status.instanceSplits may contain multiple instance splits that reference the same revision, either directly by name or indirectly by referencing the latest ready revision. The spec and status instance splits for a revision may differ after a failed split update or during a successful one. A InstanceSplitPair holds all spec and status InstanceSplits that reference the same revision by name or reference the latest ready revision. Both the spec and status instance splits can be empty. The latest revision can be included in the spec instance splits two ways o by revisionName o by setting latestRevision to True. Managed cloud run provides a single combined status instance split for both spec entries with: o revisionName set to the latest revision's name o percent set to combined percentage for both spec entries o latestRevision not set In this case both spec targets are paired with the combined status target and a status_percent_override value is used to allocate the combined instance split. Attributes: spec_splits: The spec instance splits for the referenced revision. status_splits: The status instance splits for the referenced revision. revision_name: The name of the referenced revision. latest: Boolean indicating if the instance splits reference the latest ready revision. status_percent_override: The percent of splits allocated to the referenced revision in the worker pool's status. spec_splits status_splits revision_namelatestNstatus_percent_overridecd|jrtjStj|S)z,Returns the key for the instance split pair.)latestRevisionr LATEST_REVISION_KEYGetKeyselfs rkeyzInstanceSplitPair.keyis2    ** " "4 (rc|jS)zBReturns true if the instance splits reference the latest revision.)r#r)s rr&z InstanceSplitPair.latestRevisionrs ;;rc|jS)z,Returns the name of the referenced revision.)r"r)s r revisionNamezInstanceSplitPair.revisionNamews   rcv|jr(tjt|jStS)z7Returns the sum of the spec instance split percentages.)r six text_typerrr)s r specPercentzInstanceSplitPair.specPercent|s-  ]];t'7'78 99 rc|jtj|jS|jr(tjt |jSt S)z9Returns the sum of the status instance split percentages.)r$r0r1r!rrr)s r statusPercentzInstanceSplitPair.statusPercentsN ##/ ]]477 88   ]];t'9'9: ;; rc|j|jk(rt|jSdjt|jt|jS)z(Returns human readable revision percent.z{:4} (currently {}))r4r2rrr)s rdisplayPercentz InstanceSplitPair.displayPercentsX T--- t11 22 " ) ) D,, - D.. /rc||jr%tj|d|jdS|jS)z+Returns human readable revision identifier.z (currently ))r&r r(r.r)s rdisplayRevisionIdz#InstanceSplitPair.displayRevisionIds<     %      r)__name__ __module__ __qualname____doc__rmessages InstanceSplit__annotations__strboolr$r intpropertyr+r&r.r2r4r6r9rrrr-s(hH**++h,,-- ,+/8C=/              rr spec_dict status_dictlatest_ready_revision_namecd}tj|vrBtj|vr0||vr,||}||tj<||vr t|}|S||=|S)a:Splits the fully-managed latest status target. For managed the status target for the latest revision is included by revisionName only and may hold the combined splits percent for both latestRevisionName and latestRevision spec splits. Here we adjust keys in status_dict to match with spec_dict. Args: spec_dict: Dictionary mapping revision name or 'LATEST' to the spec instance split referencing that revision. status_dict: Dictionary mapping revision name or 'LATEST' to the status instance split referencing that revision. Modified by this function. latest_ready_revision_name: The name of the latest ready revision. Returns: Optionally, the id of the list of status splits containing the combined instance splits referencing the latest ready revision by name and by latest. N)r r'id)rFrGrHcombined_status_splits_idlatest_status_splitss r_SplitManagedLatestStatusSplitsrMsw.#((I5  , ,K ? $ 3&'AB6JK223!Y."$%9": #" 0 1 ""rcd}t||k(rOt|tj}t|}t ||}|tjk(r|}|S||z }|S)zFComputes the optional override percent to apply to the status percent.N)rJrr r'min)r+rFr!rKpercent_overridespec_by_latest_percentstatus_percentstatus_by_latest_percents r_PercentOverriderTsy 33(.445!/N"#9>J n0001 (*BB r spec_split status_splitc t|}t|}t|||}g}t|j|D]n}|j |g}|j |g} t ||| |} |t jk(r|} d} n|} d} |jt|| | | | pt|t jS)avReturns a list of InstanceSplitPairs for a WorkerPool. Given the spec and status instance splits wrapped in a InstanceSplits instance for a sevice, this function pairs up all spec and status instance splits that reference the same revision (either by name or the latest ready revision) into InstanceSplitPairs. This allows the caller to easily see any differences between the spec and status split. For fully-managed Cloud Run, the status target for the latest revision is included by revisionName only and may hold the combined split percent for both latestRevisionName and latestRevision spec targets. This function splits the fully-managed status target for the latest revision into a target for the percent allocated to the latest revision by name and a target for the percent allocated to the latest revision because it is latest. Args: spec_split: A instance_split.InstanceSplits instance wrapping the spec split. status_split: A instance_split.InstanceSplits instance wrapping the status split. latest_ready_revision_name: The name of the worker pool's latest ready revision. Returns: A list of InstanceSplitPairs representing the current state of the worker pool's instance split assignments. The InstanceSplitPairs are sorted by revision name, with targets referencing the latest ready revision at the end. TF)r+) dictrMsetuniongetrTr r'appendrsortedSortKeyFromSplit) rUrVrHrFrGrKresultkr r!rPr"r#s rGetInstanceSplitPairsrasP:)\"+=8 & y>   ,a--2&KOOAr*M' 9m%> N . ..0mfmf MM  v?O  -$ N;; <r0rrr dataclassobjectrrAr?rMrTInstanceSplitsrarErrrjsH&%'**5T !5 d#tt$tn##sD!7!7889##d8#9#9::;##!$##L  A=--A= //A=!$A=r