3434import org .jooq .SelectSeekStepN ;
3535import org .jooq .impl .DSL ;
3636import org .jooq .lambda .tuple .Tuple3 ;
37+ import org .jooq .lambda .tuple .Tuple4 ;
3738import org .springframework .beans .factory .annotation .Autowired ;
3839import org .springframework .stereotype .Service ;
3940
4041import java .util .ArrayList ;
4142import java .util .List ;
4243import java .util .stream .Collectors ;
43- import java .util .stream .Stream ;
4444
4545import static org .finos .waltz .common .Checks .checkNotNull ;
4646import static org .finos .waltz .schema .Tables .ENTITY_HIERARCHY ;
@@ -107,11 +107,12 @@ public void register() {
107107 }
108108
109109 // Prepare select fields for each level
110- List <Tuple3 <Field <String >, Field <String >, Field <Long >>> levelFields = levelTuples .stream ()
110+ List <Tuple4 <Field <String >, Field <String >, Field <Long >, Field < String >>> levelFields = levelTuples .stream ()
111111 .map (t -> tuple (
112112 _m (t ).NAME .as ("Level " + t .v1 + " Name" ),
113113 _m (t ).EXTERNAL_ID .as ("Level " + t .v1 + " External Id" ),
114- _m (t ).ID .as ("Level " + t .v1 + " Waltz Id" )
114+ _m (t ).ID .as ("Level " + t .v1 + " Waltz Id" ),
115+ _m (t ).DESCRIPTION .as ("Level " + t .v1 + " Description" )
115116 ))
116117 .collect (Collectors .toList ());
117118
@@ -127,10 +128,12 @@ public void register() {
127128 selectFields .add (levelFields .get (i ).v1 );
128129 selectFields .add (levelFields .get (i ).v2 );
129130 selectFields .add (levelFields .get (i ).v3 );
131+ selectFields .add (levelFields .get (i ).v4 );
130132 } else {
131133 selectFields .add (DSL .val ("" ).as ("Level " + (i +1 ) + " Name" ));
132134 selectFields .add (DSL .val ("" ).as ("Level " + (i +1 ) + " External Id" ));
133135 selectFields .add (DSL .val (null , Long .class ).as ("Level " + (i +1 ) + " Waltz Id" ));
136+ selectFields .add (DSL .val ("" ).as ("Level " + (i +1 ) + " Description" ));
134137 }
135138 }
136139
0 commit comments