controllers/member.go

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package controllers

import (
	"fmt"
	"strconv"
	"strings"
)

type Member struct {
	Name string
	// Kubernetes namespace this member runs in.
	Namespace string
	// ID field can be 0, which is unknown ID.
	// We know the ID of a member when we get the member information from dbmaster,
	// but not from Kubernetes pod list.
	ID              uint64
	Phase           string
	Created         bool
	RunningAndReady bool
	Version         string
}

func (m Member) Ordinal() int {
	idx := strings.LastIndex(m.Name, "-")
	id, _ := strconv.ParseInt(m.Name[idx+1:], 10, 32)
	return int(id)
}

type MemberSet map[string]*Member

// the set of all members of s1 that are not members of s2
func (ms MemberSet) Diff(other MemberSet) MemberSet {
	diff := MemberSet{}
	for n, m := range ms {
		if _, ok := other[n]; !ok {
			diff[n] = m
		}
	}
	return diff
}

func (ms MemberSet) Get(id int) *Member {
	for _, m := range ms {
		idx := strings.LastIndex(m.Name, "-")
		mid, _ := strconv.ParseInt(m.Name[idx+1:], 10, 32)
		if int(mid) == id {
			return m
		}
	}
	return nil
}

// IsEqual tells whether two member sets are equal by checking
// - they have the same set of members and member equality are judged by Name only.
func (ms MemberSet) IsEqual(other MemberSet) bool {
	if ms.Size() != other.Size() {
		return false
	}
	for n := range ms {
		if _, ok := other[n]; !ok {
			return false
		}
	}
	return true
}

func (ms MemberSet) Size() int {
	return len(ms)
}

func (ms MemberSet) String() string {
	var mstring []string

	for m, v := range ms {
		mstring = append(mstring, fmt.Sprintf("%s:%s", m, v.Version))
	}
	return strings.Join(mstring, ",")
}

func (ms MemberSet) PickOne() *Member {
	for _, m := range ms {
		return m
	}
	panic("empty")
}

func (ms MemberSet) Add(m *Member) {
	ms[m.Name] = m
}

func (ms MemberSet) Remove(name string) {
	delete(ms, name)
}

func (ms MemberSet) Ordinals() map[int]bool {
	ids := map[int]bool{}
	for _, m := range ms {
		ids[m.Ordinal()] = true
	}
	return ids
}

func (ms MemberSet) Names() []string {
	names := make([]string, 0)
	for _, m := range ms {
		names = append(names, m.Name)
	}
	return names
}

func (ms MemberSet) Duplicate() MemberSet {
	r := MemberSet{}
	for k, v := range ms {
		r[k] = v
	}
	return r
}

func allMembersHealth(ms MemberSet) bool {
	for _, m := range ms {
		if !m.RunningAndReady {
			return false
		}
	}
	return true
}