Return last item of strings.Split() slice in Golang

patrick picture patrick · May 13, 2018 · Viewed 15.4k times · Source

I am splitting file names in Go to get at the file extension (e.g. import ("strings") ; strings.Split("example.txt", ".")). For this reason, I would like to return the last item in the slice returned by the split, i.e.

for strings.Split("ex.txt", "."), I want txt

This question suggests that doing

strings.Split("ex.txt", ".")[len(strings.Split("ex.txt", ".")) - 1]

is the only way to get at it. That is, there is no -1 as in Python. This seems very wasteful to me, as I feel we are doing the same splitting operation twice.

  • Is there no better command for getting the last item of a slice in Go?
  • If no, would the best approach be to write the result of Split into a variable, or just do the above?

Thank you!

Answer

abarnert picture abarnert · May 13, 2018

You should assign the results of the split to a variable, instead of calling it twice.

ss := strings.Split(msg, ".")
s := ss[len(ss)-1]

(Notice that this allows (or maybe forces) you to deal with the case where ss is empty or something else unexpected explicitly, before indexing it.)

If you're doing this over and over again, and it offends you having to use two lines (or two lines plus error handling) instead of one, you can abstract it into a function easily:

func lastString(ss []string) string {
    return ss[len(ss)-1]
}

s1 := lastString(strings.Split("example.txt", "."))
s2 := lastString(strings.Split("example.jpg", "."))

After all, passing the result of a function as an argument has essentially the same effect as binding it to a variable.