Merge pull request #82 from logic-building/RestPtr_UserDefinedType

Auto-generated code has pointer version of function - Rest

Author: logic-building

gofp/gofp.go

@@ -231,6 +231,9 @@ func generateFPCode(pkg, dataTypes, imports string) (string, error) {
 		template += template2.Rest()
 		template = r.Replace(template)
 
+		template += template2.RestPtr()
+		template = r.Replace(template)
+
 		template += template2.Reduce()
 		template = r.Replace(template)
 

internal/employee/fp.go

@@ -23,3 +23,29 @@ func Rest<CONDITIONAL_TYPE>(l []<TYPE>) []<TYPE> {
 }
 `
 }
+
+// RestPtr removes 1st item of the list and return new list having rest of the items
+func RestPtr() string {
+	return `
+
+// Rest<CONDITIONAL_TYPE> removes 1st item of the list and return new list having rest of the items
+func Rest<CONDITIONAL_TYPE>Ptr(l []*<TYPE>) []*<TYPE> {
+	if l == nil {
+		return []*<TYPE>{}
+	}
+
+	len := len(l)
+	if len == 0 || len == 1 {
+		return []*<TYPE>{}
+	}
+
+	newList := make([]*<TYPE>, len-1)
+
+	for i, v := range l[1:] {
+		newList[i] = v
+	}
+
+	return newList
+}
+`
+}