All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.anyline.data.milvus.adapter.MilvusActuator Maven / Gradle / Ivy

Go to download

There is a newer version: 8.7.2-20240916
Show newest version
/*
 * Copyright 2006-2023 www.anyline.org
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.anyline.data.milvus.adapter;

import io.milvus.v2.client.MilvusClientV2;
import io.milvus.v2.service.rbac.request.*;
import io.milvus.v2.service.rbac.response.DescribeRoleResp;
import io.milvus.v2.service.rbac.response.DescribeUserResp;
import org.anyline.annotation.Component;
import org.anyline.data.adapter.DriverActuator;
import org.anyline.data.adapter.DriverAdapter;
import org.anyline.data.param.ConfigStore;
import org.anyline.data.run.Run;
import org.anyline.data.runtime.DataRuntime;
import org.anyline.entity.DataRow;
import org.anyline.entity.DataSet;
import org.anyline.entity.authorize.Privilege;
import org.anyline.entity.authorize.Role;
import org.anyline.entity.authorize.User;
import org.anyline.metadata.*;
import org.anyline.metadata.refer.MetadataFieldRefer;
import org.anyline.util.BasicUtil;

import javax.sql.DataSource;
import java.sql.Connection;
import java.util.*;

@Component("anyline.environment.data.driver.actuator.milvus")
public class MilvusActuator implements DriverActuator {
    @Override
    public Class supportAdapterType() {
        return MilvusAdapter.class;
    }

    @Override
    public DataSource getDataSource(DriverAdapter adapter, DataRuntime runtime) {
        return null;
    }

    @Override
    public Connection getConnection(DriverAdapter adapter, DataRuntime runtime, DataSource datasource) {
        return null;
    }

    @Override
    public void releaseConnection(DriverAdapter adapter, DataRuntime runtime, Connection connection, DataSource datasource) {

    }

    private MilvusClientV2 client(DataRuntime runtime) {
        return (MilvusClientV2) runtime.getProcessor();
    }
    @Override
    public  void checkSchema(DriverAdapter adapter, DataRuntime runtime, DataSource datasource, T meta) {

    }

    @Override
    public  void checkSchema(DriverAdapter adapter, DataRuntime runtime, T meta) {

    }

    @Override
    public  void checkSchema(DriverAdapter adapter, DataRuntime runtime, Connection con, T meta) {

    }

    @Override
    public String product(DriverAdapter adapter, DataRuntime runtime, boolean create, String product) {
        return null;
    }

    @Override
    public String version(DriverAdapter adapter, DataRuntime runtime, boolean create, String version) {
        return null;
    }

    @Override
    public DataSet select(DriverAdapter adapter, DataRuntime runtime, String random, boolean system, ACTION.DML action, Table table, ConfigStore configs, Run run, String cmd, List values, LinkedHashMap columns) throws Exception {
        return null;
    }

    @Override
    public List> maps(DriverAdapter adapter, DataRuntime runtime, String random, ConfigStore configs, Run run) throws Exception {
        return null;
    }

    @Override
    public Map map(DriverAdapter adapter, DataRuntime runtime, String random, ConfigStore configs, Run run) throws Exception {
        return null;
    }

    @Override
    public long insert(DriverAdapter adapter, DataRuntime runtime, String random, Object data, ConfigStore configs, Run run, String generatedKey, String[] pks) throws Exception {
        return 0;
    }

    @Override
    public long update(DriverAdapter adapter, DataRuntime runtime, String random, Table dest, Object data, ConfigStore configs, Run run) throws Exception {
        return 0;
    }

    @Override
    public long execute(DriverAdapter adapter, DataRuntime runtime, String random, ConfigStore configs, Run run) throws Exception {
        return 0;
    }


    public  List  databases(DataRuntime runtime, String random, boolean greedy, Database query) {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Database/listDatabases.md
        List list = new ArrayList<>();
        return list;
    }

    /* *****************************************************************************************************************
     * 													role
     * -----------------------------------------------------------------------------------------------------------------
     * boolean create(DataRuntime runtime, Role role) throws Exception
     * boolean rename(DataRuntime runtime, Role origin, Role update) throws Exception;
     * boolean delete(DataRuntime runtime, Role role) throws Exception
     *  List roles(Catalog catalog, Schema schema, String pattern) throws Exception
     ******************************************************************************************************************/
    /**
     * role[调用入口]

     * 创建角色
     * @param role 角色
     * @return boolean
     */
    public boolean create(DataRuntime runtime, Role role) throws Exception {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Authentication/createRole.md
        CreateRoleReq req = CreateRoleReq.builder()
            .roleName(role.getName())
            .build();
        client(runtime).createRole(req);
        return true;
    }
    /**
     * role[调用入口]

     * 删除角色
     * @param role 角色
     * @return boolean
     */
    public boolean drop(DataRuntime runtime, Role role) throws Exception {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Authentication/dropRole.md
        DropRoleReq req = DropRoleReq.builder()
            .roleName(role.getName())
            .build();
        client(runtime).dropRole(req);
        return true;
    }
    public  List  roles(DataRuntime runtime, String random, boolean greedy, Role query) {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Authentication/listRoles.md
        List list = new ArrayList<>();
        String user = query.getUserName();
        if(null != user) {
            //用户相关角色
            DescribeUserResp response = client(runtime).describeUser(DescribeUserReq.builder()
                    .userName(user)
                    .build()
            );
            if(null != response) {
                List roles = response.getRoles();
                for(String role:roles) {
                    list.add((T)new Role(role));
                }
            }
        }else{
            //全部角色
            List roles = client(runtime).listRoles();
            for(String role:roles) {
                list.add((T)new Role(role));
            }
        }

        return list;
    }

    /* *****************************************************************************************************************
     * 													user
     * -----------------------------------------------------------------------------------------------------------------
     * boolean create(DataRuntime runtime, User user) throws Exception
     * boolean rename(DataRuntime runtime, User origin, User update) throws Exception;
     * boolean drop(DataRuntime runtime, User user) throws Exception
     * List users(Catalog catalog, Schema schema, String pattern) throws Exception
     ******************************************************************************************************************/

    /**
     * user[调用入口]

     * 创建 用户
     * @param user 用户
     * @return boolean
     */
    public boolean create(DataRuntime runtime, User user) throws Exception {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Authentication/createUser.md
        CreateUserReq req = CreateUserReq.builder()
            .userName(user.getName())
            .password(user.getPassword())
            .build();
        client(runtime).createUser(req);
        return true;
    }
    /**
     * user[调用入口]

     * 删除 用户
     * @param user 用户
     * @return boolean
     */
    public boolean drop(DataRuntime runtime, User user) throws Exception {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Authentication/dropUser.md
        DropUserReq req = DropUserReq.builder()
            .userName(user.getName())
            .build();
        client(runtime).dropUser(req);
        return true;
    }
    public  List  users(DataRuntime runtime, String random, boolean greedy, User query) {
        //https://milvus.io/api-reference/java/v2.4.x/v2/Authentication/listUsers.md
        List list = new ArrayList<>();
        List users = client(runtime).listUsers();
        for(String user:users) {
            list.add((T)new User(user));
        }
        return list;
    }
    /* *****************************************************************************************************************
     * 													privilege
     * -----------------------------------------------------------------------------------------------------------------
     *  List privileges(DataRuntime runtime, User user)
     ******************************************************************************************************************/

    /**
     * privilege[调用入口]

     * 查询用户权限
     * @param query 查询条件 根据metadata属性
     * @return List
     */

    public  List privileges(DataRuntime runtime, String random, boolean greedy, Privilege query) throws Exception {
        List list = new ArrayList<>();
        String role = query.getRoleName();
        String user = query.getUserName();
        if(null != user) {
            Map map = new HashMap<>();
            List roles = roles(runtime, random, false, new Role().setUser(user));
            for(Role r:roles) {
                List ps = privileges(runtime, random, greedy, new Privilege().setRole(r));
                for(T p:ps) {
                    map.put(p.getName(), p);
                }
            }
            list.addAll(map.values());
        }else if(null != role) {
            //角色相关权限
            DescribeRoleReq describe = DescribeRoleReq.builder()
                    .roleName(role)
                    .build();
            DescribeRoleResp response = client(runtime).describeRole(describe);
            if(null != response) {
                List grants = response.getGrantInfos();
                for(DescribeRoleResp.GrantInfo grant:grants) {
                    Privilege privilege = new Privilege();
                    privilege.setRole(new Role(role));
                    privilege.setObjectName(grant.getObjectName());
                    privilege.setObjectType(grant.getObjectType());
                    privilege.setName(grant.getPrivilege());
                    privilege.setDatabase(grant.getDbName());
                    list.add((T)privilege);
                }
            }
        }
        return list;
    }

    /* *****************************************************************************************************************
     * 													grant
     * -----------------------------------------------------------------------------------------------------------------
     * boolean grant(DataRuntime runtime, User user, Privilege ... privileges)  throws Exception
     * boolean grant(DataRuntime runtime, User user, Role ... roles) throws Exception
     * boolean grant(DataRuntime runtime, Role role, Privilege ... privileges) throws Exception
     ******************************************************************************************************************/

    /**
     * privilege[调用入口]

     * 授权
     * @param user 用户
     * @param privileges 权限
     * @return boolean
     */

    public boolean grant(DataRuntime runtime, User user, Privilege ... privileges)  throws Exception {
        //没有实现 可以通过 给角色授权 给用户赋角色
        return true;
    }

    /**
     * privilege[调用入口]

     * 授权
     * @param user 用户
     * @param roles 角色
     * @return boolean
     */

    public boolean grant(DataRuntime runtime, User user, Role ... roles)  throws Exception {
        for(Role role:roles) {
            client(runtime).grantRole(GrantRoleReq.builder()
                    .roleName(role.getName())
                    .userName(user.getName())
                    .build()
            );
        }
        return true;
    }

    /**
     * privilege[调用入口]

     * 授权
     * @param role 角色
     * @param privileges 权限
     * @return boolean
     */

    public boolean grant(DataRuntime runtime, Role role, Privilege ... privileges)  throws Exception {
        for(Privilege privilege:privileges) {
            GrantPrivilegeReq.GrantPrivilegeReqBuilder build = GrantPrivilegeReq.builder();
            build.roleName(role.getName());
            if(BasicUtil.isNotEmpty(privilege.getObjectName())) {
                build.objectName(privilege.getObjectName());
            }
            if(BasicUtil.isNotEmpty(privilege.getObjectType())) {
                build.objectType(privilege.getObjectType());
            }
            build.privilege(privilege.getName());
            GrantPrivilegeReq req = build.build();
            client(runtime).grantPrivilege(req);
        }
        return true;
    }

    /* *****************************************************************************************************************
     * 													revoke
     * -----------------------------------------------------------------------------------------------------------------
     * boolean revoke(DataRuntime runtime, User user, Privilege ... privileges)  throws Exception
     * boolean revoke(DataRuntime runtime, User user, Role ... roles) throws Exception
     * boolean revoke(DataRuntime runtime, Role role, Privilege ... privileges) throws Exception
     ******************************************************************************************************************/

    /**
     * revoke[调用入口]

     * 撤销授权
     * @param user 用户
     * @param privileges 权限
     * @return boolean
     */

    public boolean revoke(DataRuntime runtime, User user, Privilege ... privileges) throws Exception {
        //没有实现 可以通过 给用户删除角色 或给角色删除权限
        return true;
    }

    /**
     * revoke[调用入口]

     * 撤销授权
     * @param user 用户
     * @param roles 角色
     * @return boolean
     */

    public boolean revoke(DataRuntime runtime, User user, Role ... roles) throws Exception {
        for(Role role:roles) {
            client(runtime).revokeRole(RevokeRoleReq.builder()
                    .roleName(role.getName())
                    .userName(user.getName())
                    .build()
            );
        }
        return true;
    }
    /**
     * revoke[调用入口]

     * 撤销授权
     * @param role 角色
     * @param privileges 权限
     * @return boolean
     */

    public boolean revoke(DataRuntime runtime, Role role, Privilege ... privileges) throws Exception {
        for (Privilege privilege:privileges) {
            client(runtime).revokePrivilege(RevokePrivilegeReq.builder()
                    .dbName(privilege.getDatabaseName())
                    .roleName(role.getName())
                    .objectType(privilege.getObjectType())
                    .privilege(privilege.getName())
                    .objectName(privilege.getObjectName())
                    .build()
            );
        }
        return true;
    }


}
 















© 2015 - 2024 Weber Informatics LLC | Privacy Policy