File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/solrcloud.cpython-310.pyc
Back
o ;j�: � @ s d Z ddlZddlm mZ ddlmZ e�e�Z g d�Z ddgZ g d�Z g d�Z d d gZd d gZd gZ g d�Zdd� Zd4dd�Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd5d d!�Zd"d#� Zd$d%� Z d&d'� Z d5d(d)�Z d5d*d+�Z d5d,d-�Z d.d/� Z!d0d1� Z"d2d3� Z#dS )6z� Module for solrcloud configuration .. versionadded:: 2017.7.0 For now, module is limited to http-exposed API. It doesn't implement config upload via Solr zkCli � N)�SaltInvocationError) ZconfigZschemaZdataDirZ configSet� propertiesZ coreNodeNameZulogDirZshard� collectionZrolesZ transientZ loadOnStartup)�collection.configNamez router.field�async�rule�snitch)Z numShards�replicationFactor�maxShardsPerNode�autoAddReplicaszcreateNodeSet.shuffleZshardsZ createNodeSetr )r r r r r r c C s dS )NZ solrcloud� r r r �J/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/solrcloud.py� __virtual__@ s r �http://localhost:8983/solr/c K s� t |t�s td��|dd� dkr|d }tj|| fdddd�|��}d|v rH|d d kr7td j|| d ���td jt|d �|| |d d���|d S )z� Internal function to query solrcloud :param url: relative solr URL :param solr_url: solr base URL :param kwargs: additional args passed to http.query call :return: Query JSON answer converted to python dict :rtype: dict zsolr_url must be a string���N�/�jsonT)Z decode_type�decodeZ raise_error�error�statusi� zUGot a 404 when trying to contact solr at {solr_url}{url}. Please check your solr URL.)�solr_url�urlz;Got a {status} error when calling {solr_url}{url} : {error})r r r r �dict)� isinstance�str� ValueError�http�queryr �format)r r �kwargsZ query_resultr r r �_queryD s6 ��� �� �� r c C s� d}| � � D ]N\}}|tv r*t|t�std| d ��|d | d |r&dnd }q|tv rFt|t�s;td| d ��|d | d | }q|d t|� d t|� }q|S ) z9 Internal function to validate core properties � �Option "�" value must be an booleanz &property.�=�true�falsez'In option "properties", core property "�" value must be a string)�items�BOOL_PROPS_LISTr �boolr �STRING_PROPS_LISTr )r Z props_stringZ prop_nameZ prop_valuer r r �_validate_core_propertiesm s8 ��� �� ����r, c C sJ d}| � � D ]�\}}|tv r&t|t�std| d ��|d | d | }q|tv rDt|t�s7td| d ��|d | d t|� }q|tv rdt|t�sUtd| d ��|d | d |r`dnd }q|t v r�t|t �sutd| d ��|d | d d � |� }q|t v r�t|t �s�td| d ��|t|� }qtd | d ��|S )z= Internal function to validate collections options r! r" r' �&r$ z" value must be an intr# r% r& z!" value must be a list of strings�, z" value must be an dictzUnknown option "�")r( �STRING_OPTIONS_LISTr r r �INT_OPTIONS_LIST�int�BOOL_OPTIONS_LISTr* �LIST_OPTIONS_LIST�list�join�DICT_OPTIONS_LISTr r, )�options�options_stringZ option_nameZ option_valuer r r � _validate_collection_options� sN � ���� �� �� r: c C s t S )z� Get collection option list that can only be defined at creation CLI Example: .. code-block:: bash salt '*' solrcloud.collection_creation_options )�CREATION_ONLY_OPTIONr r r r �collection_creation_options� s r< c K � t di | ��d S )z� Get cluster status Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.cluster_status �.admin/collections?action=CLUSTERSTATUS&wt=json�clusterN)r> �r �r r r r �cluster_status� s rB c K s4 t | t�s td��tdi |��}d|v o| |d v S )z� Check alias existence Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.alias_exists my_alias �Alias name must be a string�aliasesNr )r r r rB )� alias_namer r? r r r � alias_exists� s rF c sR t � t�s td��dd� tdi |��d �� D �}dd� � fdd�|D �D �}|S ) z� Get collection list for an alias Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.alias_get my_alias rC c S s, g | ]}d |d v r|d |d d f�qS )rD � r r ��.0Zk_vr r r � <listcomp> s �z)alias_get_collections.<locals>.<listcomp>� collectionsc S s g | ]}|d �qS )r r )rI Zk_v1r r r rJ s �c s g | ] }� |d v r|�qS )rG r rH �rE r r rJ s Nr )r r r rB r( )rE r Zcollection_aliasesrD r rL r �alias_get_collections s ��rM c K sN t |t�s td��|D ] }t |t�std��q tdj| d�|�d�fi |��S )z� Define an alias Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.alias_set my_alias collections=[collection1, colletion2] zJCollection parameter must be defined and contain a list of collection name� Collection name must be a stringzSadmin/collections?action=CREATEALIAS&name={alias}&wt=json&collections={collections}r. )�aliasrK )r r5 r r r r r r6 )rE rK r r r r r �alias_set_collections" s � � ���rP c K s t dj| d�fi |�� dS )z� Check if a collection exists Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.collection_reload collection_name z9admin/collections?action=RELOAD&name={collection}&wt=json)r N)r r )r r r r r �collection_reloadA s �� �rQ c K r= )z� List all collections Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.collection_list �%admin/collections?action=LIST&wt=jsonrK N)rR r@ rA r r r �collection_listX s rS c K s$ t | t�s td��| tdi |��v S )z� Check if a collection exists Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.collection_exists collection_name rN Nr )r r r rS )�collection_namer r r r �collection_existsj s rU c K sP t | fi |��s td��|durd|� �}nd}tdj| ||d�fi |�� dS )z� Create a backup for a collection. Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.core_backup collection_name /mnt/nfs_backup zCollection doesn't existsNz&name=r! �P{collection}/replication?command=BACKUP&location={location}{backup_name}&wt=json�r � backup_name�location)rU r r r )rT rY rX r r r r �collection_backup s �� �rZ c K sP t di |��D ] }|durdj||d�}nd}tdj||| d�fi |�� qdS )z� Create a backup for all collection present on the server. Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.core_backup /mnt/nfs_backup Nz&name={backup}.{collection})Zbackupr r! rV rW r )rS r r )rY rX r rT r r r �collection_backup_all� s �����r[ c K sB |du ri }t |t�std��t|�}td| | fi |�� dS )aY Create a collection, Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.collection_create collection_name Collection creation options may be passed using the "options" parameter. Do not include option "name" since it already specified by the mandatory parameter "collection_name" .. code-block:: bash salt '*' solrcloud.collection_create collection_name options={"replicationFactor":2, "numShards":3} Cores options may be passed using the "properties" key in options. Do not include property "name" .. code-block:: bash salt '*' solrcloud.collection_create collection_name options={"replicationFactor":2, "numShards":3, "properties":{"dataDir":"/srv/solr/hugePartitionSollection"}} Nz&options parameter must be a dictionaryz-admin/collections?action=CREATE&wt=json&name=)r r r r: r )rT r8 r r9 r r r �collection_create� s ��� �r\ c C s$ zt | � W dS ty Y dS w )z� Check collections options CLI Example: .. code-block:: bash salt '*' solrcloud.collection_check_options '{"replicationFactor":4}' TF)r: r )r8 r r r �collection_check_options� s �r] c K s� t d i |��}|d | d |d | d d t|d | d �t|d | d �|d | d du d �}d |d | v rG|d | d |d < d |d | v rY|d | d |d < |S )z� Get collection options Additional parameters (kwargs) may be passed, they will be proxied to http.query CLI Example: .. code-block:: bash salt '*' solrcloud.collection_get_options collection_name rK Z configNameZrouter�namer r r T)r z router.namer r r r r Nr )rB r2 )rT r r? r8 r r r �collection_get_options� s$ ���� r_ c K sN t |�� �D ]}|tvrtd| d ��qt|�}td| | fi |�� dS )aT Change collection options Additional parameters (kwargs) may be passed, they will be proxied to http.query Note that not every parameter can be changed after collection creation CLI Example: .. code-block:: bash salt '*' solrcloud.collection_set_options collection_name options={"replicationFactor":4} zOption z- can't be modified after collection creation.z=admin/collections?action=MODIFYCOLLECTION&wt=json&collection=N)r5 �keysr; r r: r )rT r8 r �optionr9 r r r �collection_set_options s ����� �rb )r )N)$�__doc__�loggingZsalt.utils.http�utilsr Zsalt.exceptionsr � getLogger�__name__�logr+ r) r0 r1 r3 r4 r7 r; r r r, r: r< rB rF rM rP rQ rS rU rZ r[ r\ r] r_ rb r r r r �<module> sD )'; - $
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings