lightning:treegrid select child rows upon selecting parent row in Salesforce

lightning:treegrid select child rows upon selecting parent row in Salesforce

Sample Code:

Example.cmp:

<aura:component implements=”force:appHostable”
                controller=”AccountListController”>
 
    <aura:attribute name=”gridColumns” type=”List” />
    <aura:attribute name=”gridData” type=”Object” />
    <aura:attribute name=”selectedRows” type=”list” />
   
    <aura:handler name=”init” value=”{!this}” action=”{!c.fetchAccounts}”/>
   
    <lightning:treeGrid columns=”{! v.gridColumns }”
        data=”{! v.gridData }”
        keyField=”Id”
        aura:id=”mytree”
        onrowselection=”{! c.onSelected }”
        selectedRows=”{! v.selectedRows }”/>
       
</aura:component>


Example.js:

({
    fetchAccounts : function( component, event, helper ) {
        var columns = [
            {
                type: ‘text’,
                fieldName: ‘Name’,
                label: ‘Account Name’
            },
            {
                type: ‘text’,
                fieldName: ‘Industry’,
                label: ‘Industry’
            },
            {
                type: ‘text’,
                fieldName: ‘FirstName’,
                label: ‘First Name’
            },
            {
                type: ‘text’,
                fieldName: ‘LastName’,
                label: ‘Last Name’
            },
            {
                type: ’email’,
                fieldName: ‘Email’,
                label: ‘Email’
            }
        ];
        component.set(‘v.gridColumns’, columns);
        var action = component.get(“c.fetchAccts”);
        action.setCallback(this, function(response){
            var state = response.getState();
            if ( state === “SUCCESS” ) {
                var data = response.getReturnValue();
                for ( var i = 0; i < data.length; i++ ) {
                    data[i]._children = data[ i ][ ‘Contacts’ ];
                    delete data[i].Contacts;

                }
                component.set( ‘v.gridData’, data );
            }
        });
        $A.enqueueAction(action);
    },
   
    onSelected : function( component, event, helper ) {
       
        var selectedRows = event.getParam( ‘selectedRows’ );
        var data = component.get( ‘v.gridData’ );
        var selectedData = [];
       
        for ( var i = 0; i < selectedRows.length; i++ ) {
           
            for ( var j = 0; j < data.length; j++ ){
               
                if ( selectedRows[ i ].Id == data[ j ].Id ) {
                   
                    var childrenRecs = data[ j ][ ‘_children’ ];
                    selectedData.push( data[ j ].Id );
                   
                    for ( var k = 0; k < childrenRecs.length; k++ )
                        selectedData.push( childrenRecs[ k ].Id );   
                   
                }
               
            }
           
        }
       
        component.set( ‘v.gridData’, data );
        component.set( ‘v.selectedRows’, selectedData );
       
    }

})


AccountListController Apex Class:

public class AccountListController {

    @AuraEnabled
    public static List < Account > fetchAccts() {
        return [ SELECT Id, Name, Industry,
                ( SELECT Id, FirstName, LastName FROM Contacts )
                FROM Account LIMIT 100 ];
    }
   

}


Output:


Leave a Reply