[go lang] 인터페이스

인터페이스 정의, 구현

인터페이스는 메소드 시그니처의 집합으로, 어떤 타입이 특정 인터페이스를 구현하기 위해서는 인터페이스에 명시된 메소드를 모두 구현하면 된다. 아래는 DBManager 라는 인터페이스를 정의하고 *MySQLDBManager 타입이 이를 구현하는 예제이다.

// 인터페이스 정의
type DBManager interface {
    Connect(ip string, user string, pw string) bool
    Query(q string) Data
    Commit() bool
    Disconnect() bool
}

// Concrete Type 정의
type MySQLDBManager struct {
    ...
}

// MySQLDBManager의 DBManager 구현
func (m *MySQLDBManager) Connect(ip string, user string, pw string) bool {
    ...
}

func (m *MySQLDBManager) Query(q string) Data {
    ...
}

func (m *MySQLDBManager) Commit() bool {
    ...
}

func (m *MySQLDBManager) Disconnect() bool {
    ...
}

이를 이용해 다음과 같은 클라이언트 구현이 가능하다.

var m DBManager

// MySQL 데이터베이스 매니저 구현 이용
m = MySQLDBManager{}
m.connect(...)
...

// Oracle 데이터베이스 매니저 구현 이용
m = OracleDBManager{}
m.connect(...)
...

Type Assertions, Type Switch

빈 인터페이스(empty interface; interface{}) 변수의 경우 모든 타입의 값을 담을 수 있다.(왜냐하면 모든 타입은 적어도 0개 이상의 메소드를 구현하므로) Type Assertion(타입 단언)을 통해 인터페이스 변수에 담긴 값을 특정 타입으로 다룰 수 있다. 아래 예제 코드는 s = i.(string) 또는 s, ok = i.(string) 로 값을 가져오는데, 첫 번째 문법의 경우 값을 단언한 타입으로 가져올 수 없다면 런타임 에러를 발생시킨다. 두 번째 문법의 경우 단언한 타입으로 값을 가져올 수 있다면 값과 true를, 그럴 수 없다면 단언한 타입의 zero-value와 false를 반환한다.

var i interface{}

i = "Hello world"

s = i.(string)
s, ok = i.(string) // "Hello world", true

f, ok = i.(float64) // 0.0, false

switch 문에서 타입 단언 이용은 다음과 같다.

switch v := i.(type) {
case int:
    ...
case string:
    ...
default:
    ...
}

예제: Error 구현

type error interface {
    Error() string
}

'간단한' 제곱근 함수의 반환에 에러 반환을 추가하자. Error 인터페이스는 위와 같이 정의되었다.

type ErrNegativeSqrt float64

func (e ErrNegativeSqrt) Error() string {
    return fmt.Sprintf("cannot Sqrt negative number: %v", float64(e))
}

func Sqrt(x float64) (float64, error) {
    if x < 0.0 {
        return 0.0, ErrNegativeSqrt(x)
    }
    z := 1.0
    for i := 0; i < 10; i++ {
        z -= (z*z - x) / (2 * x)
    }
    return z, nil
}

예제: rot13 암호화 바이트 스트림 구현

type Reader {
    func Read(b []byte) (n int, err error)
}

Reader 인터페이스는 위와 같이 정의되었다. 데이터를 읽어 b []byte 에 기록하고 읽은 바이트수 n을 반환한다. 만약 에러가 발생했다면 error를 반환하고, 에러가 없다면 nil 을 반환한다.

type Rot13Reader struct {
    r io.Reader
}

func rot(b byte) byte {
    switch {
    case 'a' <= b && b <= 'z':
        return (b - 'a' + 13) % 26 + 'a'
    case 'A' <= b && b <= 'Z':
        return (b - 'A' + 13) % 26 + 'A'
    default:
        return b
    }
}

func (r Rot13Reader) Read(buffer []byte) (int, error) {
    n, err := r.r.Read(buffer)
    if err != nil {
        return n, err
    }
    for i := range n {
        buffer[i] = rot(buffer[i])
    }
    return n, nil
}