Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Generic type with custom object in Swift language

Tags:

generics

swift

I wonder is there any way to compare two generic type instances with == operator in following generic function:

 func compare<T>(T a, T b) -> Bool {
    if a == b{
       // do something
       return true; 
    }else{
       // do another thing
       return false;
    }   
 }

Here is my custom object:

class MyObj{
  var id = 3
  var name: String?
}
like image 273
ductran Avatar asked Jun 03 '14 15:06

ductran


2 Answers

From Apple Developer Resources,

Not every type in Swift can be compared with the equal to operator (==). If you create your own class or structure to represent a complex data model, for example, then the meaning of “equal to” for that class or structure is not something that Swift can guess for you. Because of this, it is not possible to guarantee that this code will work for every possible type T, and an appropriate error is reported when you try to compile the code.

All is not lost, however. The Swift standard library defines a protocol called
Equatable, which requires any conforming type to implement the equal to operator (==) and the not equal to operator (!=) to compare any two values of that type. All of Swift’s standard types automatically support the Equatable protocol.

Any type that is Equatable can be used safely with the findIndex function, because it is guaranteed to support the equal to operator. To express this fact, you write a type constraint of Equatable as part of the type parameter’s definition when you define the function:

func findIndex<T: Equatable>(array: T[], valueToFind: T) -> Int? {
    for (index, value) in enumerate(array) {
        if value == valueToFind {
            return index
        }
    }
    return nil
}

Here is a sample from their documentation explaining how to override ==

struct MyStruct: Equatable {
    var name = "Untitled"
}
func == (lhs: MyStruct, rhs: MyStruct) -> Bool {
    return lhs.name == rhs.name
}

let value1 = MyStruct()
var value2 = MyStruct()
let firstCheck = value1 == value2
// firstCheck is true

value2.name = "A New Name"
let secondCheck = value1 == value2
// secondCheck is false

In your case you would do,

class MyObj{
  var id = 3
  var name: String?
}

func == (lhs: MyObj, rhs: MyObj) -> Bool {
    return lhs.id == rhs.id
}
like image 126
Gokhan Arik Avatar answered Nov 15 '22 07:11

Gokhan Arik


You can create your own implementation of the equivalence operators == and !=, like this:

@infix func == (left: MyObj, right: MyObj) -> Bool {
    return (left.id == right.id) && (left.name == right.name)
}
@infix func != (left: MyObj, right: MyObj) -> Bool {
    return !(left == right)
}

var obj1 = MyObj()
var obj2 = MyObj()
obj1.id = 5
obj1 == obj2 // false
obj2.id = 5
obj1 == obj2 // true

obj1.name = "John"
obj1 == obj2 // false
like image 23
Nate Cook Avatar answered Nov 15 '22 07:11

Nate Cook